Update veryfinal.py
Browse files- veryfinal.py +21 -21
veryfinal.py
CHANGED
@@ -281,30 +281,30 @@ class HybridLangGraphAgnoSystem:
|
|
281 |
g.add_node("agno_research",agno_research)
|
282 |
g.add_node("lg_retrieval",lg_retrieval)
|
283 |
g.add_node("agno_general",agno_general)
|
284 |
-
g.set_entry_point("router")
|
285 |
-
g.add_conditional_edges("router",pick,{
|
286 |
-
"lg_math":"lg_math","agno_research":"agno_research",
|
287 |
-
"lg_retrieval":"lg_retrieval","agno_general":"agno_general"
|
288 |
-
})
|
289 |
for n in ["lg_math","agno_research","lg_retrieval","agno_general"]:
|
290 |
-
g.add_edge(n,"END")
|
291 |
return g.compile(checkpointer=MemorySaver())
|
292 |
def process_query(self, q: str) -> Dict[str,Any]:
|
293 |
-
state={
|
294 |
-
"messages":[HumanMessage(content=q)],
|
295 |
-
"query":q,"agent_type":"","final_answer":"",
|
296 |
-
"perf":{},"agno_resp":""
|
297 |
-
}
|
298 |
-
cfg={"configurable":{"thread_id":f"hyb_{hash(q)}"}}
|
299 |
-
try:
|
300 |
-
out=self.graph.invoke(state,cfg)
|
301 |
-
return {
|
302 |
-
"answer":out["final_answer"],
|
303 |
-
"performance_metrics":out["perf"],
|
304 |
-
"provider_used":out["perf"].get("prov")
|
305 |
-
}
|
306 |
-
|
307 |
-
|
308 |
|
309 |
def build_graph(provider: str="hybrid"):
|
310 |
if provider=="hybrid":
|
|
|
281 |
g.add_node("agno_research",agno_research)
|
282 |
g.add_node("lg_retrieval",lg_retrieval)
|
283 |
g.add_node("agno_general",agno_general)
|
284 |
+
g.set_entry_point("router")
|
285 |
+
g.add_conditional_edges("router",pick,{
|
286 |
+
"lg_math":"lg_math","agno_research":"agno_research",
|
287 |
+
"lg_retrieval":"lg_retrieval","agno_general":"agno_general"
|
288 |
+
})
|
289 |
for n in ["lg_math","agno_research","lg_retrieval","agno_general"]:
|
290 |
+
g.add_edge(n,"END")
|
291 |
return g.compile(checkpointer=MemorySaver())
|
292 |
def process_query(self, q: str) -> Dict[str,Any]:
|
293 |
+
state={
|
294 |
+
"messages":[HumanMessage(content=q)],
|
295 |
+
"query":q,"agent_type":"","final_answer":"",
|
296 |
+
"perf":{},"agno_resp":""
|
297 |
+
}
|
298 |
+
cfg={"configurable":{"thread_id":f"hyb_{hash(q)}"}}
|
299 |
+
try:
|
300 |
+
out=self.graph.invoke(state,cfg)
|
301 |
+
return {
|
302 |
+
"answer":out["final_answer"],
|
303 |
+
"performance_metrics":out["perf"],
|
304 |
+
"provider_used":out["perf"].get("prov")
|
305 |
+
}
|
306 |
+
except Exception as e:
|
307 |
+
return {"answer":f"Error: {e}","performance_metrics":{},"provider_used":"Error"}
|
308 |
|
309 |
def build_graph(provider: str="hybrid"):
|
310 |
if provider=="hybrid":
|