runs but guest info tool needs debuging

.gitignore

@@ -1,5 +1,6 @@
 # omit IDE configs
 .vscode/settings.json
+.gradio
 
 # OS clutter
 **/.DS_Store

app.py

@@ -2,28 +2,40 @@
 
 import os
 
-from smolagents import GradioUI, CodeAgent, InferenceClientModel
+from smolagents import CodeAgent, OpenAIServerModel, FinalAnswerTool
 
-from retriever import load_guest_dataset
+from guest_info import guest_info_tool_factory
 
-OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
-BASE_URL = os.environ["BASE_URL"]
 
+MODEL_ID = os.environ["MODEL_ID"]
+BASE_URL = os.environ["BASE_URL"]
+API_KEY = os.environ["OPENAI_API_KEY"]
+API_BASE = "/".join([BASE_URL, "v1"])
 
-model = InferenceClientModel(api_key=OPENAI_API_KEY, base_url=BASE_URL)
+model = OpenAIServerModel(model_id=MODEL_ID, api_base=API_BASE, api_key=API_KEY)
 
 
 # Load the guest dataset and initialize the guest info tool
-guest_info_tool = load_guest_dataset()
+guest_info_tool = guest_info_tool_factory()
+
+final_answer_tool = FinalAnswerTool()
 
 # Create Alfred with all the tools
 alfred = CodeAgent(
-    tools=[guest_info_tool],
     model=model,
-    add_base_tools=True,  # Add any additional base tools
+    tools=[guest_info_tool, final_answer_tool],
+    add_base_tools=False,  # Add any additional base tools
     planning_interval=3,  # Enable planning every 3 steps
 )
 
+# Question for agent
+# Tell me about our guest named 'Lady Ada Lovelace'
 
 if __name__ == "__main__":
-    GradioUI(alfred).launch()
+    #    GradioUI(alfred).launch(share=False)
+
+    # Example query Alfred might receive during the gala
+    response = alfred.run("Tell me about our guest named 'Lady Ada Lovelace'.")
+
+    print("🎩 Alfred's Response:")
+    print(response)

retriever.py → guest_info.py

@@ -30,7 +30,7 @@ class GuestInfoRetrieverTool(Tool):
         return "No matching guest information found."
 
 
-def load_guest_dataset():
+def guest_info_tool_factory():
     """Load the dataset"""
 
     DATASETS = [  # pylint: disable=invalid-name

requirements.txt

@@ -1,3 +1,5 @@
 datasets 
+duckduckgo-search
 langchain-community 
-smolagents
+rank_bm25
+smolagents[litellm]