修改token统计失败时的显示
This commit is contained in:
@@ -498,11 +498,14 @@ class Agent:
|
|||||||
|
|
||||||
# 新增:打印token使用统计
|
# 新增:打印token使用统计
|
||||||
token_stats = self.token_counter.get_stats()
|
token_stats = self.token_counter.get_stats()
|
||||||
self.logger.info(
|
if token_stats['input_tokens']<0:
|
||||||
f"Token使用统计 - 输入: {token_stats['input_tokens'] / 1000:.2f}K(含cached: {token_stats['cached_tokens'] / 1000:.2f}K), "
|
self.logger.info("Token统计失败")
|
||||||
f"输出: {token_stats['output_tokens'] / 1000:.2f}K(含reasoning: {token_stats['reasoning_tokens'] / 1000:.2f}K), "
|
else:
|
||||||
f"总计: {token_stats['total_tokens'] / 1000:.2f}K"
|
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
|
return results
|
||||||
|
|
||||||
@@ -732,11 +735,14 @@ class Agent:
|
|||||||
|
|
||||||
# 新增:打印token使用统计
|
# 新增:打印token使用统计
|
||||||
token_stats = self.token_counter.get_stats()
|
token_stats = self.token_counter.get_stats()
|
||||||
self.logger.info(
|
if token_stats['input_tokens'] < 0:
|
||||||
f"Token使用统计 - 输入: {token_stats['input_tokens'] / 1000:.2f}K(含cached: {token_stats['cached_tokens'] / 1000:.2f}K), "
|
self.logger.info("Token统计失败")
|
||||||
f"输出: {token_stats['output_tokens'] / 1000:.2f}K(含reasoning: {token_stats['reasoning_tokens'] / 1000:.2f}K), "
|
else:
|
||||||
f"总计: {token_stats['total_tokens'] / 1000:.2f}K"
|
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
|
return output_list
|
||||||
|
|
||||||
|
|||||||
File diff suppressed because one or more lines are too long
Reference in New Issue
Block a user