Update veryfinal.py

veryfinal.py

@@ -1,5 +1,4 @@
 """Enhanced LangGraph + Agno Hybrid Agent System"""
-
 import os
 import time
 import random
@@ -160,7 +159,7 @@ def modulus(a: int, b: int) -> int:
 
 @tool
 def optimized_web_search(query: str) -> str:
-    """Optimized Tavily web search for speed."""
+    """Optimized Tavily web search."""
     try:
         time.sleep(random.uniform(1, 2))
         docs = TavilySearchResults(max_results=2).invoke(query=query)
@@ -173,7 +172,7 @@ def optimized_web_search(query: str) -> str:
 
 @tool
 def optimized_wiki_search(query: str) -> str:
-    """Optimized Wikipedia search for speed."""
+    """Optimized Wikipedia search."""
     try:
         time.sleep(random.uniform(0.5, 1))
         docs = WikipediaLoader(query=query, load_max_docs=1).load()
@@ -283,35 +282,43 @@ class HybridLangGraphAgnoSystem:
         g.add_node("agno_general",agno_general)
         g.set_entry_point("router")
         g.add_conditional_edges("router",pick,{
-        "lg_math":"lg_math","agno_research":"agno_research",
-        "lg_retrieval":"lg_retrieval","agno_general":"agno_general"
+            "lg_math":"lg_math","agno_research":"agno_research",
+            "lg_retrieval":"lg_retrieval","agno_general":"agno_general"
         })
         for n in ["lg_math","agno_research","lg_retrieval","agno_general"]:
             g.add_edge(n,"END")
         return g.compile(checkpointer=MemorySaver())
-def process_query(self, q: str) -> Dict[str,Any]:
-    state={
-    "messages":[HumanMessage(content=q)],
-    "query":q,"agent_type":"","final_answer":"",
-    "perf":{},"agno_resp":""
-    }
-    cfg={"configurable":{"thread_id":f"hyb_{hash(q)}"}}
-    try:
-        out=self.graph.invoke(state,cfg)
-        return {
-        "answer":out["final_answer"],
-        "performance_metrics":out["perf"],
-        "provider_used":out["perf"].get("prov")
-        }
-    except Exception as e:
-        return {"answer":f"Error: {e}","performance_metrics":{},"provider_used":"Error"}
 
-def build_graph(provider: str="hybrid"):
-    if provider=="hybrid":
+    def process_query(self, q: str) -> Dict[str,Any]:
+        state={
+            "messages":[HumanMessage(content=q)],
+            "query":q,"agent_type":"","final_answer":"",
+            "perf":{},"agno_resp":""
+        }
+        cfg={"configurable":{"thread_id":f"hyb_{hash(q)}"}}
+        try:
+            out=self.graph.invoke(state,cfg)
+            return {
+                "answer":out["final_answer"],
+                "performance_metrics":out["perf"],
+                "provider_used":out["perf"].get("prov")
+            }
+        except Exception as e:
+            return {"answer":f"Error: {e}","performance_metrics":{},"provider_used":"Error"}
+
+def build_graph(provider: str = "hybrid"):
+    """
+    Build and return the StateGraph for the requested provider.
+    - "hybrid", "groq", "google", and "nvidia" are all valid and
+      will return the full HybridLangGraphAgnoSystem graph.
+    """
+    if provider in ("hybrid", "groq", "google", "nvidia"):
         return HybridLangGraphAgnoSystem().graph
-    raise ValueError("Only 'hybrid' supported")
+    else:
+        raise ValueError(f"Unsupported provider: '{provider}'. Please use 'hybrid', 'groq', 'google', or 'nvidia'.")
 
-if __name__ == "__main__":
+# Test
+if __name__=="__main__":
     graph=build_graph()
     msgs=[HumanMessage(content="What are the names of the US presidents who were assassinated?")]
     res=graph.invoke({"messages":msgs},{"configurable":{"thread_id":"test"}})