Update agent.py

agent.py

@@ -20,9 +20,9 @@ langsmith_api_key = os.getenv("LANGSMITH_API_KEY")
 langsmith_tracing = os.getenv("LANGSMITH_TRACING")
 
 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
+    base_url="https://api.groq.com/openai/v1",
+    api_key=os.getenv("GROQ_API_KEY"),
+    model="openai/gpt-oss-120b", # Model must support function calling in OpenRouter
     temperature=1
 )
 
@@ -104,10 +104,10 @@ gaia_agent = builder.compile()  # converts my builder into a runnable agent by u
 class LangGraphAgent:
     def __init__(self):
         self.api_keys = [
-            os.getenv("OPENROUTER_API_KEY"),
-            os.getenv("OPENROUTER_API_KEY_1"),
-            os.getenv("OPENROUTER_API_KEY_2"),
-            os.getenv("OPENROUTER_API_KEY_3")
+            os.getenv("GROQ_API_KEY"),
+            os.getenv("GROQ_API_KEY_1"),
+            os.getenv("GROQ_API_KEY_2"),
+            os.getenv("GROQ_API_KEY_3")
         ]
         self.key_index = 0
         print("LangGraphAgent initialized.")
@@ -119,7 +119,7 @@ class LangGraphAgent:
         llm = ChatOpenAI(
             base_url="https://openrouter.ai/api/v1",
             api_key=api_key,
-            model="moonshotai/kimi-k2:free",
+            model="openai/gpt-oss-120b",
             temperature=1
         )
         llm_with_tools = llm.bind_tools(tools)