Update agent.py

agent.py

@@ -152,7 +152,9 @@ import requests
 import mimetypes
 from PIL import Image
 from langchain_core.tools import tool
-from langchain.chat_models import ChatOpenAI
+# from langchain.chat_models import ChatOpenAI
+from langchain_community.chat_models import ChatOpenAI
+
 from langchain_core.pydantic_v1 import BaseModel
 from langgraph.graph import StateGraph, START, END
 from langchain_experimental.utilities import PythonREPL
@@ -162,13 +164,15 @@ from transformers import Blip2Processor, Blip2ForConditionalGeneration
 
 from tavily import TavilyClient
 
-DeepSeek_API_KEY= os.environ["DeepSeek_API_KEY"]
 GPT_4o_API_KEY  = os.environ["GPT_4o_API_KEY"]
 TAVILY_API_KEY  = os.environ["TAVILY_API_KEY"]
 Pixtral12B_API  = os.environ["Pixtral12B_API"]
 API_URL   = "https://openrouter.ai/api/v1/chat/completions"
 
-model = ChatOpenAI(base_url="https://openrouter.ai/api/v1", temperature=0)
+model = ChatOpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"),
+                   base_url="https://openrouter.ai/api/v1",
+                   model_name="gpt-4o",
+                   temperature=0)
 tavily = TavilyClient(api_key=TAVILY_API_KEY)