3928b4872a
Merged via squash. Prepared head SHA: df8f292039e27edec45b8ed2ad65ab0ac7f56194 Co-authored-by: uf-hy <41638541+uf-hy@users.noreply.github.com> Co-authored-by: jalehman <550978+jalehman@users.noreply.github.com> Reviewed-by: @jalehman
38 lines
1.2 KiB
TypeScript
38 lines
1.2 KiB
TypeScript
import { deriveSessionTotalTokens, type NormalizedUsage } from "../../agents/usage.js";
|
|
import { incrementCompactionCount } from "./session-updates.js";
|
|
import { persistSessionUsageUpdate } from "./session-usage.js";
|
|
|
|
type PersistRunSessionUsageParams = Parameters<typeof persistSessionUsageUpdate>[0];
|
|
|
|
type IncrementRunCompactionCountParams = Omit<
|
|
Parameters<typeof incrementCompactionCount>[0],
|
|
"tokensAfter"
|
|
> & {
|
|
amount?: number;
|
|
lastCallUsage?: NormalizedUsage;
|
|
contextTokensUsed?: number;
|
|
};
|
|
|
|
export async function persistRunSessionUsage(params: PersistRunSessionUsageParams): Promise<void> {
|
|
await persistSessionUsageUpdate(params);
|
|
}
|
|
|
|
export async function incrementRunCompactionCount(
|
|
params: IncrementRunCompactionCountParams,
|
|
): Promise<number | undefined> {
|
|
const tokensAfterCompaction = params.lastCallUsage
|
|
? deriveSessionTotalTokens({
|
|
usage: params.lastCallUsage,
|
|
contextTokens: params.contextTokensUsed,
|
|
})
|
|
: undefined;
|
|
return incrementCompactionCount({
|
|
sessionEntry: params.sessionEntry,
|
|
sessionStore: params.sessionStore,
|
|
sessionKey: params.sessionKey,
|
|
storePath: params.storePath,
|
|
amount: params.amount,
|
|
tokensAfter: tokensAfterCompaction,
|
|
});
|
|
}
|