Spaces:
Sleeping
Sleeping
Update agents.py
Browse files
agents.py
CHANGED
@@ -13,7 +13,6 @@ from langchain_community.vectorstores import SupabaseVectorStore
|
|
13 |
|
14 |
from langchain_core.messages import SystemMessage, HumanMessage
|
15 |
from langchain_core.tools import tool
|
16 |
-
from langchain.tools.retriever import create_retriever_tool
|
17 |
|
18 |
from supabase.client import create_client, Client
|
19 |
|
@@ -129,8 +128,7 @@ def create_agent_flow(provider: str = "groq"):
|
|
129 |
graph = StateGraph(MessagesState)
|
130 |
graph.add_node("assistant", assistant_node)
|
131 |
graph.add_node("tools", ToolNode(toolset))
|
132 |
-
graph.add_edge(START, "
|
133 |
-
graph.add_edge("retriever", "assistant")
|
134 |
graph.add_conditional_edges("assistant", tools_condition)
|
135 |
graph.add_edge("tools", "assistant")
|
136 |
|
|
|
13 |
|
14 |
from langchain_core.messages import SystemMessage, HumanMessage
|
15 |
from langchain_core.tools import tool
|
|
|
16 |
|
17 |
from supabase.client import create_client, Client
|
18 |
|
|
|
128 |
graph = StateGraph(MessagesState)
|
129 |
graph.add_node("assistant", assistant_node)
|
130 |
graph.add_node("tools", ToolNode(toolset))
|
131 |
+
graph.add_edge(START, "assistant")
|
|
|
132 |
graph.add_conditional_edges("assistant", tools_condition)
|
133 |
graph.add_edge("tools", "assistant")
|
134 |
|