Update agent.py

agent.py

@@ -19,28 +19,12 @@ from langchain_core.runnables import RunnableConfig  # for LangSmith tracking
 langsmith_api_key = os.getenv("LANGSMITH_API_KEY")
 langsmith_tracing = os.getenv("LANGSMITH_TRACING")
 
-def get_llm():
-    api_keys = [os.getenv("OPENROUTER_API_KEY"), os.getenv("OPENROUTER_API_KEY_1")]
-    last_exception = None
-    for key in api_keys:
-        if not key:
-            continue
-        try:
-            llm = ChatOpenAI(
-                base_url="https://openrouter.ai/api/v1",
-                api_key=key,
-                model="moonshotai/kimi-k2:free",
-                temperature=1
-            )
-            # Optionally, test the key with a trivial call to ensure it's valid
-            # llm.invoke([SystemMessage(content="ping")])
-            return llm
-        except Exception as e:
-            last_exception = e
-            continue
-    raise RuntimeError(f"All OpenRouter API keys failed: {last_exception}")
-
-llm = get_llm()
+llm = ChatOpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key=os.getenv("OPENROUTER_API_KEY"),
+    model="moonshotai/kimi-k2:free", # Model must support function calling in OpenRouter
+    temperature=1
+)
 python_tool = PythonAstREPLTool()
 search_tool = BraveSearch.from_api_key(
                     api_key=os.getenv("BRAVE_SEARCH_API"),