fix: fix tools type in response model

Signed-off-by: Palaniappan R <[email protected]>
The-OpenROAD-Project · Oct 28, 2024 · 915e20b · 915e20b
1 parent b10177f
commit 915e20b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/backend/src/api/models/response_model.py b/backend/src/api/models/response_model.py
@@ -12,11 +12,11 @@ class ChatResponse(BaseModel):
     response: str
     sources: list[str] = []
     context: list[str] = []
-    tool: str
+    tools: list[str] = []
 
 
 class ChatToolResponse(BaseModel):
     response: str
     sources: list[str] = []
     context: list[str] = []
-    tool: str
+    tools: list[str] = []
diff --git a/backend/src/api/routers/graphs.py b/backend/src/api/routers/graphs.py
@@ -103,7 +103,6 @@ async def get_agent_response(user_input: UserInput) -> ChatResponse:
         output = list(rg.graph.stream(inputs))
     else:
         raise ValueError("RetrieverGraph not initialized.")
-
     urls: list[str] = []
     context: list[str] = []