entl
commited on
Commit
·
1683024
1
Parent(s):
dcc2c9a
disable rag
Browse files
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, "
|
| 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,
|