Update tool.py

tool.py

@@ -1,6 +1,20 @@
 from smolagents import Tool
 from typing import Optional
 
+from enum import Enum
+
+class Diet(Enum):
+    NONE = "none"
+    VEGETARIAN = "vegetarian"
+    VEGAN = "vegan"
+    GLUTEN_FREE = "gluten free"
+    KETOGENIC = "ketogenic"
+    PALEO = "paleo"
+    PESCETARIAN = "pescetarian"
+    WHOLE30 = "whole30"
+    HALAL = "halal"
+    LOW_FODMAP = "low fodmap"
+
 class SimpleTool(Tool):
     name = "CulinAI -- your AI Recipe Assistant"
     description = "Gets a recipe suggestion based on provided ingredients, dietary preference, and your laziness level (1=active chef, 10=super lazy). After finding a recipe, itretrieves detailed recipe information."
@@ -10,20 +24,9 @@ class SimpleTool(Tool):
             "description": "A comma-separated string of available ingredients."
         },
         "diet": {
-            "type": "string",
-            "enum": [
-                "none",
-                "vegetarian", 
-                "vegan", 
-                "gluten free",
-                "ketogenic",
-                "paleo",
-                "pescetarian",
-                "whole30",
-                "halal",
-                "low fodmap"
-            ],
-            "default": "none",
+            "type": "enum",
+            "enum": Diet,
+            "default": Diet.NONE,
             "nullable": True,
             "description": "Select dietary restriction from the available options."
         },
@@ -35,7 +38,7 @@ class SimpleTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, ingredients: str, diet: Optional[str] = "none", laziness: Optional[int] = 5) -> str:
+    def forward(self, ingredients: str, diet: Optional[Diet] = Diet.NONE, laziness: Optional[int] = 5) -> str:
         """
         Gets a recipe suggestion based on provided ingredients, dietary preference, 
         and your laziness level (1=active chef, 10=super lazy). After finding a recipe, it
@@ -51,14 +54,6 @@ class SimpleTool(Tool):
         """
 
 
-        # Validate diet option
-        valid_diets = {
-            "none", "vegetarian", "vegan", "gluten free", "ketogenic",
-            "paleo", "pescetarian", "whole30", "halal", "low fodmap"
-        }
-        if diet and diet not in valid_diets:
-            return f"Invalid diet option. Please choose from: {', '.join(sorted(valid_diets))}"
-            
         import os
         import requests
 
@@ -76,8 +71,8 @@ class SimpleTool(Tool):
         }
 
         # Add diet if provided and not "none"
-        if diet and diet != "none":
-            params["diet"] = diet
+        if diet and diet != Diet.NONE:
+            params["diet"] = diet.value 
 
         # Incorporate the laziness factor: filter by maxReadyTime if needed.
         try: