增加鉴权系统

2026-07-03 21:31:25 +08:00
parent 8073dd63a7
commit 7e65fadd83
12 changed files with 301 additions and 20 deletions
@@ -46,10 +46,15 @@ Message = Annotated[
 class Session(BaseModel):
    model_config = ConfigDict(extra="forbid")

+    user_id: str | None = None
    session_id: str
    system_prompt: str
    created_at: datetime
    updated_at: datetime
+    api_calls: int = 0
+    prompt_tokens: int = 0
+    completion_tokens: int = 0
+    total_tokens: int = 0
    messages: list[Message] = Field(default_factory=list)

    @model_validator(mode="before")