Update text_generator.py

text_generator.py

@@ -7,8 +7,8 @@ class TextGenerationTool(Tool):
     name = "text_generator"
     description = "This is a tool for text generation. It takes a prompt as input and returns the generated text."
     
-    inputs = ["text"]
-    outputs = ["text"]
+    inputs = {"text": {"type": "text", "description": "The prompt for text generation"}}
+    outputs = {"text": {"type": "text", "description": "The generated text"}}
     
     # Available text generation models
     models = {