c249763d7e
- Add sessionAgentMap.clear() to _resetForTesting() - Prevents test pollution when tests run in parallel in CI
39 lines
969 B
TypeScript
39 lines
969 B
TypeScript
export const subagentSessions = new Set<string>()
|
|
|
|
let _mainSessionID: string | undefined
|
|
|
|
export function setMainSession(id: string | undefined) {
|
|
_mainSessionID = id
|
|
}
|
|
|
|
export function getMainSessionID(): string | undefined {
|
|
return _mainSessionID
|
|
}
|
|
|
|
/** @internal For testing only */
|
|
export function _resetForTesting(): void {
|
|
_mainSessionID = undefined
|
|
subagentSessions.clear()
|
|
sessionAgentMap.clear()
|
|
}
|
|
|
|
const sessionAgentMap = new Map<string, string>()
|
|
|
|
export function setSessionAgent(sessionID: string, agent: string): void {
|
|
if (!sessionAgentMap.has(sessionID)) {
|
|
sessionAgentMap.set(sessionID, agent)
|
|
}
|
|
}
|
|
|
|
export function updateSessionAgent(sessionID: string, agent: string): void {
|
|
sessionAgentMap.set(sessionID, agent)
|
|
}
|
|
|
|
export function getSessionAgent(sessionID: string): string | undefined {
|
|
return sessionAgentMap.get(sessionID)
|
|
}
|
|
|
|
export function clearSessionAgent(sessionID: string): void {
|
|
sessionAgentMap.delete(sessionID)
|
|
}
|