Spaces:
Running
Running
File size: 3,593 Bytes
84c328d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 |
"""MCP tool for deleting tasks from the todo list.
[Task]: T047
[From]: specs/004-ai-chatbot/tasks.md
This tool allows the AI agent to permanently delete tasks
through natural language conversations.
"""
from typing import Optional, Any
from uuid import UUID
from datetime import datetime
from sqlalchemy import select
from models.task import Task
from core.database import engine
from sqlmodel import Session
# Tool metadata for MCP registration
tool_metadata = {
"name": "delete_task",
"description": """Delete a task from the user's todo list permanently.
Use this tool when the user wants to:
- Delete, remove, or get rid of a task
- Clear a task from their list
- Permanently remove a task
Parameters:
- user_id (required): User ID (UUID) who owns the task
- task_id (required): Task ID (UUID) of the task to delete
Returns: Confirmation of deletion with task details.
""",
"inputSchema": {
"type": "object",
"properties": {
"user_id": {
"type": "string",
"description": "User ID (UUID) who owns this task"
},
"task_id": {
"type": "string",
"description": "Task ID (UUID) of the task to delete"
}
},
"required": ["user_id", "task_id"]
}
}
async def delete_task(
user_id: str,
task_id: str
) -> dict[str, Any]:
"""Delete a task from the user's todo list.
[From]: specs/004-ai-chatbot/spec.md - US5
Args:
user_id: User ID (UUID string) who owns the task
task_id: Task ID (UUID string) of the task to delete
Returns:
Dictionary with deletion confirmation
Raises:
ValueError: If validation fails or task not found
"""
# Get database session (synchronous)
with Session(engine) as db:
try:
# Fetch the task
stmt = select(Task).where(
Task.id == UUID(task_id),
Task.user_id == UUID(user_id)
)
task = db.scalars(stmt).first()
if not task:
return {
"success": False,
"error": "Task not found",
"message": f"Could not find task with ID {task_id}"
}
# Store task details for confirmation
task_details = {
"id": str(task.id),
"title": task.title,
"description": task.description,
"due_date": task.due_date.isoformat() if task.due_date else None,
"priority": task.priority,
"completed": task.completed,
"created_at": task.created_at.isoformat(),
"updated_at": task.updated_at.isoformat()
}
# Delete the task
db.delete(task)
db.commit()
# Build success message
message = f"✅ Task '{task.title}' deleted successfully"
return {
"success": True,
"task": task_details,
"message": message
}
except ValueError as e:
db.rollback()
raise ValueError(f"Failed to delete task: {str(e)}")
# Register tool with MCP server
def register_tool(mcp_server: Any) -> None:
"""Register this tool with the MCP server.
[From]: backend/mcp_server/server.py
Args:
mcp_server: MCP server instance
"""
mcp_server.tool(
name=tool_metadata["name"],
description=tool_metadata["description"]
)(delete_task)
|