mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-04-25 00:51:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(acp): populate usage from top-level result fields

Browse source
This commit is contained in:
Yuhan Lei 2026-04-10 16:43:35 +08:00 • committed by Teknium
parent eaa21a8275
commit f92298fe95
2 changed files with 39 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
acp_adapter/server.py
View file

@ -460,6 +460,14 @@ class HermesACPAgent(acp.Agent):
thought_tokens=usage_data.get("reasoning_tokens"),
cached_read_tokens=usage_data.get("cached_tokens"),
)
elif any(result.get(key) is not None for key in ("prompt_tokens", "completion_tokens", "total_tokens")):
usage = Usage(
input_tokens=result.get("prompt_tokens", 0),
output_tokens=result.get("completion_tokens", 0),
total_tokens=result.get("total_tokens", 0),
thought_tokens=result.get("reasoning_tokens"),
cached_read_tokens=result.get("cache_read_tokens"),
)
stop_reason = "cancelled" if state.cancel_event and state.cancel_event.is_set() else "end_turn"
return PromptResponse(stop_reason=stop_reason, usage=usage)