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 }