From fdffcf65050ab606d9eb996ee1564ebf47866f28 Mon Sep 17 00:00:00 2001
From: huangzhuang_3rd <huangzhuang_3rd@tphai.com>
Date: Wed, 12 Aug 2026 13:30:45 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20AI=20Worker=20=E5=B9=B3=E5=8F=B0=20MVP?=
 =?UTF-8?q?=20v1.0.0=20-=20=E5=A4=9A=E7=A7=9F=E6=88=B7/=E9=A1=B9=E7=9B=AE?=
 =?UTF-8?q?=E7=AE=A1=E7=90=86/AI=20Worker/=E4=BB=BB=E5=8A=A1=E7=BC=96?=
 =?UTF-8?q?=E6=8E=92/HITL=E5=AE=A1=E6=A0=B8/=E6=88=90=E6=9C=AC=E6=B2=BB?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/app/models/cost.py | 42 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 backend/app/models/cost.py

diff --git a/backend/app/models/cost.py b/backend/app/models/cost.py
new file mode 100644
index 0000000..7550274
--- /dev/null
+++ b/backend/app/models/cost.py
@@ -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}>"