Update langgraph_agent.py

langgraph_agent.py

@@ -66,7 +66,7 @@ def arvix_search(query: str) -> dict:
     )
     return {"arvix_results": formatted}
 
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+API_KEY = os.getenv("GEMINI_API_KEY")
 HF_SPACE_TOKEN = os.getenv("HF_SPACE_TOKEN")
 
 
@@ -82,13 +82,16 @@ sys_msg = SystemMessage(content=system_prompt)
 
 
 def build_graph(provider: str = "openai"):
-    """Build the LangGraph agent with chosen LLM (default: OpenAI)."""
+    """Build the LangGraph agent with chosen LLM (default: Gemini)."""
     if provider == "openai":
-        llm = ChatOpenAI(
-            model_name="o4-mini-2025-04-16",
-            openai_api_key=OPENAI_API_KEY,
-            # no temperature override here
-        )
+        lllm = ChatGoogleGenerativeAI(
+        model= "gemini-2.5-pro-preview-05-06",
+        temperature=1.0,
+        max_retries=2,
+        google_api_key=api_key,
+)
+
+        
     elif provider == "huggingface":
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(