Feat: species differences in ST

app.py

@@ -46,23 +46,27 @@ def get_star_trek_quote(character: str)-> str: #it's import to specify the retur
     else:
         return f"Sorry, no quotes available for {character}. Please provide a valid Star Trek character name."
 
-# To invoke this tool, just type something like "How many tribbles are there? or "Check the Tribble infestation!" 
 @tool
-def tribbles_counter() -> str:
+def analyze_species_difference(species1: str, species2: str) -> str:
     """
-    Simulates the number of Tribbles aboard a starship.
+    Analyze the physiological difference between two Star Trek species using a search engine.
+
+    Args:
+        species1: The first species (e.g., "Vulkan", "Human").
+        species2: The second species (e.g., "Klingon", "Romulan").
 
     Returns:
-        A string with the count of Tribbles aboard.
+        A detailed analysis based on the search of results.
     """
-    tribbles_counter = random.randint(1,1000)
-    if tribbles_count > 500:
-        return f"There are {tribbles_count} Tribbles aboard! It's a *serious* infestation!"
-    elif tribbles_count > 200:
-        return f"There are {tribbles_count} Tribbles aboard. It's getting out of hand!"
-    else:
-        return f"There are {tribbles_count} Tribbles aboard. It's under control...for now."
+    query = f"Physiological differences between {species1} and {species2} in Star Trek"
 
+    search_results = DuckDuckGoSearchTool(query)
+
+    if search_results:
+        return f"Based on the research, here's a comparison between {species1} and {species2}"
+    else:
+        return f"Sorry, I couldn't find information on the physiological differences between {species1} and {species2}. Please, try another query."
+        
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -100,7 +104,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_star_trek_quote, tribbles_counter], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_star_trek_quote, analyze_species_difference], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,