samoye16 commited on
Commit
9258ac0
·
verified ·
1 Parent(s): 831badd

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +1 -3
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, "retriever")
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