Ethan GreenfeldandGitHub
d2308f4835
Pi and OMP share only the JSONL child-process transport while retaining provider-owned launch, RPC, runtime, session, history, and permission behavior. Removes captured fixtures in favor of typed harnesses and real-provider coverage. Closes #2006 Closes #2060
25 lines
947 B
TypeScript
25 lines
947 B
TypeScript
import type { AgentUsage } from "../../agent-sdk-types.js";
|
|
import type { OmpSessionState, OmpSessionStats } from "./rpc-types.js";
|
|
|
|
export function mapOmpUsage(input: {
|
|
stats: OmpSessionStats;
|
|
state: OmpSessionState;
|
|
baseUsage: AgentUsage | undefined;
|
|
}): AgentUsage | undefined {
|
|
const contextWindowUsedTokens = finiteNumber(input.state.contextUsage?.tokens);
|
|
const contextWindowMaxTokens = finiteNumber(input.state.contextUsage?.contextWindow);
|
|
if (contextWindowUsedTokens === undefined && contextWindowMaxTokens === undefined) {
|
|
return input.baseUsage;
|
|
}
|
|
|
|
return {
|
|
...input.baseUsage,
|
|
...(contextWindowMaxTokens !== undefined ? { contextWindowMaxTokens } : {}),
|
|
...(contextWindowUsedTokens !== undefined ? { contextWindowUsedTokens } : {}),
|
|
};
|
|
}
|
|
|
|
function finiteNumber(value: number | null | undefined): number | undefined {
|
|
return typeof value === "number" && Number.isFinite(value) ? value : undefined;
|
|
}
|