generated from coulomb/repo-seed
tegwick
58e1bafce9
Introduces end-to-end token consumption tracking so agent work is visible as a cost/effort metric alongside tasks and workplans. - Migration o2j3k4l5m6n7: token_events table with FK indexes on task_id, workstream_id, repo_id, created_at - ORM model, Pydantic schemas (TokenEventCreate, TokenEventRead with computed tokens_total, TokenSummary) - Router: POST /token-events/, GET /token-events/ (7 filters), GET /token-events/summary/ (task|workstream|repo|commit|release scope) - MCP tools: record_token_event, get_token_summary (formatted table) - update_task_status enriched with optional tokens_in/tokens_out passthrough — one call creates status update + token event - Dashboard token-cost.md page: by-repo bar, by-workplan table, by-model bar, top-10 tasks by tokens - ralph-workplan skill updated with token reporting guidance and per-task heuristics for estimating counts - Tests: test_token_events.py + test_token_passthrough.py (182 pass) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
import os
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from api.database import engine
|
|
from api.routers import decisions, extension_points, progress, state, tasks, technical_debt, topics, workstreams, workstream_dependencies
|
|
from api.routers import domains, repos, contributions, sbom, policy, domain_goals, repo_goals, messages, capability_requests, tpsc
|
|
from api.routers import token_events
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
yield
|
|
await engine.dispose()
|
|
|
|
|
|
app = FastAPI(
|
|
title="Custodian State Hub",
|
|
description="Local-first state API for the Custodian agent system.",
|
|
version="0.6.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
_cors_env = os.getenv("CORS_ORIGINS", "http://localhost:3000,http://127.0.0.1:3000")
|
|
_cors_origins = [o.strip() for o in _cors_env.split(",") if o.strip()]
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=_cors_origins,
|
|
allow_methods=["GET", "POST", "PATCH", "DELETE", "PUT"],
|
|
allow_headers=["Content-Type"],
|
|
)
|
|
|
|
app.include_router(domains.router)
|
|
app.include_router(repos.router)
|
|
app.include_router(topics.router)
|
|
app.include_router(workstreams.router)
|
|
app.include_router(workstream_dependencies.router)
|
|
app.include_router(tasks.router)
|
|
app.include_router(decisions.router)
|
|
app.include_router(extension_points.router)
|
|
app.include_router(technical_debt.router)
|
|
app.include_router(progress.router)
|
|
app.include_router(domain_goals.router)
|
|
app.include_router(repo_goals.router)
|
|
app.include_router(contributions.router)
|
|
app.include_router(sbom.router)
|
|
app.include_router(messages.router)
|
|
app.include_router(capability_requests.router)
|
|
app.include_router(tpsc.router)
|
|
app.include_router(token_events.router)
|
|
app.include_router(state.router)
|
|
app.include_router(policy.router)
|
|
|
|
|
|
@app.get("/", include_in_schema=False)
|
|
async def root():
|
|
return {"service": "state-hub", "docs": "/docs"}
|