Update tools.py

tools.py

@@ -65,23 +65,44 @@ class HubStatsTool(Tool):
         except Exception as e:
             return f"Error fetching models for {author}: {str(e)}"
 
-class CalculatorTool(Tool):
-    name = "calculator"
-    description = "Performs mathematical calculations."
+class CalendarTool(Tool):
+    name = "calendar"
+    description = "Manages and retrieves information about dates and events."
     inputs = {
-        "expression": {
+        "action": {
+            "type": "string",
+            "description": "The action to perform (e.g., 'add', 'get', 'delete')."
+        },
+        "date": {
             "type": "string",
-            "description": "The mathematical expression to evaluate."
+            "description": "The date of the event (format: YYYY-MM-DD)."
+        },
+        "event": {
+            "type": "string",
+            "description": "The event description.",
+            "nullable": True  # Add this line to specify that 'event' is nullable
         }
     }
     output_type = "string"
 
-    def forward(self, expression: str):
-        try:
-            result = eval(expression)
-            return f"The result of the expression '{expression}' is {result}."
-        except Exception as e:
-            return f"Error evaluating expression: {str(e)}"
+    def __init__(self):
+        self.events = {}
+
+    def forward(self, action: str, date: str, event: str = None):
+        if action == "add":
+            self.events[date] = event
+            return f"Event '{event}' added to {date}."
+        elif action == "get":
+            return f"Event on {date}: {self.events.get(date, 'No event found.')}"
+        elif action == "delete":
+            if date in self.events:
+                del self.events[date]
+                return f"Event on {date} deleted."
+            else:
+                return f"No event found on {date}."
+        else:
+            return "Invalid action."
+
 
 class CalendarTool(Tool):
     name = "calendar"