/**

* Manages active embedded-agent run handles, queues, aborts, and waiters.

*/

import { normalizeOptionalString } from "@openclaw/normalization-core/string-coerce";

import {

abortActiveReplyRuns,

abortReplyRunBySessionId,

ABANDONED_EMBEDDED_RUNS_BY_SESSION_ID,

ABANDONED_EMBEDDED_RUN_SESSION_IDS_BY_FILE,

ABANDONED_EMBEDDED_RUN_SESSION_IDS_BY_KEY,

EMBEDDED_RUN_MODEL_SWITCH_REQUESTS,

EMBEDDED_RUN_WAITERS,

getActiveEmbeddedRunCount,

type ActiveEmbeddedRunSnapshot,

type AbandonedEmbeddedRun,

type EmbeddedAgentQueueHandle,

type EmbeddedAgentQueueMessageOptions,

type EmbeddedRunModelSwitchRequest,

type EmbeddedRunWaiter,

} from "./run-state.js";

import { resolveEmbeddedSessionFileKey } from "./session-file-key.js";

type ActiveEmbeddedRunSnapshot,

type EmbeddedAgentQueueHandle,

type EmbeddedAgentQueueMessageOptions,

type EmbeddedRunModelSwitchRequest,

} from "./run-state.js";

export type EmbeddedAgentQueueFailureReason =

return ACTIVE_EMBEDDED_RUN_SNAPSHOTS.get(sessionId);

}

export function requestEmbeddedRunModelSwitch(

sessionId: string,

request: EmbeddedRunModelSwitchRequest,

): boolean {

const normalizedSessionId = sessionId.trim();

const provider = request.provider.trim();

const model = request.model.trim();

if (!normalizedSessionId || !provider || !model) {

return false;

}

EMBEDDED_RUN_MODEL_SWITCH_REQUESTS.set(normalizedSessionId, {

provider,

model,

authProfileId: normalizeOptionalString(request.authProfileId),

authProfileIdSource: normalizeOptionalString(request.authProfileId)

? request.authProfileIdSource

: undefined,

});

diag.debug(

`model switch requested: sessionId=${normalizedSessionId} provider=${provider} model=${model}`,

);

return true;

}

export function consumeEmbeddedRunModelSwitch(

sessionId: string,

): EmbeddedRunModelSwitchRequest | undefined {

const normalizedSessionId = sessionId.trim();

if (!normalizedSessionId) {

return undefined;

}

const request = EMBEDDED_RUN_MODEL_SWITCH_REQUESTS.get(normalizedSessionId);

if (request) {

EMBEDDED_RUN_MODEL_SWITCH_REQUESTS.delete(normalizedSessionId);

}

return request;

}

/**

* Wait for active embedded runs to drain.

*

if (activeHandle === handle) {

ACTIVE_EMBEDDED_RUNS.delete(sessionId);

ACTIVE_EMBEDDED_RUN_SNAPSHOTS.delete(sessionId);

EMBEDDED_RUN_MODEL_SWITCH_REQUESTS.delete(sessionId);

clearActiveRunSessionKeys(sessionId, sessionKey);

clearActiveRunSessionFiles(sessionId, sessionFile);

logSessionStateChange({

if (ACTIVE_EMBEDDED_RUNS.has(sessionId)) {

ACTIVE_EMBEDDED_RUNS.delete(sessionId);

ACTIVE_EMBEDDED_RUN_SNAPSHOTS.delete(sessionId);

EMBEDDED_RUN_MODEL_SWITCH_REQUESTS.delete(sessionId);

clearActiveRunSessionKeys(sessionId, sessionKey);

clearActiveRunSessionFiles(sessionId);

logSessionStateChange({ sessionId, sessionKey, state: "idle", reason });

ABANDONED_EMBEDDED_RUNS_BY_SESSION_ID.clear();

ABANDONED_EMBEDDED_RUN_SESSION_IDS_BY_KEY.clear();

ABANDONED_EMBEDDED_RUN_SESSION_IDS_BY_FILE.clear();

EMBEDDED_RUN_MODEL_SWITCH_REQUESTS.clear();

},

};

export { testing as __testing };