Update tool.py

tool.py

@@ -1,14 +1,54 @@
 from smolagents import Tool
 from typing import Optional
 
+# Define valid diet options as a Literal type
+DietType = Literal[
+    "none", 
+    "vegetarian", 
+    "vegan", 
+    "gluten free",
+    "ketogenic",
+    "paleo",
+    "pescetarian",
+    "whole30",
+    "halal",
+    "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."
-    inputs = {"ingredients":{"type":"string","description":"A comma-separated string of available ingredients."},"diet":{"type":"string","nullable":True,"description":"Dietary restrictions such as 'vegetarian', 'vegan', or 'gluten free'. Defaults to None."},"laziness":{"type":"integer","nullable":True,"description":"An integer from 1 (active) to 10 (super lazy); higher means recipes with quicker prep."}}
+    inputs = {
+        "ingredients": {
+            "type": "string",
+            "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",
+            "description": "Select dietary restriction from the available options."
+        },
+        "laziness": {
+            "type": "integer",
+            "nullable": True,
+            "description": "An integer from 1 (active) to 10 (super lazy); higher means recipes with quicker prep."
+        }
+    }
     output_type = "string"
 
-
-    def forward(self, ingredients: str, diet: Optional[str] = None, laziness: Optional[int] = 5) -> str:
+    def forward(self, ingredients: str, diet: DietType = "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
@@ -38,8 +78,8 @@ class SimpleTool(Tool):
             "apiKey": api_key,
         }
 
-        # Add diet if provided
-        if diet:
+        # Add diet if provided and not "none"
+        if diet and diet != "none":
             params["diet"] = diet
 
         # Incorporate the laziness factor: filter by maxReadyTime if needed.