修改token统计失败时的显示

This commit is contained in:
xunbu
2025-09-23 23:34:00 +08:00
parent 75e85e9639
commit b6c316425d
3 changed files with 19 additions and 11 deletions

View File

@@ -498,11 +498,14 @@ class Agent:
# 新增打印token使用统计
token_stats = self.token_counter.get_stats()
self.logger.info(
f"Token使用统计 - 输入: {token_stats['input_tokens'] / 1000:.2f}K(含cached: {token_stats['cached_tokens'] / 1000:.2f}K), "
f"输出: {token_stats['output_tokens'] / 1000:.2f}K(含reasoning: {token_stats['reasoning_tokens'] / 1000:.2f}K), "
f"总计: {token_stats['total_tokens'] / 1000:.2f}K"
)
if token_stats['input_tokens']<0:
self.logger.info("Token统计失败")
else:
self.logger.info(
f"Token使用统计 - 输入: {token_stats['input_tokens'] / 1000:.2f}K(含cached: {token_stats['cached_tokens'] / 1000:.2f}K), "
f"输出: {token_stats['output_tokens'] / 1000:.2f}K(含reasoning: {token_stats['reasoning_tokens'] / 1000:.2f}K), "
f"总计: {token_stats['total_tokens'] / 1000:.2f}K"
)
return results
@@ -732,11 +735,14 @@ class Agent:
# 新增打印token使用统计
token_stats = self.token_counter.get_stats()
self.logger.info(
f"Token使用统计 - 输入: {token_stats['input_tokens'] / 1000:.2f}K(含cached: {token_stats['cached_tokens'] / 1000:.2f}K), "
f"输出: {token_stats['output_tokens'] / 1000:.2f}K(含reasoning: {token_stats['reasoning_tokens'] / 1000:.2f}K), "
f"总计: {token_stats['total_tokens'] / 1000:.2f}K"
)
if token_stats['input_tokens'] < 0:
self.logger.info("Token统计失败")
else:
self.logger.info(
f"Token使用统计 - 输入: {token_stats['input_tokens'] / 1000:.2f}K(含cached: {token_stats['cached_tokens'] / 1000:.2f}K), "
f"输出: {token_stats['output_tokens'] / 1000:.2f}K(含reasoning: {token_stats['reasoning_tokens'] / 1000:.2f}K), "
f"总计: {token_stats['total_tokens'] / 1000:.2f}K"
)
return output_list

File diff suppressed because one or more lines are too long

View File

@@ -4,8 +4,10 @@ v1.4.5版 2025.9.23
特性
- 支持ass字幕翻译
- 缓存先前选择的工作流
- 增加聚光AI平台支持
修复
- 修复xlsx工作流某些文件后缀不能使用的问题
- 修复token计数失败导致中断问题
----------------
v1.4.4版 2025.9.17
修复