wip

app.py

@@ -1,4 +1,7 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, InferenceClientModel, Tool, tool, VisitWebpageTool
+from huggingface_hub import login
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, Tool, tool, VisitWebpageTool, FinalAnswerTool
+from dotenv import load_dotenv
+import os
 
 @tool
 def suggest_menu(occasion: str) -> str:
@@ -60,6 +63,16 @@ class SuperheroPartyThemeTool(Tool):
         
         return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")
 
+load_dotenv()
+login(
+    token=os.environ["HF_TOKEN"]
+)
+
+model = LiteLLMModel(
+    model_id="openai/gpt-4.1",
+    temperature=0.2,
+    api_key=os.environ["OPENAI_API_KEY"]
+)
 
 # Alfred, the butler, preparing the menu for the party
 agent = CodeAgent(