nezumi/astrbot_plugin_qq_group_daily_analysis
1
0
Fork 0
mirror of https://github.com/Nezumi-2711/astrbot_plugin_qq_group_daily_analysis.git synced 2026-09-22 13:38:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix] (LLM Token 获取) 增加对获取不到 token 各个字段的情况的处理，每部分失败后使用默认值 0

Browse Source
This commit is contained in:
SXP-Simon
2025-09-09 18:41:55 +08:00
parent 394b0a5243
commit f2b45a4fdf
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/analysis/llm_analyzer.py
+9 -9
View File
@@ -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'):
src/reports/generators.py
+3 -3
View File
@@ -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
}