fix: persist usage from rpc

This commit is contained in:
Peter Steinberger
2025-12-10 01:37:57 +00:00
parent 49e70746f0
commit c4b02645f5
3 changed files with 26 additions and 12 deletions

View File

@@ -95,10 +95,11 @@ function parsePiJson(raw: string): AgentParseResult {
.trim();
if (isAssistantMessage) {
// Always remember the last assistant message so we keep usage/model meta even when no text.
lastAssistant = msg;
if (msgText && msgText !== lastPushed) {
texts.push(msgText);
lastPushed = msgText;
lastAssistant = msg;
}
} else if (isToolResult && msg.content) {
const toolText = msg.content
@@ -119,15 +120,14 @@ function parsePiJson(raw: string): AgentParseResult {
}
}
const meta: AgentMeta | undefined =
lastAssistant && texts.length
? {
model: lastAssistant.model,
provider: lastAssistant.provider,
stopReason: lastAssistant.stopReason,
usage: lastAssistant.usage,
}
: undefined;
const meta: AgentMeta | undefined = lastAssistant
? {
model: lastAssistant.model,
provider: lastAssistant.provider,
stopReason: lastAssistant.stopReason,
usage: lastAssistant.usage,
}
: undefined;
return {
texts,