Update retriever.py

retriever.py

@@ -3,9 +3,15 @@ from smolagents import Tool
 class FrugalAI_methods(Tool):
     name = "FrugalAI_methods"
     description = "Retrieves methods for model frugalization."
+    inputs = {
+        "method": {
+            "type": "string",
+            "description": "Always set it to 'Frugal' or it will raise an error."
+        }
+    }
     output_type = "string"
 
-    def pruning(self):
+    def pruning(self,method: str):
         """
         Optimizes models by removing unnecessary components, such as certain weights in a neural network. 
         This function demonstrates how to apply pruning.
@@ -18,7 +24,7 @@ class FrugalAI_methods(Tool):
             f"can be increased or decreased depending on the initial number of weights and the complexity of your use case (minimu value: 0, maximum value: 1)."
         )
 
-    def quantization(self):
+    def quantization(self, method: str):
         """
         Converts high-precision weights into lower-precision one to reduce cost. 
         """