Update tools.py

tools.py

@@ -20,6 +20,50 @@ import re
 from datetime import datetime, timedelta
 from langchain_core.tools import tool
 
+@tool
+def categorize_grocery_items(items: list[str]) -> dict:
+    """
+    Categorizes grocery items into food groups: vegetables, fruits, grains, dairy, snacks, nuts/legumes, spices, and beverages.
+
+    Args:
+        items (list[str]): List of grocery item names.
+
+    Returns:
+        dict: Dictionary mapping category names to sorted lists of items in that category.
+    """
+    print(f"🛠️ categorize_grocery_items called with: {items}")
+    
+    categories = {
+        "vegetables": {"broccoli", "celery", "green beans", "lettuce", "sweet potatoes", "zucchini"},
+        "fruits": {"plums", "bananas", "strawberries", "blueberries", "bell pepper"},
+        "grains": {"rice", "flour", "corn"},
+        "dairy": {"milk", "eggs"},
+        "snacks": {"Oreos"},
+        "nuts_and_legumes": {"peanuts", "acorns"},
+        "spices": {"whole allspice", "fresh basil"},
+        "beverages": {"whole bean coffee"},
+    }
+
+    result = {category: [] for category in categories}
+    result["other"] = []
+
+    for item in items:
+        found = False
+        for category, items_set in categories.items():
+            if item in items_set:
+                result[category].append(item)
+                found = True
+                break
+        if not found:
+            result["other"].append(item)
+
+    # Sort each category list
+    for key in result:
+        result[key] = sorted(result[key])
+
+    return result
+
+
 @tool
 def search_featured_articles_by_date_range(start_date: str, end_date: str) -> list[str]:
     """
@@ -232,6 +276,7 @@ def filter_vegetables(items: list[str]) -> list[str]:
 # List of all tools
 all_tools = [
     # search_featured_articles_by_date_range,
+    categorize_grocery_items,
     detect_non_commutative_subset,
     reverse_sentence,
     filter_vegetables,