feat(avatar): add private vision chat support

2026-08-31 15:10:50 +08:00
parent 094f8cd40f
commit 016bc22c05
22 changed files with 1616 additions and 59 deletions
@@ -79,12 +79,17 @@ def reserve_avatar_tokens(
    model: str,
    messages: list[dict],
    max_output_tokens: int,
+    *,
+    minimum_reserve_tokens: int = 0,
 ) -> TokenReservation:
    user = avatar_owner_user(db, avatar)
    if not user:
        raise InsufficientTokensError("分身尚未关联有效用户，暂时无法使用积分")
    account = get_or_create_account(db, user.id)
-    reserved = estimate_request_tokens(messages, max_output_tokens)
+    reserved = max(
+        estimate_request_tokens(messages, max_output_tokens),
+        max(0, int(minimum_reserve_tokens or 0)),
+    )
    updated = (
        db.query(TokenAccount)
        .filter(TokenAccount.id == account.id, TokenAccount.balance >= reserved)