From 4089d5e40b9c3c65bf57c94a62bc3679dfa2d9bf Mon Sep 17 00:00:00 2001
From: Antonino Lorenzo <94693967+antoninoLorenzo@users.noreply.github.com>
Date: Sat, 9 Nov 2024 14:43:19 +0100
Subject: [PATCH] Fixed import in api.py

---
 src/api.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/api.py b/src/api.py
index b150f82..c33c448 100644
--- a/src/api.py
+++ b/src/api.py
@@ -1,7 +1,9 @@
 """
 API Interface for AI-OPS, here is provided the list of available endpoints.
+# TODO: dependency injection of Agent
+# TODO: separate routes
 
-Session Related:
+Session routes:
 - /session/list                    : Return all sessions.
 - /session/get/{sid}               : Return a specific session by ID.
 - /session/new/{name}              : Creates a new session.
@@ -12,7 +14,7 @@
 Agent Related:
 - /session/{sid}/query/{q}: Makes a query to the Agent.
 
-RAG Related:
+RAG Routes:
 - /collections/list    : Returns available Collections.
 - /collections/new     : Creates a new Collection.
 - /collections/upload/ : Upload document to an existing Collection
@@ -33,7 +35,6 @@
 from src.agent import Agent
 from src.agent.knowledge import Store, Collection
 from src.agent.llm import LLM, AVAILABLE_PROVIDERS
-from src.agent.plan import TaskStatus
 
 load_dotenv()
 TR = ToolRegistry()