feat: AI Worker 平台 MVP v1.0.0 - 多租户/项目管理/AI Worker/任务编排/HITL审核/成本治理
This commit is contained in:
@@ -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}>"
|
||||||
Reference in New Issue
Block a user