fix

app.py

@@ -6,7 +6,7 @@ from transformers import logging
 
 from langgraph.checkpoint.memory import MemorySaver
 # from langchain_community.chat_models import ChatOpenRouter
-from langchain_openai import ChatOpenRouter
+from langchain_openai import ChatOpenAI
 
 from interface import create_demo
 from medrax.agent import *
@@ -57,8 +57,8 @@ def initialize_agent(
             tools_dict[tool_name] = all_tools[tool_name]()
 
     checkpointer = MemorySaver()
-    model = ChatOpenRouter(
-        model_name=model,
+    model = ChatOpenAI(
+        model_name=model,  # "google/gemini-2.5-pro-exp-03-25:free"
         api_key=os.getenv("OPENAI_API_KEY"),
         base_url=os.getenv("OPENAI_BASE_URL"),
         temperature=temperature,