Add task description field to CreateTaskTool and UpdateTaskTool for enhanced task details

agency_ai_demo/agents/NotionProjectAgent/tools/CreateTask.py

@@ -28,6 +28,11 @@ class CreateTaskTool(BaseTool):
         description="The title of the task (required).",
     )
 
+    task_description: str = Field(
+        default=None,
+        description="A text description of the task.",
+    )
+
     status: str = Field(
         default=None,
         description="Status of the task. Options: Backlog, In Progress, In Review, Testing, Completed.",
@@ -105,6 +110,14 @@ class CreateTaskTool(BaseTool):
             "title": [{"type": "text", "text": {"content": self.title}}]
         }
 
+        # Add task description if provided
+        if self.task_description:
+            properties["Task Description"] = {
+                "rich_text": [
+                    {"type": "text", "text": {"content": self.task_description}}
+                ]
+            }
+
         # Add status if provided
         if self.status:
             properties["Status"] = {"status": {"name": self.status}}

agency_ai_demo/agents/NotionProjectAgent/tools/GetTask.py

@@ -13,7 +13,7 @@ class GetTaskTool(BaseTool):
     """
     Tool for retrieving a specific task (page) from Notion.
     This tool fetches the properties of a specific task using its page ID and
-    by default, also fetches the page content (blocks) with all nested blocks.
+    by default, does not fetch the page content (blocks). Content can be included by setting include_content to True.
     """
 
     # Add example_field with a default value to satisfy BaseTool validation
@@ -28,7 +28,7 @@ class GetTaskTool(BaseTool):
     )
 
     include_content: bool = Field(
-        default=True,
+        default=False,
         description="Whether to include the page content (blocks) in the response. When True, all nested blocks will be included.",
     )
 

agency_ai_demo/agents/NotionProjectAgent/tools/GetTasks.py

@@ -19,37 +19,37 @@ class GetTasksTool(BaseTool):
     # Add example_field with a default value to satisfy BaseTool validation
     example_field: str = Field(
         default="notion_tasks",
-        description="Identifier for this tool. Can be left at its default value."
+        description="Identifier for this tool. Can be left at its default value.",
     )
 
     status: str = Field(
         default=None,
-        description="Filter tasks by status. Options: Backlog, In Progress, In Review, Testing, Completed."
+        description="Filter tasks by status. Options: Backlog, In Progress, In Review, Testing, Completed.",
     )
-    
+
     priority: str = Field(
         default=None,
-        description="Filter tasks by priority. Options: High, Medium, Low."
+        description="Filter tasks by priority. Options: High, Medium, Low.",
     )
-    
+
     due_date_before: str = Field(
         default=None,
-        description="Filter tasks due before this date (format: YYYY-MM-DD)."
+        description="Filter tasks due before this date (format: YYYY-MM-DD).",
     )
-    
+
     due_date_after: str = Field(
         default=None,
-        description="Filter tasks due after this date (format: YYYY-MM-DD)."
+        description="Filter tasks due after this date (format: YYYY-MM-DD).",
     )
-    
+
     sort_by: str = Field(
         default="Due Date",
-        description="Property to sort by. Options: Due Date, Status, Priority, Task Name."
+        description="Property to sort by. Options: Due Date, Status, Priority, Task Name.",
     )
-    
+
     sort_direction: str = Field(
         default="ascending",
-        description="Sort direction. Options: ascending, descending."
+        description="Sort direction. Options: ascending, descending.",
     )
 
     def run(self):
@@ -60,7 +60,7 @@ class GetTasksTool(BaseTool):
             dict: The JSON response from the Notion API containing the tasks.
         """
         import requests
-        
+
         # Use the database ID from the environment variable
         database_id = os.getenv("NOTION_DATABASE_ID")
 
@@ -76,61 +76,42 @@ class GetTasksTool(BaseTool):
 
         # Prepare the request body
         data = {}
-        
+
         # Build filter
         filters = []
-        
+
         if self.status:
-            filters.append({
-                "property": "Status",
-                "status": {
-                    "equals": self.status
-                }
-            })
-            
+            filters.append({"property": "Status", "status": {"equals": self.status}})
+
         if self.priority:
-            filters.append({
-                "property": "Priority",
-                "select": {
-                    "equals": self.priority
-                }
-            })
-            
+            filters.append(
+                {"property": "Priority", "select": {"equals": self.priority}}
+            )
+
         if self.due_date_before:
-            filters.append({
-                "property": "Due Date",
-                "date": {
-                    "before": self.due_date_before
-                }
-            })
-            
+            filters.append(
+                {"property": "Due Date", "date": {"before": self.due_date_before}}
+            )
+
         if self.due_date_after:
-            filters.append({
-                "property": "Due Date",
-                "date": {
-                    "after": self.due_date_after
-                }
-            })
-        
+            filters.append(
+                {"property": "Due Date", "date": {"after": self.due_date_after}}
+            )
+
         if filters:
             if len(filters) > 1:
-                data["filter"] = {
-                    "and": filters
-                }
+                data["filter"] = {"and": filters}
             else:
                 data["filter"] = filters[0]
-        
+
         # Add sorting
         if self.sort_by:
             data["sorts"] = [
-                {
-                    "property": self.sort_by,
-                    "direction": self.sort_direction
-                }
+                {"property": self.sort_by, "direction": self.sort_direction}
             ]
 
         # Make the request
         response = requests.post(url, headers=headers, json=data)
 
         # Return the JSON response
-        return response.json()
+        return response.json()

agency_ai_demo/agents/NotionProjectAgent/tools/UpdateTask.py

@@ -32,6 +32,11 @@ class UpdateTaskTool(BaseTool):
         description="The new title of the task.",
     )
 
+    task_description: str = Field(
+        default=None,
+        description="The new text description of the task.",
+    )
+
     status: str = Field(
         default=None,
         description="New status of the task. Options: Backlog, In Progress, In Review, Testing, Completed.",
@@ -91,6 +96,14 @@ class UpdateTaskTool(BaseTool):
                 "title": [{"type": "text", "text": {"content": self.title}}]
             }
 
+        # Add task description if provided
+        if self.task_description is not None:
+            properties["Task Description"] = {
+                "rich_text": [
+                    {"type": "text", "text": {"content": self.task_description}}
+                ]
+            }
+
         # Add status if provided
         if self.status is not None:
             properties["Status"] = {"status": {"name": self.status}}