Skip to content

Commit 8334dec

Browse files
committed
refactor: inline compression timing state
1 parent 4671b5b commit 8334dec

1 file changed

Lines changed: 5 additions & 9 deletions

File tree

lib/state/state.ts

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import type { CompressionTimingState, SessionState, ToolParameterEntry, WithParts } from "./types"
1+
import type { SessionState, ToolParameterEntry, WithParts } from "./types"
22
import type { Logger } from "../logger"
33
import { attachCompressionDuration } from "../compress/state"
44
import { loadSessionState, saveSessionState } from "./persistence"
@@ -14,13 +14,6 @@ import {
1414
} from "./utils"
1515
import { getLastUserMessage } from "../messages/query"
1616

17-
function createCompressionTimingState(): CompressionTimingState {
18-
return {
19-
startsByCallId: new Map(),
20-
pendingBySessionId: new Map(),
21-
}
22-
}
23-
2417
export const checkSession = async (
2518
client: any,
2619
state: SessionState,
@@ -89,7 +82,10 @@ export function createSessionState(): SessionState {
8982
pruneTokenCounter: 0,
9083
totalPruneTokens: 0,
9184
},
92-
compressionTiming: createCompressionTimingState(),
85+
compressionTiming: {
86+
startsByCallId: new Map(),
87+
pendingBySessionId: new Map(),
88+
},
9389
toolParameters: new Map<string, ToolParameterEntry>(),
9490
subAgentResultCache: new Map<string, string>(),
9591
toolIdList: [],

0 commit comments

Comments
 (0)