Update rag_langgraph.py
Browse files- rag_langgraph.py +2 -3
rag_langgraph.py
CHANGED
@@ -128,18 +128,17 @@ def run_multi_agent(prompt):
|
|
128 |
|
129 |
graph = workflow.compile()
|
130 |
|
131 |
-
result = ""
|
132 |
|
133 |
for s in graph.stream(
|
134 |
{
|
135 |
"messages": [
|
136 |
-
HumanMessage(content=
|
137 |
]
|
138 |
}
|
139 |
):
|
140 |
if "__end__" not in s:
|
141 |
print(s)
|
142 |
-
result = str(s)
|
143 |
print("----")
|
144 |
|
145 |
return result
|
|
|
128 |
|
129 |
graph = workflow.compile()
|
130 |
|
131 |
+
result = "DONE"
|
132 |
|
133 |
for s in graph.stream(
|
134 |
{
|
135 |
"messages": [
|
136 |
+
HumanMessage(content=prompt)
|
137 |
]
|
138 |
}
|
139 |
):
|
140 |
if "__end__" not in s:
|
141 |
print(s)
|
|
|
142 |
print("----")
|
143 |
|
144 |
return result
|