feat: AI Worker 平台 MVP v1.0.0 - 多租户/项目管理/AI Worker/任务编排/HITL审核/成本治理

This commit is contained in:
2026-08-12 13:30:45 +08:00
parent a6bd7a6689
commit fdffcf6505
+42
View File
@@ -0,0 +1,42 @@
"""Cost tracking model - per-call usage logs."""
from sqlalchemy import Column, String, Integer, Text, ForeignKey, Float
from app.models.base import Base, TimestampMixin, TenantMixin
class CostLog(Base, TimestampMixin, TenantMixin):
"""Records every LLM call's token usage and cost.
Enables multi-dimensional cost analysis:
tenant x project x task x worker x model
"""
__tablename__ = "cost_logs"
id = Column(Integer, primary_key=True, autoincrement=True)
project_id = Column(Integer, ForeignKey("projects.id"), nullable=True, index=True)
task_id = Column(Integer, ForeignKey("tasks.id"), nullable=True, index=True)
worker_id = Column(Integer, ForeignKey("ai_workers.id"), nullable=True, index=True)
# Model info
provider = Column(String(100), nullable=False)
model_name = Column(String(200), nullable=False)
# Token counts
prompt_tokens = Column(Integer, default=0)
completion_tokens = Column(Integer, default=0)
total_tokens = Column(Integer, default=0)
# Cost in cents
cost_cents = Column(Integer, default=0)
# Duration
duration_ms = Column(Integer, default=0)
# Status
status = Column(String(50), default="success") # success, error, timeout
# Error message if any
error_message = Column(Text, nullable=True)
def __repr__(self):
return f"<CostLog {self.provider}/{self.model_name} tokens={self.total_tokens}>"