entl commited on
Commit
1683024
·
1 Parent(s): dcc2c9a

disable rag

Browse files
Files changed (1) hide show
  1. agent.py +1 -4
agent.py CHANGED
@@ -1,4 +1,3 @@
1
- """LangGraph Agent"""
2
  import os
3
  from dotenv import load_dotenv
4
  from langgraph.graph import START, StateGraph, MessagesState
@@ -188,11 +187,9 @@ def build_graph(provider: str = "groq"):
188
  return {"messages": [sys_msg] + state["messages"] + [example_msg]}
189
 
190
  builder = StateGraph(MessagesState)
191
- builder.add_node("retriever", retriever)
192
  builder.add_node("assistant", assistant)
193
  builder.add_node("tools", ToolNode(tools))
194
- builder.add_edge(START, "retriever")
195
- builder.add_edge("retriever", "assistant")
196
  builder.add_conditional_edges(
197
  "assistant",
198
  tools_condition,
 
 
1
  import os
2
  from dotenv import load_dotenv
3
  from langgraph.graph import START, StateGraph, MessagesState
 
187
  return {"messages": [sys_msg] + state["messages"] + [example_msg]}
188
 
189
  builder = StateGraph(MessagesState)
 
190
  builder.add_node("assistant", assistant)
191
  builder.add_node("tools", ToolNode(tools))
192
+ builder.add_edge(START, "assistant")
 
193
  builder.add_conditional_edges(
194
  "assistant",
195
  tools_condition,