Update app.py

app.py

@@ -9,6 +9,7 @@ from typing import TypedDict, List, Dict, Any, Optional
 from langgraph.graph import StateGraph, END
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import HumanMessage
+from smol_agents.llms import InferenceClientModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -140,10 +141,14 @@ class BasicAgent:
         If you are not able to find the answer using the tools privided, you can use the web_search tool.
         If you are given a task to create an image,you can use the image_generation_tool.
         """
-        llm = ChatOpenAI(model="gpt-4o", temperature=0)
+        model = InferenceClientModel(
+            model_id="gpt-4.5-preview",
+            token=os.environ["OPENAI_API_KEY"],
+            provider="openai"
+        )
         self.agent= CodeAgent(
             tools = [add, subtract, multiply, divide, web_search, image_generation_tool],
-            model=llm,
+            model=model,
         )
     def __call__(self, question: str, context: str = "") -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")