Lesson 02 — Reliability floor

"""Pydantic schemas — reliability floor for the workshop."""

from __future__ import annotations

from datetime import datetime, timezone
from typing import Any, Literal

from pydantic import BaseModel, Field, field_validator

Actor = Literal["user", "agent", "system"]
AllowedAction = Literal["read", "write", "network"]


class BusinessQuestion(BaseModel):
    question: str = Field(min_length=1)
    tenant: str = Field(min_length=1)
    deadline: datetime

    @field_validator("deadline")
    @classmethod
    def _deadline_must_be_future(cls, v: datetime) -> datetime:
        now = datetime.now(tz=v.tzinfo or timezone.utc)
        if v <= now:
            raise ValueError("deadline must be in the future")
        return v


class SkillManifest(BaseModel):
    name: str
    description: str = Field(min_length=20)
    when_to_use: str = Field(min_length=1)
    tools: list[str] = Field(min_length=1)
    allowed_actions: list[AllowedAction] = Field(min_length=1)

    @field_validator("name")
    @classmethod
    def _snake_case(cls, v: str) -> str:
        if not v.replace("_", "").isalnum() or not v.islower():
            raise ValueError("name must be snake_case (lowercase + underscores)")
        return v


class Receipt(BaseModel):
    ts: datetime
    actor: Actor
    event: str
    payload: dict[str, Any]

"""Reliability floor tests."""

from __future__ import annotations

from datetime import datetime, timedelta, timezone

import pytest
from pydantic import ValidationError

from src.schemas import BusinessQuestion, Receipt, SkillManifest


def _future() -> datetime:
    return datetime.now(tz=timezone.utc) + timedelta(days=1)


def test_business_question_rejects_empty_question() -> None:
    with pytest.raises(ValidationError, match="at least 1 character"):
        BusinessQuestion(question="", tenant="acme", deadline=_future())


def test_business_question_rejects_past_deadline() -> None:
    past = datetime.now(tz=timezone.utc) - timedelta(days=1)
    with pytest.raises(ValidationError, match="future"):
        BusinessQuestion(question="q?", tenant="acme", deadline=past)


def test_skill_manifest_rejects_non_snake_case_name() -> None:
    with pytest.raises(ValidationError, match="snake_case"):
        SkillManifest(
            name="QueryCognee",
            description="Answer business questions from the graph.",
            when_to_use="user asks a data question",
            tools=["cognee_search"],
            allowed_actions=["read"],
        )


def test_receipt_accepts_arbitrary_payload() -> None:
    r = Receipt(
        ts=datetime.now(tz=timezone.utc),
        actor="agent",
        event="skill_created",
        payload={"skill": "query_cognee", "path": ".skills/query_cognee/SKILL.md"},
    )
    assert r.event == "skill_created"