Update task_agent.py

task_agent.py

@@ -7,7 +7,7 @@ class TaskAgent:
         self.tasks = []
         self.calendar_service = get_calendar_service()
     
-    def add_task(self, name, description, email):  # Removed phone parameter
+    def add_task(self, name, description, email):
         suggestion = self._safe_get_deadline(description)
         
         deadline = self._extract_field(suggestion, "Recommended Deadline")
@@ -25,4 +25,19 @@ class TaskAgent:
         if self.calendar_service:
             add_to_calendar(self.calendar_service, name, deadline)
         
-        return suggestion
+        return suggestion
+    
+    def _safe_get_deadline(self, description):
+        try:
+            return llm.get_deadline_suggestion(description)
+        except Exception as e:
+            return f"LLM unavailable: {str(e)}. Set deadline manually."
+    
+    def _extract_field(self, text, field):
+        for line in text.split("\n"):
+            if field in line:
+                return line.split(":", 1)[-1].strip()
+        return None
+
+# This line is CRITICAL — don't remove it
+agent = TaskAgent()