From f2b45a4fdf88809a1caceed82c588461b93b3472 Mon Sep 17 00:00:00 2001 From: SXP-Simon Date: Tue, 9 Sep 2025 18:41:55 +0800 Subject: [PATCH] =?UTF-8?q?[fix]=20(LLM=20Token=20=E8=8E=B7=E5=8F=96)=20?= =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=AF=B9=E8=8E=B7=E5=8F=96=E4=B8=8D=E5=88=B0?= =?UTF-8?q?=20token=20=E5=90=84=E4=B8=AA=E5=AD=97=E6=AE=B5=E7=9A=84?= =?UTF-8?q?=E6=83=85=E5=86=B5=E7=9A=84=E5=A4=84=E7=90=86=EF=BC=8C=E6=AF=8F?= =?UTF-8?q?=E9=83=A8=E5=88=86=E5=A4=B1=E8=B4=A5=E5=90=8E=E4=BD=BF=E7=94=A8?= =?UTF-8?q?=E9=BB=98=E8=AE=A4=E5=80=BC=200?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/analysis/llm_analyzer.py | 18 +++++++++--------- src/reports/generators.py | 6 +++--- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/src/analysis/llm_analyzer.py b/src/analysis/llm_analyzer.py index 45f92b4..1e5a0d1 100644 --- a/src/analysis/llm_analyzer.py +++ b/src/analysis/llm_analyzer.py @@ -118,9 +118,9 @@ class LLMAnalyzer: token_usage = TokenUsage() if response.raw_completion and hasattr(response.raw_completion, 'usage'): usage = response.raw_completion.usage - token_usage.prompt_tokens = usage.prompt_tokens - token_usage.completion_tokens = usage.completion_tokens - token_usage.total_tokens = usage.total_tokens + token_usage.prompt_tokens = usage.prompt_tokens if usage.prompt_tokens else 0 + token_usage.completion_tokens = usage.completion_tokens if usage.completion_tokens else 0 + token_usage.total_tokens = usage.total_tokens if usage.total_tokens else 0 # 解析响应 if hasattr(response, 'completion_text'): @@ -343,9 +343,9 @@ class LLMAnalyzer: token_usage = TokenUsage() if response.raw_completion and hasattr(response.raw_completion, 'usage'): usage = response.raw_completion.usage - token_usage.prompt_tokens = usage.prompt_tokens - token_usage.completion_tokens = usage.completion_tokens - token_usage.total_tokens = usage.total_tokens + token_usage.prompt_tokens = usage.prompt_tokens if usage.prompt_tokens else 0 + token_usage.completion_tokens = usage.completion_tokens if usage.completion_tokens else 0 + token_usage.total_tokens = usage.total_tokens if usage.total_tokens else 0 # 解析响应 if hasattr(response, 'completion_text'): @@ -444,9 +444,9 @@ class LLMAnalyzer: token_usage = TokenUsage() if response.raw_completion and hasattr(response.raw_completion, 'usage'): usage = response.raw_completion.usage - token_usage.prompt_tokens = usage.prompt_tokens - token_usage.completion_tokens = usage.completion_tokens - token_usage.total_tokens = usage.total_tokens + token_usage.prompt_tokens = usage.prompt_tokens if usage.prompt_tokens else 0 + token_usage.completion_tokens = usage.completion_tokens if usage.completion_tokens else 0 + token_usage.total_tokens = usage.total_tokens if usage.total_tokens else 0 # 解析响应 if hasattr(response, 'completion_text'): diff --git a/src/reports/generators.py b/src/reports/generators.py index 240bba6..a107672 100644 --- a/src/reports/generators.py +++ b/src/reports/generators.py @@ -189,9 +189,9 @@ class ReportGenerator: "titles_html": titles_html, "quotes_html": quotes_html, "hourly_chart_html": hourly_chart_html, - "total_tokens": stats.token_usage.total_tokens, - "prompt_tokens": stats.token_usage.prompt_tokens, - "completion_tokens": stats.token_usage.completion_tokens + "total_tokens": stats.token_usage.total_tokens if stats.token_usage.total_tokens else 0, + "prompt_tokens": stats.token_usage.prompt_tokens if stats.token_usage.prompt_tokens else 0, + "completion_tokens": stats.token_usage.completion_tokens if stats.token_usage.completion_tokens else 0 }