Add scheduled task endpoints (#1450)

* Improve JWT expiration time

* Add scheduled task endpoints

* Add router

* get json response

Author: Josh-XT

agixt/app.py

@@ -18,6 +18,7 @@
 from endpoints.Provider import app as provider_endpoints
 from endpoints.Auth import app as auth_endpoints
 from endpoints.Health import app as health_endpoints
+from endpoints.Tasks import app as tasks_endpoints
 from endpoints.TeslaIntegration import register_tesla_routes
 from Globals import getenv
 from contextlib import asynccontextmanager
@@ -86,6 +87,7 @@ def signal_handler(signum, frame):
 
 
 app.include_router(agent_endpoints)
+app.include_router(tasks_endpoints)
 app.include_router(chain_endpoints)
 app.include_router(completions_endpoints)
 app.include_router(conversation_endpoints)

agixt/endpoints/Tasks.py

@@ -0,0 +1,156 @@
+from fastapi import APIRouter, Depends, Header
+from ApiClient import verify_api_key, get_api_client
+from Models import ResponseMessage
+from pydantic import BaseModel
+from Globals import getenv
+import logging
+import json
+
+
+app = APIRouter()
+
+logging.basicConfig(
+    level=getenv("LOG_LEVEL"),
+    format=getenv("LOG_FORMAT"),
+)
+
+
+class TaskModel(BaseModel):
+    agent_name: str
+    title: str
+    task_description: str
+    days: int = 0
+    hours: int = 0
+    minutes: int = 0
+    conversation_id: str = None
+
+
+class ReoccurringTaskModel(BaseModel):
+    agent_name: str
+    title: str
+    task_description: str
+    start_date: str
+    end_date: str
+    frequency: str = "daily"
+    conversation_id: str = None
+
+
+class ModifyTaskModel(BaseModel):
+    task_id: str
+    title: str = None
+    description: str = None
+    due_date: str = None
+    estimated_hours: str = None
+    priority: str = None
+    cancel_task: str = "false"
+    agent_name: str = None
+    conversation_id: str = None
+
+
+@app.post(
+    "/v1/task",
+    tags=["Tasks"],
+    response_model=ResponseMessage,
+    summary="Create a new task",
+    description="Create a new task with the specified parameters.",
+    dependencies=[Depends(verify_api_key)],
+)
+async def new_task(
+    task: TaskModel,
+    user=Depends(verify_api_key),
+    authorization: str = Header(None),
+) -> ResponseMessage:
+    agixt = get_api_client(authorization=authorization)
+    response = agixt.execute_command(
+        agent_name=task.agent_name,
+        command_name="Schedule Task",
+        command_args={
+            "task_description": task.task_description,
+            "days": task.days,
+            "hours": task.hours,
+            "minutes": task.minutes,
+        },
+        conversation_name=task.conversation_id,
+    )
+    return ResponseMessage(message=f"Task created for agent '{task.agent_name}'.")
+
+
+@app.post(
+    "/v1/reoccurring_task",
+    tags=["Tasks"],
+    response_model=ResponseMessage,
+    summary="Create a new reoccurring task",
+    description="Create a new reoccurring task with the specified parameters.",
+    dependencies=[Depends(verify_api_key)],
+)
+async def new_reoccurring_task(
+    task: ReoccurringTaskModel,
+    user=Depends(verify_api_key),
+    authorization: str = Header(None),
+) -> ResponseMessage:
+    agixt = get_api_client(authorization=authorization)
+    response = agixt.execute_command(
+        agent_name=task.agent_name,
+        command_name="Schedule Reoccurring Task",
+        command_args={
+            "task_description": task.task_description,
+            "start_date": task.start_date,
+            "end_date": task.end_date,
+            "frequency": task.frequency,
+        },
+        conversation_name=task.conversation_id,
+    )
+    return ResponseMessage(
+        message=f"Reoccurring task created for agent '{task.agent_name}'."
+    )
+
+
+@app.put(
+    "/v1/task",
+    tags=["Tasks"],
+    response_model=ResponseMessage,
+    summary="Modify an existing task",
+    description="Modify an existing task with new information, or cancel it.",
+    dependencies=[Depends(verify_api_key)],
+)
+async def modify_task(
+    task: ModifyTaskModel,
+    user=Depends(verify_api_key),
+    authorization: str = Header(None),
+) -> ResponseMessage:
+    agixt = get_api_client(authorization=authorization)
+    response = agixt.execute_command(
+        agent_name=task.agent_name,
+        command_name="Modify Scheduled Task",
+        command_args={
+            "task_id": task.task_id,
+            "title": task.title,
+            "description": task.description,
+            "due_date": task.due_date,
+            "estimated_hours": task.estimated_hours,
+            "priority": task.priority,
+            "cancel_task": task.cancel_task,
+        },
+        conversation_name=task.conversation_id,
+    )
+    return ResponseMessage(message=f"Task modified for agent '{task.agent_name}'.")
+
+
+@app.get(
+    "/v1/tasks",
+    tags=["Tasks"],
+    summary="Get all scheduled tasks",
+    description="Get all scheduled tasks for the current agent.",
+    dependencies=[Depends(verify_api_key)],
+)
+async def get_scheduled_tasks(
+    user=Depends(verify_api_key), authorization: str = Header(None)
+):
+    agixt = get_api_client(authorization=authorization)
+    response = agixt.execute_command(
+        agent_name="",
+        command_name="Get Scheduled Tasks",
+        command_args={},
+        conversation_name="",
+    )
+    return json.loads(response)

agixt/extensions/scheduled_tasks.py

@@ -103,7 +103,7 @@ async def schedule_task(
             memory_collection=self.conversation_id,  # This ensures context preservation
         )
 
-        return f"Scheduled follow-up task {task_id} for {due_date.strftime('%Y-%m-%d %H:%M:%S')}"
+        return f"Scheduled task {task_id} for {due_date.strftime('%Y-%m-%d %H:%M:%S')}"
 
     async def schedule_reoccurring_task(
         self,

agixt/version

@@ -1 +1 @@
-v1.7.8
+v1.7.9