From b34cfb676d4cb44a51a22e6d522ad13b1601e677 Mon Sep 17 00:00:00 2001
From: fawney19 <elky0401@gmail.com>
Date: Mon, 5 Jan 2026 09:12:03 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20streaming=20=E7=8A=B6=E6=80=81=E6=9B=B4?=
 =?UTF-8?q?=E6=96=B0=E6=97=B6=E4=BC=A0=E9=80=92=20provider=20=E7=9B=B8?=
 =?UTF-8?q?=E5=85=B3=20ID=20=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

在 update_usage_status 方法中增加 provider_id、provider_endpoint_id
和 provider_api_key_id 参数，确保流式请求进入 streaming 状态时
能正确记录这些字段。
---
 src/services/usage/service.py | 12 ++++++++++++
 src/services/usage/stream.py  |  6 ++++++
 2 files changed, 18 insertions(+)

diff --git a/src/services/usage/service.py b/src/services/usage/service.py
index bdb2628..190fde8 100644
--- a/src/services/usage/service.py
+++ b/src/services/usage/service.py
@@ -1476,6 +1476,9 @@ class UsageService:
         provider: Optional[str] = None,
         target_model: Optional[str] = None,
         first_byte_time_ms: Optional[int] = None,
+        provider_id: Optional[str] = None,
+        provider_endpoint_id: Optional[str] = None,
+        provider_api_key_id: Optional[str] = None,
     ) -> Optional[Usage]:
         """
         快速更新使用记录状态
@@ -1488,6 +1491,9 @@ class UsageService:
             provider: 提供商名称（可选，streaming 状态时更新）
             target_model: 映射后的目标模型名（可选）
             first_byte_time_ms: 首字时间/TTFB（可选，streaming 状态时更新）
+            provider_id: Provider ID（可选，streaming 状态时更新）
+            provider_endpoint_id: Endpoint ID（可选，streaming 状态时更新）
+            provider_api_key_id: Provider API Key ID（可选，streaming 状态时更新）
 
         Returns:
             更新后的 Usage 记录，如果未找到则返回 None
@@ -1513,6 +1519,12 @@ class UsageService:
             usage.target_model = target_model
         if first_byte_time_ms is not None:
             usage.first_byte_time_ms = first_byte_time_ms
+        if provider_id is not None:
+            usage.provider_id = provider_id
+        if provider_endpoint_id is not None:
+            usage.provider_endpoint_id = provider_endpoint_id
+        if provider_api_key_id is not None:
+            usage.provider_api_key_id = provider_api_key_id
 
         db.commit()
 
diff --git a/src/services/usage/stream.py b/src/services/usage/stream.py
index 8fef64f..ac6761e 100644
--- a/src/services/usage/stream.py
+++ b/src/services/usage/stream.py
@@ -484,6 +484,9 @@ class StreamUsageTracker:
                             status="streaming",
                             provider=self.provider,
                             first_byte_time_ms=first_byte_time_ms,
+                            provider_id=self.provider_id,
+                            provider_endpoint_id=self.provider_endpoint_id,
+                            provider_api_key_id=self.provider_api_key_id,
                         )
                     except Exception as e:
                         logger.warning(f"更新使用记录状态为 streaming 失败: {e}")
@@ -942,6 +945,9 @@ class EnhancedStreamUsageTracker(StreamUsageTracker):
                             status="streaming",
                             provider=self.provider,
                             first_byte_time_ms=first_byte_time_ms,
+                            provider_id=self.provider_id,
+                            provider_endpoint_id=self.provider_endpoint_id,
+                            provider_api_key_id=self.provider_api_key_id,
                         )
                     except Exception as e:
                         logger.warning(f"更新使用记录状态为 streaming 失败: {e}")