Feature: API for Accessing & Managing Action Items

backend/database/action_items.py

@@ -0,0 +1,83 @@
+from datetime import datetime, timezone
+from typing import List, Optional
+from uuid import uuid4
+
+from google.cloud import firestore
+from google.cloud.firestore_v1.base_query import FieldFilter
+
+from models.action_item import ActionItem, ActionItemStatus
+from ._client import db
+
+def create_action_item(uid: str, description: str, memory_id: Optional[str] = None, due_date: Optional[datetime] = None) -> ActionItem:
+    now = datetime.now(timezone.utc)
+    action_item = ActionItem(
+        id=str(uuid4()),
+        memory_id=memory_id,
+        uid=uid,
+        description=description,
+        created_at=now,
+        updated_at=now,
+        due_date=due_date
+    )
+
+    db.collection('users').document(uid).collection('action_items').document(action_item.id).set(action_item.dict())
+    return action_item
+
+def get_action_items(
+    uid: str,
+    status: Optional[ActionItemStatus] = None,
+    memory_id: Optional[str] = None,
+    start_date: Optional[datetime] = None,
+    end_date: Optional[datetime] = None,
+    limit: int = 100,
+    offset: int = 0
+) -> List[ActionItem]:
+    query = db.collection('users').document(uid).collection('action_items')
+
+    filters = []
+    if status:
+        filters.append(FieldFilter('status', '==', status))
+    if memory_id:
+        filters.append(FieldFilter('memory_id', '==', memory_id))
+    if start_date:
+        filters.append(FieldFilter('created_at', '>=', start_date))
+    if end_date:
+        filters.append(FieldFilter('created_at', '<=', end_date))
+
+    if filters:
+        for f in filters:
+            query = query.where(filter=f)
+
+    query = query.order_by('created_at', direction=firestore.Query.DESCENDING).limit(limit).offset(offset)
+
+    return [ActionItem(**doc.to_dict()) for doc in query.stream()]
+
+def update_action_item(uid: str, action_item_id: str, updates: dict) -> Optional[ActionItem]:
+    doc_ref = db.collection('users').document(uid).collection('action_items').document(action_item_id)
+    doc = doc_ref.get()
+
+    if not doc.exists:
+        return None
+
+    updates['updated_at'] = datetime.now(timezone.utc)
+    if updates.get('status') == ActionItemStatus.COMPLETED:
+        updates['completed_at'] = datetime.now(timezone.utc)
+    doc_ref.update(updates)
+
+    updated_doc = doc_ref.get()
+    return ActionItem(**updated_doc.to_dict())
+
+def delete_action_item(uid: str, action_item_id: str) -> bool:
+    doc_ref = db.collection('users').document(uid).collection('action_items').document(action_item_id)
+    doc = doc_ref.get()
+
+    if not doc.exists:
+        return False
+
+    updates = {
+        'status': ActionItemStatus.DELETED,
+        'deleted_at': datetime.now(timezone.utc),
+        'updated_at': datetime.now(timezone.utc)
+    }
+    doc_ref.update(updates)
+    return True

backend/main.py

@@ -6,7 +6,7 @@
 
 from modal import Image, App, asgi_app, Secret
 from routers import workflow, chat, firmware, plugins, memories, transcribe_v2, notifications, \
-    speech_profile, agents, facts, users, processing_memories, trends, sdcard, sync, apps, custom_auth, payment
+    speech_profile, agents, facts, users, processing_memories, trends, sdcard, sync, apps, custom_auth, payment, action_items
 
 if os.environ.get('SERVICE_ACCOUNT_JSON'):
     service_account_info = json.loads(os.environ["SERVICE_ACCOUNT_JSON"])
@@ -37,6 +37,7 @@
 
 app.include_router(apps.router)
 app.include_router(custom_auth.router)
+app.include_router(action_items.router)
 
 app.include_router(payment.router)
 

backend/models/action_item.py

@@ -0,0 +1,21 @@
+from datetime import datetime
+from enum import Enum
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class ActionItemStatus(str, Enum):
+    OPEN = "open"
+    COMPLETED = "completed"
+    DELETED = "deleted"
+
+class ActionItem(BaseModel):
+    id: str = Field(description="Unique identifier for the action item")
+    memory_id: Optional[str] = Field(description="Associated memory ID if created from a memory")
+    uid: str = Field(description="User ID who owns this action item")
+    description: str = Field(description="The action item text/description")
+    status: ActionItemStatus = Field(default=ActionItemStatus.OPEN)
+    created_at: datetime = Field(description="When the action item was created")
+    updated_at: datetime = Field(description="Last time the action item was updated")
+    completed_at: Optional[datetime] = Field(default=None, description="When the action item was completed")
+    deleted_at: Optional[datetime] = Field(default=None, description="When the action item was deleted")
+    due_date: Optional[datetime] = Field(default=None, description="Optional due date for the action item")

backend/routers/action_items.py

@@ -0,0 +1,77 @@
+from datetime import datetime
+from typing import List, Optional
+from fastapi import APIRouter, Depends, HTTPException
+from pydantic import BaseModel
+
+from models.action_item import ActionItem, ActionItemStatus
+import database.action_items as action_items_db
+from utils.other import endpoints as auth
+
+router = APIRouter(
+    prefix="/v1/action-items",
+    tags=["action-items"]
+)
+
+class CreateActionItemRequest(BaseModel):
+    description: str
+    memory_id: Optional[str] = None
+    due_date: Optional[datetime] = None
+
+class UpdateActionItemRequest(BaseModel):
+    description: Optional[str] = None
+    status: Optional[ActionItemStatus] = None
+    due_date: Optional[datetime] = None
+
+@router.post("", response_model=ActionItem)
+def create_action_item(
+    request: CreateActionItemRequest,
+    uid: str = Depends(auth.get_current_user_uid)
+):
+    return action_items_db.create_action_item(
+        uid=uid,
+        description=request.description,
+        memory_id=request.memory_id,
+        due_date=request.due_date
+    )
+
+@router.get("", response_model=List[ActionItem])
+def get_action_items(
+    status: Optional[ActionItemStatus] = None,
+    memory_id: Optional[str] = None,
+    start_date: Optional[datetime] = None,
+    end_date: Optional[datetime] = None,
+    limit: int = 100,
+    offset: int = 0,
+    uid: str = Depends(auth.get_current_user_uid)
+):
+    return action_items_db.get_action_items(
+        uid=uid,
+        status=status,
+        memory_id=memory_id,
+        start_date=start_date,
+        end_date=end_date,
+        limit=limit,
+        offset=offset
+    )
+
+@router.patch("/{action_item_id}", response_model=ActionItem)
+def update_action_item(
+    action_item_id: str,
+    request: UpdateActionItemRequest,
+    uid: str = Depends(auth.get_current_user_uid)
+):
+    updates = request.dict(exclude_unset=True)
+    updated_item = action_items_db.update_action_item(uid, action_item_id, updates)
+    if not updated_item:
+        raise HTTPException(status_code=404, detail="Action item not found")
+    return updated_item
+
+@router.delete("/{action_item_id}")
+def delete_action_item(
+    action_item_id: str,
+    uid: str = Depends(auth.get_current_user_uid)
+):
+    success = action_items_db.delete_action_item(uid, action_item_id)
+    if not success:
+        raise HTTPException(status_code=404, detail="Action item not found")
+    return {"message": "Action item deleted successfully"}