state-hub/api/schemas/capability_request.py

import uuid
from datetime import datetime

from pydantic import BaseModel, ConfigDict


# ---------------------------------------------------------------------------
# Capability Catalog schemas
# ---------------------------------------------------------------------------

class CatalogCreate(BaseModel):
    domain: str  # slug, resolved to domain_id in router
    capability_type: str
    title: str
    description: str | None = None
    keywords: list[str] = []


class CatalogRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    id: uuid.UUID
    domain_slug: str
    capability_type: str
    title: str
    description: str | None = None
    keywords: list[str] = []
    status: str
    created_at: datetime
    updated_at: datetime


# ---------------------------------------------------------------------------
# Capability Request schemas
# ---------------------------------------------------------------------------

class CapabilityRequestCreate(BaseModel):
    title: str
    description: str | None = None
    capability_type: str
    priority: str = "medium"
    requesting_domain: str  # slug, resolved to domain_id in router
    requesting_agent: str
    requesting_workstream_id: uuid.UUID | None = None
    blocking_task_id: uuid.UUID | None = None


class CapabilityRequestAccept(BaseModel):
    fulfilling_agent: str
    fulfilling_workstream_id: uuid.UUID | None = None


class CapabilityRequestStatusPatch(BaseModel):
    status: str  # in_progress | ready_for_review | completed | rejected | withdrawn
    note: str | None = None


class CapabilityRequestPatch(BaseModel):
    catalog_entry_id: uuid.UUID | None = None
    priority: str | None = None
    blocking_task_id: uuid.UUID | None = None
    fulfilling_workstream_id: uuid.UUID | None = None


class CapabilityRequestRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    id: uuid.UUID
    title: str
    description: str | None = None
    capability_type: str
    priority: str
    status: str
    requesting_domain_slug: str
    requesting_agent: str
    requesting_workstream_id: uuid.UUID | None = None
    fulfilling_domain_slug: str | None = None
    fulfilling_agent: str | None = None
    fulfilling_workstream_id: uuid.UUID | None = None
    blocking_task_id: uuid.UUID | None = None
    catalog_entry_id: uuid.UUID | None = None
    resolution_note: str | None = None
    routing_note: str | None = None
    accepted_at: datetime | None = None
    completed_at: datetime | None = None
    created_at: datetime
    updated_at: datetime