Update agent.py

agent.py

@@ -4,16 +4,16 @@ from langgraph.graph.message import add_messages
 from langchain_core.messages import AnyMessage, SystemMessage
 from langgraph.prebuilt import ToolNode, tools_condition
 from langgraph.graph import START, StateGraph
-# from langchain_openai import ChatOpenAI
-from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_openai import ChatOpenAI
+# from langchain_google_genai import ChatGoogleGenerativeAI
 
 from tools import tavily_search_tool, repl_tool
 
-# OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
-GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
+# GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
-# chat_model = ChatOpenAI(model='gpt-4o-mini', temperature=0, api_key=OPENAI_API_KEY)
-chat_model = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=GOOGLE_API_KEY)
+chat_model = ChatOpenAI(model='gpt-4.1-mini', temperature=0, api_key=OPENAI_API_KEY)
+# chat_model = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=GOOGLE_API_KEY)
 tools = [tavily_search_tool, repl_tool]
 chat_model_with_tools = chat_model.bind_tools(tools)