submit

app.py

@@ -1,27 +1,42 @@
-import gradio as gr
+# Import necessary libraries
 import random
-from smolagents import GradioUI, CodeAgent, HfApiModel
+from smolagents import CodeAgent, InferenceClientModel
 
+# Import our custom tools from their modules
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
 from retriever import load_guest_dataset
+from dotenv import load_dotenv
+import os
+load_dotenv()
 
-model = HfApiModel()
+# Load the Hugging Face API key from environment variables
+api_key = os.getenv("HUGGINGFACE_API_KEY")
 
+# Initialize the Hugging Face model
+model = InferenceClientModel(token=api_key)
+
+# Initialize the web search tool
 search_tool = DuckDuckGoSearchTool()
 
+# Initialize the weather tool
 weather_info_tool = WeatherInfoTool()
 
+# Initialize the Hub stats tool
 hub_stats_tool = HubStatsTool()
 
+# Load the guest dataset and initialize the guest info tool
 guest_info_tool = load_guest_dataset()
 
 # Create Alfred with all the tools
 alfred = CodeAgent(
-    tools = [guest_info_tool, weather_info_tool, hub_stats_tool, search_tool],
-    model = model,
-    add_base_tools = True,
-    planning_interval = 3
+    tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], 
+    model=model,
+    add_base_tools=True,  # Add any additional base tools
+    planning_interval=3   # Enable planning every 3 steps
 )
 
-if __name__="__name__":
-    GradioUI(alfred).launch()
+query = "I need to speak with Dr. Nikola Tesla about recent advancements in wireless energy. Can you help me prepare for this conversation?"
+response = alfred.run(query)
+
+print("🎩 Alfred's Response:")
+print(response)

retriever.py

@@ -0,0 +1,74 @@
+import datasets
+from langchain.docstore.document import Document
+from smolagents import Tool
+from langchain_community.retrievers import BM25Retriever
+from smolagents import CodeAgent, InferenceClientModel
+import os
+# from huggingface_hub import HfApi, InferenceClient
+from dotenv import load_dotenv
+# import os
+
+load_dotenv()
+# Load the Hugging Face API key from environment variables
+api_key = os.getenv("HUGGINGFACE_API_KEY")
+
+
+class GuestInfoRetrieverTool(Tool):
+    name = "guest_info_retriever"
+    description = "Retrieves detailed information about gala guests based on their name or relation."
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "The name or relation of the guest you want information about."
+        }
+    }
+    output_type = "string"
+
+    def __init__(self, docs):
+        self.is_initialized = False
+        self.retriever = BM25Retriever.from_documents(docs)
+
+    def forward(self, query: str):
+        results = self.retriever.get_relevant_documents(query)
+        if results:
+            return "\n\n".join([doc.page_content for doc in results[:3]])
+        else:
+            return "No matching guest information found."
+        
+def load_guest_dataset():
+    # Load the dataset
+    guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
+
+    # Convert dataset entries into Document objects
+    docs = [
+        Document(
+            page_content="\n".join([
+                f"Name: {guest['name']}",
+                f"Relation: {guest['relation']}",
+                f"Description: {guest['description']}",
+                f"Email: {guest['email']}"
+            ]),
+            metadata={"name": guest["name"]}
+        )
+        for guest in guest_dataset
+    ]
+
+    # Return the tool
+    return GuestInfoRetrieverTool(docs)
+
+# Initialize the tool
+# guest_info_tool = GuestInfoRetrieverTool(docs)
+
+
+# Initialize the Hugging Face model
+model = InferenceClientModel(token=api_key)
+
+# Create Alfred, our gala agent, with the guest info tool
+# alfred = CodeAgent(tools=[guest_info_tool], model=model,
+#                    )
+
+# Example query Alfred might receive during the gala
+# response = alfred.run("Tell me about our guest named 'Nhung ham'.")
+
+# print("🎩 Alfred's Response:")
+# print(response)

tools.py

@@ -0,0 +1,89 @@
+from smolagents import DuckDuckGoSearchTool
+from smolagents import Tool
+import random
+# from smolagents import Tool
+from huggingface_hub import list_models
+from dotenv import load_dotenv
+import os
+load_dotenv()
+from smolagents import CodeAgent, InferenceClientModel
+
+# Load the Hugging Face API key from environment variables
+api_key = os.getenv("HUGGINGFACE_API_KEY")
+
+# Initialize the DuckDuckGo search tool
+search_tool = DuckDuckGoSearchTool()
+
+# Example usage
+# results = search_tool("Who's the current President of France?")
+# print(results)
+
+class WeatherInfoTool(Tool):
+    name = "weather_info"
+    description = "Fetches dummy weather information for a given location."
+    inputs = {
+        "location": {
+            "type": "string",
+            "description": "The location to get weather information for."
+        }
+    }
+    output_type = "string"
+
+    def forward(self, location: str):
+        # Dummy weather data
+        weather_conditions = [
+            {"condition": "Rainy", "temp_c": 15},
+            {"condition": "Clear", "temp_c": 25},
+            {"condition": "Windy", "temp_c": 20}
+        ]
+        # Randomly select a weather condition
+        data = random.choice(weather_conditions)
+        return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"
+
+# Initialize the tool
+weather_info_tool = WeatherInfoTool()
+
+class HubStatsTool(Tool):
+    name = "hub_stats"
+    description = "Fetches the most downloaded model from a specific author on the Hugging Face Hub."
+    inputs = {
+        "author": {
+            "type": "string",
+            "description": "The username of the model author/organization to find models from."
+        }
+    }
+    output_type = "string"
+
+    def forward(self, author: str):
+        try:
+            # List models from the specified author, sorted by downloads
+            models = list(list_models(author=author, sort="downloads", direction=-1, limit=1))
+            
+            if models:
+                model = models[0]
+                return f"The most downloaded model by {author} is {model.id} with {model.downloads:,} downloads."
+            else:
+                return f"No models found for author {author}."
+        except Exception as e:
+            return f"Error fetching models for {author}: {str(e)}"
+
+# Initialize the tool
+hub_stats_tool = HubStatsTool()
+
+# Example usage
+# print(hub_stats_tool("facebook")) # Example: Get the most downloaded model by Facebook
+
+# Initialize the Hugging Face model
+model = InferenceClientModel(token=api_key)
+
+# Create Alfred with all the tools
+alfred = CodeAgent(
+    tools=[search_tool, weather_info_tool, hub_stats_tool], 
+    model=model
+)
+
+# Example query Alfred might receive during the gala
+# response = alfred.run("I'am planning a trip to Paris. What is the weathere there, and can you tell me who the current mayor is? Also, what's the most popular machine learning model from French researchers?")
+
+# print("🎩 Alfred's Response:")
+# print(response)