mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
9e05370b29
This change makes system prompt text and tool changes part of the transcript rather than silently rewriting its starting conditions. This lets Pi record when instructions changed or tools became available, restore that state after resuming or navigating branches, and preserve cached prompt prefixes where the upstream supports it.
42 lines
2 KiB
TypeScript
42 lines
2 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { streamSimple as streamAnthropic } from "../src/api/anthropic-messages.ts";
|
|
import { streamSimple as streamAzure } from "../src/api/azure-openai-responses.ts";
|
|
import { streamSimple as streamGoogle } from "../src/api/google-generative-ai.ts";
|
|
import { streamSimple as streamMistral } from "../src/api/mistral-conversations.ts";
|
|
import { streamSimple as streamCodex } from "../src/api/openai-codex-responses.ts";
|
|
import { streamSimple as streamOpenAICompletions } from "../src/api/openai-completions.ts";
|
|
import { streamSimple as streamOpenAIResponses } from "../src/api/openai-responses.ts";
|
|
import type { Api, AssistantMessageEventStream, Model } from "../src/types.ts";
|
|
import { normalizeContext } from "../src/utils/transcript.ts";
|
|
|
|
function model<TApi extends Api>(api: TApi): Model<TApi> {
|
|
return {
|
|
id: "test-model",
|
|
name: "Test",
|
|
api,
|
|
provider: "test-provider",
|
|
baseUrl: "https://example.invalid",
|
|
reasoning: false,
|
|
input: ["text"],
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
|
contextWindow: 1_000,
|
|
maxTokens: 100,
|
|
};
|
|
}
|
|
|
|
function expectMissingAuthThrows(create: () => AssistantMessageEventStream): void {
|
|
expect(create).toThrow("No API key for provider: test-provider");
|
|
}
|
|
|
|
describe("direct API authentication", () => {
|
|
it("throws synchronously when auth is missing", () => {
|
|
const context = normalizeContext({ messages: [] });
|
|
expectMissingAuthThrows(() => streamAnthropic(model("anthropic-messages"), context, {}));
|
|
expectMissingAuthThrows(() => streamAzure(model("azure-openai-responses"), context, {}));
|
|
expectMissingAuthThrows(() => streamGoogle(model("google-generative-ai"), context, {}));
|
|
expectMissingAuthThrows(() => streamMistral(model("mistral-conversations"), context, {}));
|
|
expectMissingAuthThrows(() => streamCodex(model("openai-codex-responses"), context, {}));
|
|
expectMissingAuthThrows(() => streamOpenAICompletions(model("openai-completions"), context, {}));
|
|
expectMissingAuthThrows(() => streamOpenAIResponses(model("openai-responses"), context, {}));
|
|
});
|
|
});
|