can1357
1b9d9d0851
Move bundled models, model cache/manager, thinking metadata, effort helpers, provider descriptors/discovery, wire constants, and model identity utilities into the new @oh-my-pi/pi-catalog package. Update pi-ai to keep provider runtime/auth concerns, move catalog provider metadata into CATALOG_PROVIDERS, and migrate coding-agent, agent, stats, docs, and tests to import catalog values from pi-catalog. Split coding-agent model registry helpers into discovery, roles, and models config modules while preserving registry orchestration. BREAKING CHANGE: @oh-my-pi/pi-ai no longer exports catalog subpaths such as /models, /model-cache, /model-manager, /model-thinking, /effort, /provider-models*, discovery helpers, and provider wire constants; use the matching @oh-my-pi/pi-catalog subpaths instead.
113 lines
3.4 KiB
TypeScript
113 lines
3.4 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import { streamSimple } from "@oh-my-pi/pi-ai/stream";
|
|
import type { Context, FetchImpl, Model, ProviderResponseMetadata } from "@oh-my-pi/pi-ai/types";
|
|
import { normalizeProviderResponse, notifyProviderResponse } from "@oh-my-pi/pi-ai/utils/provider-response";
|
|
import { getBundledModel } from "@oh-my-pi/pi-catalog/models";
|
|
|
|
describe("provider response metadata", () => {
|
|
it("normalizes response status, headers, and request id", () => {
|
|
const response = new Response(null, {
|
|
status: 202,
|
|
headers: {
|
|
"X-Request-ID": "req_123",
|
|
"X-RateLimit-Remaining": "42",
|
|
},
|
|
});
|
|
|
|
expect(normalizeProviderResponse(response, "req_123")).toEqual({
|
|
status: 202,
|
|
headers: {
|
|
"x-request-id": "req_123",
|
|
"x-ratelimit-remaining": "42",
|
|
},
|
|
requestId: "req_123",
|
|
});
|
|
});
|
|
|
|
it("invokes the response callback with normalized metadata", async () => {
|
|
const seen: Array<{ response: ProviderResponseMetadata; model: Model | undefined }> = [];
|
|
const model = { provider: "openai", api: "openai-responses", id: "gpt-test" } as Model;
|
|
|
|
await notifyProviderResponse(
|
|
{
|
|
onResponse: (response, responseModel) => {
|
|
seen.push({ response, model: responseModel });
|
|
},
|
|
},
|
|
new Response(null, { status: 204, headers: { "Cache-Control": "no-store" } }),
|
|
model,
|
|
null,
|
|
{ attempt: 1 },
|
|
);
|
|
|
|
expect(seen).toEqual([
|
|
{
|
|
response: {
|
|
status: 204,
|
|
headers: { "cache-control": "no-store" },
|
|
requestId: null,
|
|
metadata: { attempt: 1 },
|
|
},
|
|
model,
|
|
},
|
|
]);
|
|
});
|
|
});
|
|
|
|
function createSseResponse(events: unknown[], headers: Record<string, string> = {}): Response {
|
|
const payload = `${events.map(event => `data: ${typeof event === "string" ? event : JSON.stringify(event)}`).join("\n\n")}\n\n`;
|
|
return new Response(payload, {
|
|
status: 200,
|
|
headers: { "content-type": "text/event-stream", ...headers },
|
|
});
|
|
}
|
|
|
|
describe("streamSimple onResponse propagation", () => {
|
|
it("invokes onResponse for the default openai-completions path through streamSimple", async () => {
|
|
const model: Model<"openai-completions"> = {
|
|
...(getBundledModel("openai", "gpt-4o-mini") as Model<"openai-completions">),
|
|
api: "openai-completions",
|
|
};
|
|
|
|
const fetchMock: FetchImpl = Object.assign(
|
|
async (_input: string | URL | Request, _init?: RequestInit): Promise<Response> =>
|
|
createSseResponse(
|
|
[
|
|
{
|
|
id: "chatcmpl-onresponse",
|
|
object: "chat.completion.chunk",
|
|
created: 0,
|
|
model: model.id,
|
|
choices: [{ index: 0, delta: { content: "ok" } }],
|
|
},
|
|
{
|
|
id: "chatcmpl-onresponse",
|
|
object: "chat.completion.chunk",
|
|
created: 0,
|
|
model: model.id,
|
|
choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
|
|
},
|
|
"[DONE]",
|
|
],
|
|
{ "x-request-id": "req_stream_simple" },
|
|
),
|
|
{ preconnect: fetch.preconnect },
|
|
);
|
|
|
|
const context: Context = { messages: [{ role: "user", content: "hello", timestamp: Date.now() }] };
|
|
const seen: ProviderResponseMetadata[] = [];
|
|
const result = await streamSimple(model, context, {
|
|
apiKey: "test-key",
|
|
fetch: fetchMock,
|
|
onResponse: response => {
|
|
seen.push(response);
|
|
},
|
|
}).result();
|
|
|
|
expect(result.stopReason).toBe("stop");
|
|
expect(seen).toHaveLength(1);
|
|
expect(seen[0]?.status).toBe(200);
|
|
expect(seen[0]?.headers["x-request-id"]).toBe("req_stream_simple");
|
|
});
|
|
});
|