Tool para vegetais

agent.py

@@ -14,7 +14,7 @@ class Agent:
         
         print("Initializing Agent....")
         print("**************************************************************************************")
-        print('........ Versão: Sem Temperature no supervisor, tratamento da abertura do excel ,   .....')
+        print('........ Versão: Tool para verificar se é um vegetal   .....')
         print("**************************************************************************************")
         
         print("--> Audio Agent")
@@ -39,7 +39,7 @@ class Agent:
             agents=[self.web_search_agent, self.audio_agent],
             tools=[bird_video_count_tool,chess_image_to_fen_tool,chess_fen_get_best_next_move_tool,
                 get_excel_columns_tool, calculate_excel_sum_by_columns_tool,execute_python_code_tool,
-                text_inverter_tool, check_table_commutativity_tool],
+                text_inverter_tool, check_table_commutativity_tool, is_vegetable_tool],
             prompt= SUPERVISOR_PROMPT,
             add_handoff_back_messages=True,
             output_mode="last_message",

tools.py

@@ -2,6 +2,7 @@ from contextlib import redirect_stderr, redirect_stdout
 import io
 import json
 import os
+import re
 import subprocess
 import traceback
 from typing import Dict, List, Literal, Optional
@@ -623,4 +624,46 @@ def calculate_excel_sum_by_columns_tool(
 
     df = pd.read_excel(final_excel_path)
     total = df[include_columns].sum().sum()  # soma todas as colunas e depois soma os totais
-    return total
+    return total
+
+# Lista curada de vegetais culinários
+VEGETABLES = {
+    "lettuce", "carrot", "broccoli", "spinach", "kale", "celery", "cabbage",
+    "sweet potato", "radish", "turnip", "cauliflower", "beet", "onion", "garlic",
+    "green bean", "pea", "chard", "arugula", "basil", "parsley", "dill", "leek",
+    "asparagus", "eggplant", "okra", "pumpkin", "squash", "yam", "collard green",
+    "mustard green", "brussels sprout", "scallion", "fennel", "rhubarb", "artichoke",
+    "endive", "escarole", "bok choy", "watercress", "turnip green"
+}
+
+COMMON_ADJECTIVES = {"fresh", "raw", "organic", "chopped", "sliced", "whole"}
+
+def normalize_item(text: str) -> str:
+    # Lowercase and remove common adjectives
+    words = [w for w in re.findall(r"\w+", text.lower()) if w not in COMMON_ADJECTIVES]
+    # Singularização básica
+    singular = []
+    for word in words:
+        if word.endswith("ies"):
+            singular.append(word[:-3] + "y")
+        elif word.endswith("oes"):
+            singular.append(word[:-2])
+        elif word.endswith("s") and not word.endswith("ss"):
+            singular.append(word[:-1])
+        else:
+            singular.append(word)
+    return " ".join(singular)
+
+def is_vegetable_tool(item_name: str) -> bool:
+    """
+        Verify if an item is a vegetable
+        
+        Args:
+            item_name: Name of the item
+            
+        Returns:
+            True if is vegetable, otherwise false
+    """ 
+    norm = normalize_item(item_name)
+    return norm in VEGETABLES
+