Update app.py

app.py

@@ -10,9 +10,23 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-from smolagents.agent import ToolCallingAgent
-from smolagents.models.hf_api import HfApiModel
-from smolagents.tools.duckduckgo import DuckDuckGoSearchTool
+from smolagents.models.litellm import LiteLLMModel
+from smolagents.agent import CodeAgent
+
+class BasicAgent:
+    def __init__(self):
+        print("✅ BasicAgent with LiteLLM (OpenAI-compatible)")
+        self.model = LiteLLMModel(
+            model="gpt-3.5-turbo",
+            api_key=os.getenv("HF_API_KEY")  # Works with OpenRouter/OpenAI keys
+        )
+        self.agent = CodeAgent(
+            model=self.model,
+            verbose=True
+        )
+
+    def __call__(self, question: str) -> str:
+        return self.agent.run(question).strip()
 import os
 
 class BasicAgent: