josondev commited on
Commit
0ab2059
·
verified ·
1 Parent(s): d3bd386

Update veryfinal.py

Browse files
Files changed (1) hide show
  1. 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
- 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":
 
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":