ngskcloud
/
gcgj-dify-1.7.0
17
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: llm_usage.total_tokens stat

Browse Source
pull/19177/head
IthacaDream 1 year ago
parent 349c3cf7b8
commit 9cf5ec2a5c
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File

1
api/core/agent/cot_agent_runner.py
Unescape Escape View File

@ -87,6 +87,7 @@ class CotAgentRunner(BaseAgentRunner, ABC):
llm_usage = final_llm_usage_dict["usage"] llm_usage = final_llm_usage_dict["usage"]
llm_usage.prompt_tokens += usage.prompt_tokens llm_usage.prompt_tokens += usage.prompt_tokens
llm_usage.completion_tokens += usage.completion_tokens llm_usage.completion_tokens += usage.completion_tokens
llm_usage.total_tokens += usage.total_tokens
llm_usage.prompt_price += usage.prompt_price llm_usage.prompt_price += usage.prompt_price
llm_usage.completion_price += usage.completion_price llm_usage.completion_price += usage.completion_price
llm_usage.total_price += usage.total_price llm_usage.total_price += usage.total_price

1
api/core/agent/fc_agent_runner.py
Unescape Escape View File

@ -72,6 +72,7 @@ class FunctionCallAgentRunner(BaseAgentRunner):
llm_usage = final_llm_usage_dict["usage"] llm_usage = final_llm_usage_dict["usage"]
llm_usage.prompt_tokens += usage.prompt_tokens llm_usage.prompt_tokens += usage.prompt_tokens
llm_usage.completion_tokens += usage.completion_tokens llm_usage.completion_tokens += usage.completion_tokens
llm_usage.total_tokens += usage.total_tokens
llm_usage.prompt_price += usage.prompt_price llm_usage.prompt_price += usage.prompt_price
llm_usage.completion_price += usage.completion_price llm_usage.completion_price += usage.completion_price
llm_usage.total_price += usage.total_price llm_usage.total_price += usage.total_price

Write Preview
Loading…
Cancel
Save