Spaces:
Sleeping
Sleeping
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":
|