Update tool.py

tool.py

@@ -1,19 +1,5 @@
 from smolagents import Tool
-from typing import Optional, Literal
-
-# Define valid diet options as a Literal type
-DietType = Literal[
-    "none", 
-    "vegetarian", 
-    "vegan", 
-    "gluten free",
-    "ketogenic",
-    "paleo",
-    "pescetarian",
-    "whole30",
-    "halal",
-    "low fodmap"
-]
+from typing import Optional
 
 class SimpleTool(Tool):
     name = "CulinAI -- your AI Recipe Assistant"
@@ -48,7 +34,7 @@ class SimpleTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, ingredients: str, diet: DietType = "none", laziness: Optional[int] = 5) -> str:
+    def forward(self, ingredients: str, diet: str = "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
@@ -62,6 +48,16 @@ class SimpleTool(Tool):
         Returns:
             A string with detailed information about the recommended recipe.
         """
+
+
+        # Validate diet option
+        valid_diets = {
+            "none", "vegetarian", "vegan", "gluten free", "ketogenic",
+            "paleo", "pescetarian", "whole30", "halal", "low fodmap"
+        }
+        if diet not in valid_diets:
+            return f"Invalid diet option. Please choose from: {', '.join(sorted(valid_diets))}"
+            
         import os
         import requests