generated from coulomb/repo-seed
tegwick
0949d4c0d8
Replace the ad-hoc coordination-domain spine with the Repo Classification Standard: 14 market domains, classification columns on managed_repos, and workplans anchored by repo_id (topic_id optional). - Add Alembic migration d8e9f0a1b2c3 with data backfill and workstream→workplan rename - Add api/classification.py validation and register-from-classification tooling - Expose workplan-first REST/MCP surface with legacy workstream aliases - Add C-24 consistency rule and legacy domain frontmatter mapping - Update dashboard repos page with category/capability/stake filters - Update orientation docs; mark STATE-WP-0065 finished
80 lines
3.5 KiB
Python
80 lines
3.5 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
from typing import Any
|
|
|
|
from sqlalchemy import DateTime, Float, ForeignKey, Integer, Text, UniqueConstraint, func
|
|
from sqlalchemy.dialects.postgresql import JSONB, UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from api.models.base import Base, new_uuid
|
|
|
|
|
|
class TokenEvent(Base):
|
|
__tablename__ = "token_events"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"measurement_kind",
|
|
"source_provider",
|
|
"source_id",
|
|
name="uq_token_events_source_identity",
|
|
),
|
|
)
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), primary_key=True, default=new_uuid
|
|
)
|
|
task_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("tasks.id", ondelete="SET NULL"), nullable=True, index=True
|
|
)
|
|
workplan_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("workplans.id", ondelete="SET NULL"), nullable=True, index=True
|
|
)
|
|
repo_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("managed_repos.id", ondelete="SET NULL"), nullable=True, index=True
|
|
)
|
|
session_id: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
model: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
tokens_in: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
tokens_out: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
agent: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
ref_type: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
ref_id: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
note: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
measurement_kind: Mapped[str] = mapped_column(
|
|
Text, nullable=False, default="estimated", server_default="estimated", index=True
|
|
)
|
|
source_provider: Mapped[str] = mapped_column(
|
|
Text, nullable=False, default="manual", server_default="manual", index=True
|
|
)
|
|
source_id: Mapped[str | None] = mapped_column(Text, nullable=True, index=True)
|
|
source_path: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
source_created_at: Mapped[datetime | None] = mapped_column(
|
|
DateTime(timezone=True), nullable=True, index=True
|
|
)
|
|
ingested_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now(), nullable=False, index=True
|
|
)
|
|
parser_version: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
confidence: Mapped[float] = mapped_column(
|
|
Float, nullable=False, default=0.35, server_default="0.35"
|
|
)
|
|
cached_input_tokens: Mapped[int] = mapped_column(
|
|
Integer, nullable=False, default=0, server_default="0"
|
|
)
|
|
reasoning_output_tokens: Mapped[int] = mapped_column(
|
|
Integer, nullable=False, default=0, server_default="0"
|
|
)
|
|
raw_total_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
cost_estimated_usd: Mapped[float | None] = mapped_column(Float, nullable=True)
|
|
raw_metadata: Mapped[dict[str, Any]] = mapped_column(
|
|
JSONB, nullable=False, default=dict, server_default="{}"
|
|
)
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now(), nullable=False, index=True
|
|
)
|
|
|
|
task: Mapped["Task | None"] = relationship("Task", lazy="selectin") # noqa: F821
|
|
workplan: Mapped["Workplan | None"] = relationship("Workplan", lazy="selectin") # noqa: F821
|
|
repo: Mapped["ManagedRepo | None"] = relationship("ManagedRepo", lazy="selectin") # noqa: F821
|