Skip to content

Commit 93d3529

Browse files
committed
Fix async
1 parent 6a3fb67 commit 93d3529

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

services/chatbot/src/chatbot/langgraph_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ async def build_langgraph_agent(api_key, model_name, user_jwt):
6969
mcp_tools = await mcp_client.get_tools()
7070
db_tools = toolkit.get_tools()
7171
tools = mcp_tools + db_tools
72-
retriever_tool = await get_retriever_tool(api_key)
72+
retriever_tool = get_retriever_tool(api_key)
7373
tools.append(retriever_tool)
7474
agent_node = create_react_agent(model=llm, tools=tools, prompt=system_prompt)
7575
return agent_node

services/chatbot/src/chatbot/retrieverutils.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,7 @@ def get_chroma_vectorstore(api_key):
6262
return vectorstore
6363

6464

65-
def add_to_chroma_collection(
66-
api_key, session_id, new_messages: dict[str, str]
67-
) -> list:
65+
def add_to_chroma_collection(api_key, session_id, new_messages: dict[str, str]) -> list:
6866
vectorstore = get_chroma_vectorstore(api_key)
6967
res: list = vectorstore.add_documents(
7068
documents=[
@@ -75,7 +73,7 @@ def add_to_chroma_collection(
7573
return res
7674

7775

78-
async def get_retriever_tool(api_key):
76+
def get_retriever_tool(api_key):
7977
vectorstore = get_chroma_vectorstore(api_key)
8078
retriever = vectorstore.as_retriever()
8179
retriever_tool = create_retriever_tool(

0 commit comments

Comments
 (0)