feat(coding-agent): replace model registry with model runtime

Move provider auth and OAuth flows onto pi-ai Models, compose models.json and extension overlays through ModelRuntime, and retain ModelRegistry as an extension compatibility facade.
2026-07-14 17:48:45 +02:00
parent 6731a0ba9e
commit 9993c96907
133 changed files with 5103 additions and 4340 deletions
@@ -0,0 +1,65 @@
+import { describe, expect, it } from "vitest";
+import { cloudflareStreams } from "../src/providers/cloudflare-stream.ts";
+import type { Api, Context, Model } from "../src/types.ts";
+import { AssistantMessageEventStream } from "../src/utils/event-stream.ts";
+
+const model: Model<Api> = {
+	id: "model",
+	name: "model",
+	api: "openai-completions",
+	provider: "cloudflare-ai-gateway",
+	baseUrl: "https://gateway.ai.cloudflare.com/v1/{CLOUDFLARE_ACCOUNT_ID}/{CLOUDFLARE_GATEWAY_ID}/openai",
+	reasoning: false,
+	input: ["text"],
+	cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+	contextWindow: 1000,
+	maxTokens: 100,
+};
+
+const context: Context = { messages: [] };
+
+describe("Cloudflare provider streams", () => {
+	it("materializes the model endpoint before dispatch", () => {
+		const captured: string[] = [];
+		const streams = cloudflareStreams({
+			stream: (requestModel) => {
+				captured.push(requestModel.baseUrl);
+				return new AssistantMessageEventStream();
+			},
+			streamSimple: (requestModel) => {
+				captured.push(requestModel.baseUrl);
+				return new AssistantMessageEventStream();
+			},
+		});
+		const env = {
+			CLOUDFLARE_ACCOUNT_ID: "account",
+			CLOUDFLARE_GATEWAY_ID: "gateway",
+		};
+
+		streams.stream(model, context, { env });
+		streams.streamSimple(model, context, { env });
+
+		expect(captured).toEqual([
+			"https://gateway.ai.cloudflare.com/v1/account/gateway/openai",
+			"https://gateway.ai.cloudflare.com/v1/account/gateway/openai",
+		]);
+	});
+
+	it("keeps placeholders when the provider env does not resolve them", () => {
+		let captured: string | undefined;
+		const streams = cloudflareStreams({
+			stream: (requestModel) => {
+				captured = requestModel.baseUrl;
+				return new AssistantMessageEventStream();
+			},
+			streamSimple: (requestModel) => {
+				captured = requestModel.baseUrl;
+				return new AssistantMessageEventStream();
+			},
+		});
+
+		streams.streamSimple(model, context, {});
+
+		expect(captured).toBe(model.baseUrl);
+	});
+});