Agents: widen WS usage aliases

This commit is contained in:
复试资料 2026-03-27 21:02:12 +08:00 committed by Peter Steinberger
parent d166f2648e
commit e955cffd32
3 changed files with 7 additions and 5 deletions

View File

@ -39,9 +39,11 @@ export interface ResponseObject {
}
export interface UsageInfo {
input_tokens: number;
output_tokens: number;
total_tokens: number;
input_tokens?: number;
output_tokens?: number;
total_tokens?: number;
prompt_tokens?: number;
completion_tokens?: number;
}
export type OpenAIResponsesAssistantPhase = "commentary" | "final_answer";

View File

@ -548,7 +548,7 @@ export function buildAssistantMessageFromResponse(
usage: buildUsageWithNoCost({
input: normalizedUsage?.input ?? 0,
output: normalizedUsage?.output ?? 0,
totalTokens: normalizedUsage?.total ?? response.usage?.total_tokens ?? 0,
totalTokens: normalizedUsage?.total ?? 0,
}),
});

View File

@ -862,7 +862,7 @@ describe("buildAssistantMessageFromResponse", () => {
it("maps prompt_tokens and completion_tokens usage aliases", () => {
const response = makeResponseObject("resp_5b", "Hello");
(response as unknown as { usage?: Record<string, number> }).usage = {
response.usage = {
prompt_tokens: 44,
completion_tokens: 11,
total_tokens: 55,