Update agent.py

agent.py

@@ -1,9 +1,6 @@
 import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, LiteLLMModel
-from dotenv import load_dotenv
-load_dotenv()
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
 
-api_key = os.getenv("OPENAI_API_KEY")
 
 @tool
 def add(a: int, b: int) -> int:
@@ -59,7 +56,7 @@ def divide(a: int, b: int) -> float:
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        llm = LiteLLMModel("openai/gpt-4o-mini", api_key = api_key)
+        llm = InferenceClientModel()
         
         self.agent = CodeAgent(
             tools=[add,subtract,multiply,divide,DuckDuckGoSearchTool()],