import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";

import { LiveSessionModelSwitchError } from "../../agents/live-model-switch-error.js";

import type { SessionEntry } from "../../config/sessions.js";

import type { ModelDefinitionConfig } from "../../config/types.models.js";

import { CommandLaneClearedError, GatewayDrainingError } from "../../process/command-queue.js";

import type { TemplateContext } from "../templating.js";

import { SILENT_REPLY_TOKEN } from "../tokens.js";

const GENERIC_RUN_FAILURE_TEXT =

"⚠️ Something went wrong while processing your request. Please try again, or use /new to start a fresh session.";

function makeTestModel(id: string, contextTokens: number): ModelDefinitionConfig {

return {

id,

name: id,

reasoning: false,

input: ["text"],

cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },

contextWindow: contextTokens,

contextTokens,

maxTokens: 4096,

};

}

vi.mock("../../agents/pi-embedded.js", () => ({

runEmbeddedPiAgent: (params: unknown) => state.runEmbeddedPiAgentMock(params),

}));

models: {

providers: {

openrouter: {

models: [{ id: "qwen3.6-plus", contextTokens: 1_000_000 }],

baseUrl: "https://openrouter.test",

models: [makeTestModel("qwen3.6-plus", 1_000_000)],

},

ollama: {

models: [{ id: "qwen3.5-9b-32k:latest", contextTokens: 32_768 }],

baseUrl: "http://ollama.test",

models: [makeTestModel("qwen3.5-9b-32k:latest", 32_768)],

},

},

},