Update app.py

app.py

@@ -11,17 +11,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
-from smolagents import CodeAgent
-from smolagents.models import OpenAIChat
-from my_tools import my_tool_list  # 你需要新建/补全这个模块和列表
+from smolagents import CodeAgent, LiteLLMModel 
+from my_tools import my_tool_list  
 
 class BasicAgent:
     def __init__(self):
         api_key = os.getenv("OPENAI_API_KEY")   # ← 这里自动读取环境变量
         if not api_key:
             raise ValueError("OPENAI_API_KEY not set in environment variables!")
-        self.model = OpenAIChat(model="gpt-4o", api_key=api_key)
-        self.agent = CodeAgent(model=self.model, tools=my_tool_list, max_steps=6)
+        model = LiteLLMModel(
+            model_id="gpt-4o",
+            api_key=api_key
+        )
+        
+        self.agent = CodeAgent(model=model, tools=my_tool_list, max_steps=6)
+
 
     
     def __call__(self, question: str) -> str: