1.0.0初始化源代码

backend/app/schemas/token_usage.py

@@ -0,0 +1,54 @@
+"""
+Token 使用相关 Schema
+"""
+from pydantic import BaseModel, Field
+from typing import Optional, List
+from datetime import date, datetime
+
+
+class TokenUsageBase(BaseModel):
+    """Token 使用基础 Schema"""
+    tokens_used: int = Field(..., description="使用的 Token 数量")
+    ai_model: str = Field(..., description="使用的 AI 模型")
+    action_type: Optional[str] = Field(None, description="操作类型")
+
+
+class TokenUsageResponse(TokenUsageBase):
+    """Token 使用响应"""
+    id: int
+    virtual_user_id: Optional[int]
+    interaction_id: Optional[int]
+    tokens_prompt: int
+    tokens_completion: int
+    usage_date: date
+    created_at: datetime
+    
+    class Config:
+        from_attributes = True
+
+
+class TokenUsageStats(BaseModel):
+    """Token 使用统计"""
+    today_tokens: int = Field(0, description="今日 Token 数")
+    yesterday_tokens: int = Field(0, description="昨日 Token 数")
+    month_tokens: int = Field(0, description="当月 Token 数")
+    remaining_tokens: int = Field(0, description="剩余 Token 数")
+    total_limit: int = Field(0, description="总限额")
+
+
+class DailyTokenUsage(BaseModel):
+    """每日 Token 使用"""
+    date: str
+    tokens: int
+
+
+class MonthlyTokenUsage(BaseModel):
+    """每月 Token 使用"""
+    month: str
+    tokens: int
+
+
+class TokenUsageChartResponse(BaseModel):
+    """Token 使用图表响应"""
+    daily_usages: List[DailyTokenUsage]
+    monthly_usages: List[MonthlyTokenUsage]