import type { AssistantMessage } from "@mariozechner/pi-ai";

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

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

import { makeAttemptResult } from "./run.overflow-compaction.fixture.js";

import {

loadRunOverflowCompactionHarness,

mockedEnsureRuntimePluginsLoaded,

mockedResolveModelAsync,

mockedRunEmbeddedAttempt,

} from "./run.overflow-compaction.harness.js";

import type { EmbeddedRunAttemptResult } from "./run/types.js";

// If the bug exists, it will likely be 350

expect(usage?.total).toBe(200);

});

it("reports the resolved model provider when PI marks the assistant message as pi", async () => {

mockedResolveModelAsync.mockResolvedValueOnce({

model: {

id: "openai/gpt-5.4",

provider: "openrouter",

contextWindow: 200000,

api: "openai-completions",

},

error: null,

authStorage: {

setRuntimeApiKey: vi.fn(),

},

modelRegistry: {},

});

mockedRunEmbeddedAttempt.mockResolvedValueOnce(

makeAttemptResult({

assistantTexts: ["Response 1"],

lastAssistant: makeAssistantMessage({

provider: "pi",

model: "pi",

usage: { input: 100, output: 50, total: 150 } as unknown as AssistantMessage["usage"],

}),

attemptUsage: { input: 100, output: 50, total: 150 },

}),

);

const result = await runEmbeddedPiAgent({

sessionId: "test-session",

sessionKey: "test-key",

sessionFile: "/tmp/session.json",

workspaceDir: "/tmp/workspace",

prompt: "hello",

provider: "openrouter",

model: "openai/gpt-5.4",

timeoutMs: 30000,

runId: "run-provider-attribution",

});

expect(result.meta.agentMeta?.provider).toBe("openrouter");

expect(result.meta.agentMeta?.model).toBe("openai/gpt-5.4");

expect(result.meta.executionTrace?.winnerProvider).toBe("openrouter");

expect(result.meta.executionTrace?.winnerModel).toBe("openai/gpt-5.4");

});

});