feature: 新增API调用日志统计，首字，总耗时，token消耗等

2026-03-18 12:35:13 +08:00
parent 4e3fb6766e
commit a6e6df5073
9 changed files with 491 additions and 69 deletions
@@ -90,7 +90,12 @@ class AnthropicProvider(BaseAIProvider):
                            final_messages, model, temperature, max_tokens, system_prompt, tools, user_id
                        ):
                            yield final_chunk
+                    if chunk.get("finish_reason"):
+                        yield {"finish_reason": chunk.get("finish_reason"), "done": True}
                    break
+
+                if chunk.get("usage"):
+                    yield {"usage": chunk.get("usage")}
                
                # 输出文本内容
                if chunk.get("content"):
@@ -106,8 +111,11 @@ class AnthropicProvider(BaseAIProvider):
            max_tokens=max_tokens,
            system_prompt=system_prompt,
        ):
-            # 确保只 yield 字符串内容，避免 yield 字典导致类型错误
            if isinstance(chunk, dict):
+                if chunk.get("usage"):
+                    yield {"usage": chunk.get("usage")}
+                if chunk.get("finish_reason"):
+                    yield {"finish_reason": chunk.get("finish_reason")}
                if chunk.get("content"):
                    yield chunk["content"]
            else:
@@ -155,7 +163,12 @@ class AnthropicProvider(BaseAIProvider):
                        messages, model, temperature, max_tokens, system_prompt, tools, user_id
                    ):
                        yield final_chunk
+                if chunk.get("finish_reason"):
+                    yield {"finish_reason": chunk.get("finish_reason"), "done": True}
                break
+
+            if chunk.get("usage"):
+                yield {"usage": chunk.get("usage")}
            
            if chunk.get("content"):
                yield chunk["content"]