Update app.py

app.py

@@ -96,12 +96,7 @@ class AgentState(TypedDict):
     answer: str
     task_id: str | None = None
 
-model=OpenAIServerModel(
-    model_id=model_id,
-    api_base="https://api.groq.com/openai/v1",
-    api_key=GROQ_API_KEY,
-    timeout=60,
-)
+
 
 # class WebSearchAgent:
         
@@ -161,8 +156,13 @@ _exhausted_models: set[str] = set()  # models that hit daily rate limits are ski
 # --------------------------------------------------------------------------- #
 #                         NODES  (LangGraph  functions)                        #
 # --------------------------------------------------------------------------- #
-_llm_router = model
-_llm_answer = model
+_llm_router = ChatOpenAI(
+    model=MODEL_CONFIGS[0]["model_id"],
+    base_url="https://api.groq.com/openai/v1",
+    api_key=GROQ_API_KEY,
+    timeout=60,
+)
+_llm_answer = _llm_router
 
 def route_question(state: AgentState) -> AgentState:
     """Label the task so we know which toolchain to invoke."""
@@ -293,7 +293,16 @@ def build_graph() -> StateGraph:
 class LGAgent:
     """Callable wrapper used by run_and_submit_all."""
 
-    def __init__(self) -> None:
+    def __init__(self, model_id: str | None = None) -> None:
+        global _llm_router, _llm_answer
+        mid = model_id or MODEL_CONFIGS[0]["model_id"]
+        _llm_router = ChatOpenAI(
+            model=mid,
+            base_url="https://api.groq.com/openai/v1",
+            api_key=GROQ_API_KEY,
+            timeout=60,
+        )
+        _llm_answer = _llm_router
         self.graph = build_graph()
 
     def __call__(self, question: str, task_id: str | None = None) -> str: