@browserbasehq/orca 3.1.0-patch.4 → 3.2.0-middleware.1

package/dist/esm/tests/unit/llm-middleware.test.js

@@ -0,0 +1,460 @@
+import { describe, expect, it, vi } from "vitest";
+import { getAISDKLanguageModel, LLMProvider, } from "../../lib/v3/llm/LLMProvider.js";
+import { resolveModelConfiguration } from "../../lib/v3/v3.js";
+/**
+ * Creates a recording middleware that captures every doGenerate / doStream
+ * invocation along with the model identity and returned usage data.
+ */
+function createRecordingMiddleware() {
+    const calls = [];
+    const middleware = {
+        wrapGenerate: async ({ doGenerate, model }) => {
+            const result = await doGenerate();
+            calls.push({
+                type: "generate",
+                modelId: model.modelId,
+                provider: model.provider,
+                usage: {
+                    inputTokens: result.usage.inputTokens ?? undefined,
+                    outputTokens: result.usage.outputTokens ?? undefined,
+                },
+            });
+            return result;
+        },
+        wrapStream: async ({ doStream, model }) => {
+            const result = await doStream();
+            calls.push({
+                type: "stream",
+                modelId: model.modelId,
+                provider: model.provider,
+            });
+            return result;
+        },
+    };
+    return { middleware, calls };
+}
+/**
+ * Creates a minimal mock LanguageModelV2 that returns canned results
+ * without hitting any real provider. Useful for testing the wrapping
+ * mechanics in isolation.
+ */
+function createMockLanguageModel(modelId, provider) {
+    return {
+        specificationVersion: "v2",
+        provider,
+        modelId,
+        defaultObjectGenerationMode: "json",
+        doGenerate: vi.fn().mockResolvedValue({
+            content: [{ type: "text", text: "mock response" }],
+            finishReason: "stop",
+            usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+            warnings: [],
+        }),
+        doStream: vi.fn().mockResolvedValue({
+            stream: new ReadableStream({
+                start(controller) {
+                    controller.enqueue({
+                        type: "finish",
+                        finishReason: "stop",
+                        usage: { inputTokens: 8, outputTokens: 3, totalTokens: 11 },
+                    });
+                    controller.close();
+                },
+            }),
+        }),
+    };
+}
+// ---------------------------------------------------------------------------
+// getAISDKLanguageModel with middleware
+// ---------------------------------------------------------------------------
+describe("getAISDKLanguageModel with middleware", () => {
+    it("returns a model when no middleware is provided", () => {
+        const model = getAISDKLanguageModel("ollama", "llama3.2");
+        expect(model).toBeDefined();
+        expect(model.modelId).toBe("llama3.2");
+    });
+    it("returns a wrapped model when middleware is provided", () => {
+        const { middleware } = createRecordingMiddleware();
+        const model = getAISDKLanguageModel("ollama", "llama3.2", undefined, middleware);
+        expect(model).toBeDefined();
+        // wrapLanguageModel preserves modelId
+        expect(model.modelId).toBe("llama3.2");
+    });
+    it("wrapped model preserves doGenerate and doStream methods", () => {
+        const { middleware } = createRecordingMiddleware();
+        const model = getAISDKLanguageModel("ollama", "llama3.2", undefined, middleware);
+        expect(typeof model.doGenerate).toBe("function");
+        expect(typeof model.doStream).toBe("function");
+        expect(model.provider).toContain("ollama");
+    });
+});
+// ---------------------------------------------------------------------------
+// LLMProvider with middleware
+// ---------------------------------------------------------------------------
+describe("LLMProvider with middleware", () => {
+    const noop = () => { };
+    it("creates an AISdkClient without middleware", () => {
+        const provider = new LLMProvider(noop);
+        const client = provider.getClient("openai/gpt-4o");
+        expect(client).toBeDefined();
+        expect(client.type).toBe("aisdk");
+    });
+    it("creates an AISdkClient that carries middleware-wrapped model", () => {
+        const { middleware } = createRecordingMiddleware();
+        const provider = new LLMProvider(noop, middleware);
+        const client = provider.getClient("ollama/llama3.2");
+        expect(client).toBeDefined();
+        expect(client.type).toBe("aisdk");
+        const languageModel = client.getLanguageModel();
+        expect(languageModel).toBeDefined();
+        // Wrapped models should still expose the original modelId
+        expect(languageModel.modelId).toBe("llama3.2");
+    });
+    it("applies the same middleware to different models from getClient", () => {
+        const { middleware } = createRecordingMiddleware();
+        const provider = new LLMProvider(noop, middleware);
+        const clientA = provider.getClient("ollama/llama3.2");
+        const clientB = provider.getClient("ollama/mistral");
+        expect(clientA.getLanguageModel()).toBeDefined();
+        expect(clientB.getLanguageModel()).toBeDefined();
+        expect(clientA.getLanguageModel().modelId).toBe("llama3.2");
+        expect(clientB.getLanguageModel().modelId).toBe("mistral");
+    });
+});
+// ---------------------------------------------------------------------------
+// Middleware captures usage across act/extract/observe/agent code paths
+// ---------------------------------------------------------------------------
+describe("middleware captures usage from doGenerate and doStream", () => {
+    it("wrapGenerate fires and captures usage on doGenerate", async () => {
+        const { middleware, calls } = createRecordingMiddleware();
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        // Simulate what wrapLanguageModel does: import and wrap
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({ model: mockModel, middleware });
+        await wrapped.doGenerate({
+            prompt: [
+                {
+                    role: "user",
+                    content: [{ type: "text", text: "act: click button" }],
+                },
+            ],
+        });
+        expect(calls).toHaveLength(1);
+        expect(calls[0].type).toBe("generate");
+        expect(calls[0].modelId).toBe("gpt-4o");
+        expect(calls[0].usage).toEqual({ inputTokens: 10, outputTokens: 5 });
+    });
+    it("wrapStream fires on doStream", async () => {
+        const { middleware, calls } = createRecordingMiddleware();
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({ model: mockModel, middleware });
+        const result = await wrapped.doStream({
+            prompt: [
+                { role: "user", content: [{ type: "text", text: "stream this" }] },
+            ],
+        });
+        // Consume the stream to trigger the finish chunk
+        const reader = result.stream.getReader();
+        while (true) {
+            const { done } = await reader.read();
+            if (done)
+                break;
+        }
+        expect(calls).toHaveLength(1);
+        expect(calls[0].type).toBe("stream");
+        expect(calls[0].modelId).toBe("gpt-4o");
+    });
+    it("middleware fires for each separate doGenerate call (simulates act + extract + observe)", async () => {
+        const { middleware, calls } = createRecordingMiddleware();
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({ model: mockModel, middleware });
+        const callOpts = {
+            prompt: [
+                {
+                    role: "user",
+                    content: [{ type: "text", text: "test" }],
+                },
+            ],
+        };
+        // Simulate act call
+        await wrapped.doGenerate(callOpts);
+        // Simulate extract call
+        await wrapped.doGenerate(callOpts);
+        // Simulate observe call
+        await wrapped.doGenerate(callOpts);
+        expect(calls).toHaveLength(3);
+        expect(calls.every((c) => c.type === "generate")).toBe(true);
+        expect(calls.every((c) => c.modelId === "gpt-4o")).toBe(true);
+    });
+    it("middleware fires for agent multi-step calls (sequential doGenerate)", async () => {
+        const { middleware, calls } = createRecordingMiddleware();
+        const mockModel = createMockLanguageModel("openai/gpt-4.1", "gateway.openai");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({ model: mockModel, middleware });
+        const callOpts = {
+            prompt: [
+                {
+                    role: "user",
+                    content: [{ type: "text", text: "agent step" }],
+                },
+            ],
+        };
+        // Simulate multiple agent reasoning steps
+        await wrapped.doGenerate(callOpts);
+        await wrapped.doGenerate(callOpts);
+        await wrapped.doGenerate(callOpts);
+        await wrapped.doGenerate(callOpts);
+        await wrapped.doGenerate(callOpts);
+        expect(calls).toHaveLength(5);
+        calls.forEach((c) => {
+            expect(c.usage).toEqual({ inputTokens: 10, outputTokens: 5 });
+        });
+    });
+});
+// ---------------------------------------------------------------------------
+// ModelConfiguration-level middleware (the user-facing shape)
+// ---------------------------------------------------------------------------
+describe("middleware inside ModelConfiguration", () => {
+    it("resolveModelConfiguration extracts middleware from object config", () => {
+        const { middleware: mw } = createRecordingMiddleware();
+        const result = resolveModelConfiguration({
+            modelName: "openai/gpt-4o",
+            apiKey: "sk-test",
+            middleware: mw,
+        });
+        expect(result.modelName).toBe("openai/gpt-4o");
+        expect(result.middleware).toBe(mw);
+        expect(result.clientOptions).toEqual({ apiKey: "sk-test" });
+        expect(result.clientOptions && "middleware" in result.clientOptions).toBe(false);
+    });
+    it("string ModelConfiguration has no middleware", () => {
+        const result = resolveModelConfiguration("openai/gpt-4o");
+        expect(result.modelName).toBe("openai/gpt-4o");
+        expect(result.middleware).toBeUndefined();
+        expect(result.clientOptions).toBeUndefined();
+    });
+    it("middleware is separated from clientOptions when resolving per-method overrides", () => {
+        const { middleware: mw } = createRecordingMiddleware();
+        const result = resolveModelConfiguration({
+            modelName: "anthropic/claude-sonnet-4-20250514",
+            apiKey: "sk-ant-test",
+            middleware: mw,
+        });
+        expect(result.modelName).toBe("anthropic/claude-sonnet-4-20250514");
+        expect(result.middleware).toBe(mw);
+        expect(result.clientOptions).toEqual({ apiKey: "sk-ant-test" });
+        expect(result.clientOptions && "middleware" in result.clientOptions).toBe(false);
+    });
+});
+// ---------------------------------------------------------------------------
+// Practical middleware behaviors
+// ---------------------------------------------------------------------------
+describe("middleware that tracks duration", () => {
+    it("measures wall-clock time of doGenerate", async () => {
+        const durations = [];
+        const timingMiddleware = {
+            wrapGenerate: async ({ doGenerate }) => {
+                const start = performance.now();
+                const result = await doGenerate();
+                durations.push(performance.now() - start);
+                return result;
+            },
+        };
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({
+            model: mockModel,
+            middleware: timingMiddleware,
+        });
+        await wrapped.doGenerate({
+            prompt: [{ role: "user", content: [{ type: "text", text: "time me" }] }],
+        });
+        expect(durations).toHaveLength(1);
+        expect(durations[0]).toBeGreaterThanOrEqual(0);
+    });
+});
+describe("middleware that aggregates token usage", () => {
+    it("sums tokens across multiple calls like a billing tracker", async () => {
+        let totalInput = 0;
+        let totalOutput = 0;
+        const billingMiddleware = {
+            wrapGenerate: async ({ doGenerate }) => {
+                const result = await doGenerate();
+                totalInput += result.usage.inputTokens ?? 0;
+                totalOutput += result.usage.outputTokens ?? 0;
+                return result;
+            },
+        };
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({
+            model: mockModel,
+            middleware: billingMiddleware,
+        });
+        const callOpts = {
+            prompt: [
+                {
+                    role: "user",
+                    content: [{ type: "text", text: "go" }],
+                },
+            ],
+        };
+        // act
+        await wrapped.doGenerate(callOpts);
+        // extract
+        await wrapped.doGenerate(callOpts);
+        // observe
+        await wrapped.doGenerate(callOpts);
+        expect(totalInput).toBe(30);
+        expect(totalOutput).toBe(15);
+    });
+});
+describe("middleware that logs per-model call counts", () => {
+    it("tracks which models were called and how many times", async () => {
+        const modelCallCounts = new Map();
+        const countingMiddleware = {
+            wrapGenerate: async ({ doGenerate, model }) => {
+                const key = `${model.provider}/${model.modelId}`;
+                modelCallCounts.set(key, (modelCallCounts.get(key) ?? 0) + 1);
+                return doGenerate();
+            },
+        };
+        const modelA = createMockLanguageModel("gpt-4o", "openai.chat");
+        const modelB = createMockLanguageModel("claude-sonnet-4-20250514", "anthropic.messages");
+        const { wrapLanguageModel } = await import("ai");
+        const wrappedA = wrapLanguageModel({
+            model: modelA,
+            middleware: countingMiddleware,
+        });
+        const wrappedB = wrapLanguageModel({
+            model: modelB,
+            middleware: countingMiddleware,
+        });
+        const callOpts = {
+            prompt: [
+                {
+                    role: "user",
+                    content: [{ type: "text", text: "go" }],
+                },
+            ],
+        };
+        await wrappedA.doGenerate(callOpts);
+        await wrappedA.doGenerate(callOpts);
+        await wrappedB.doGenerate(callOpts);
+        expect(modelCallCounts.get("openai.chat/gpt-4o")).toBe(2);
+        expect(modelCallCounts.get("anthropic.messages/claude-sonnet-4-20250514")).toBe(1);
+        expect(modelCallCounts.size).toBe(2);
+    });
+});
+describe("middleware that detects errors", () => {
+    it("catches and re-throws errors from doGenerate while still recording the failure", async () => {
+        const errors = [];
+        const errorTrackingMiddleware = {
+            wrapGenerate: async ({ doGenerate }) => {
+                try {
+                    return await doGenerate();
+                }
+                catch (err) {
+                    errors.push(err);
+                    throw err;
+                }
+            },
+        };
+        const failingModel = {
+            specificationVersion: "v2",
+            provider: "openai.chat",
+            modelId: "gpt-4o",
+            defaultObjectGenerationMode: "json",
+            doGenerate: vi.fn().mockRejectedValue(new Error("rate limit exceeded")),
+            doStream: vi.fn(),
+        };
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({
+            model: failingModel,
+            middleware: errorTrackingMiddleware,
+        });
+        await expect(wrapped.doGenerate({
+            prompt: [{ role: "user", content: [{ type: "text", text: "fail" }] }],
+        })).rejects.toThrow("rate limit exceeded");
+        expect(errors).toHaveLength(1);
+        expect(errors[0].message).toBe("rate limit exceeded");
+    });
+});
+describe("chaining middleware across multiple wrapped models", () => {
+    it("same middleware instance sees calls from different models", async () => {
+        const seen = [];
+        const spyMiddleware = {
+            wrapGenerate: async ({ doGenerate, model }) => {
+                seen.push(model.modelId);
+                return doGenerate();
+            },
+        };
+        const mockA = createMockLanguageModel("gpt-4o", "openai.chat");
+        const mockB = createMockLanguageModel("claude-sonnet-4-20250514", "anthropic.messages");
+        const { wrapLanguageModel } = await import("ai");
+        const wrappedA = wrapLanguageModel({
+            model: mockA,
+            middleware: spyMiddleware,
+        });
+        const wrappedB = wrapLanguageModel({
+            model: mockB,
+            middleware: spyMiddleware,
+        });
+        const callOpts = {
+            prompt: [
+                {
+                    role: "user",
+                    content: [{ type: "text", text: "go" }],
+                },
+            ],
+        };
+        await wrappedA.doGenerate(callOpts);
+        await wrappedB.doGenerate(callOpts);
+        await wrappedA.doGenerate(callOpts);
+        expect(seen).toEqual(["gpt-4o", "claude-sonnet-4-20250514", "gpt-4o"]);
+    });
+});
+// ---------------------------------------------------------------------------
+// Edge cases
+// ---------------------------------------------------------------------------
+describe("middleware edge cases", () => {
+    it("middleware that only implements wrapGenerate still works for doStream", async () => {
+        const generateCalls = [];
+        const partialMiddleware = {
+            wrapGenerate: async ({ doGenerate, model }) => {
+                generateCalls.push(model.modelId);
+                return doGenerate();
+            },
+        };
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({
+            model: mockModel,
+            middleware: partialMiddleware,
+        });
+        const result = await wrapped.doStream({
+            prompt: [{ role: "user", content: [{ type: "text", text: "stream" }] }],
+        });
+        expect(result.stream).toBeDefined();
+        expect(generateCalls).toHaveLength(0);
+    });
+    it("middleware does not alter the response data", async () => {
+        const { middleware } = createRecordingMiddleware();
+        const mockModel = createMockLanguageModel("gpt-4o", "openai.chat");
+        const { wrapLanguageModel } = await import("ai");
+        const wrapped = wrapLanguageModel({ model: mockModel, middleware });
+        const result = await wrapped.doGenerate({
+            prompt: [{ role: "user", content: [{ type: "text", text: "test" }] }],
+        });
+        expect(result.content).toEqual([{ type: "text", text: "mock response" }]);
+        expect(result.usage).toEqual({
+            inputTokens: 10,
+            outputTokens: 5,
+            totalTokens: 15,
+        });
+        expect(result.finishReason).toBe("stop");
+    });
+});
+//# sourceMappingURL=llm-middleware.test.js.map

package/dist/esm/tests/unit/llm-middleware.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm-middleware.test.js","sourceRoot":"","sources":["../../../../tests/unit/llm-middleware.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAKlD,OAAO,EACL,qBAAqB,EACrB,WAAW,GACZ,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE/D;;;GAGG;AACH,SAAS,yBAAyB;IAChC,MAAM,KAAK,GAKL,EAAE,CAAC;IAET,MAAM,UAAU,GAA8B;QAC5C,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;YAC5C,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE;oBACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,SAAS;oBAClD,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,IAAI,SAAS;iBACrD;aACF,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE;YACxC,MAAM,MAAM,GAAG,MAAM,QAAQ,EAAE,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;IAEF,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AAC/B,CAAC;AAED;;;;GAIG;AACH,SAAS,uBAAuB,CAC9B,OAAe,EACf,QAAgB;IAEhB,OAAO;QACL,oBAAoB,EAAE,IAAI;QAC1B,QAAQ;QACR,OAAO;QACP,2BAA2B,EAAE,MAAM;QACnC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACpC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;YAClD,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;YAC5D,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAClC,MAAM,EAAE,IAAI,cAAc,CAAC;gBACzB,KAAK,CAAC,UAAU;oBACd,UAAU,CAAC,OAAO,CAAC;wBACjB,IAAI,EAAE,QAAQ;wBACd,YAAY,EAAE,MAAM;wBACpB,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;qBAC5D,CAAC,CAAC;oBACH,UAAU,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC;aACF,CAAC;SACH,CAAC;KAC2B,CAAC;AAClC,CAAC;AAED,8EAA8E;AAC9E,wCAAwC;AACxC,8EAA8E;AAE9E,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACrD,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,KAAK,GAAG,qBAAqB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,qBAAqB,CACjC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5B,sCAAsC;QACtC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,qBAAqB,CACjC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;QAEF,MAAM,CAAC,OAAO,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,8BAA8B;AAC9B,8EAA8E;AAE9E,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,MAAM,IAAI,GAAG,GAAG,EAAE,GAAE,CAAC,CAAC;IAEtB,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,eAAwB,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,iBAA0B,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAElC,MAAM,aAAa,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAChD,MAAM,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;QACpC,0DAA0D;QAC1D,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEnD,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,iBAA0B,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,gBAAyB,CAAC,CAAC;QAE9D,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5D,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,wEAAwE;AACxE,8EAA8E;AAE9E,QAAQ,CAAC,wDAAwD,EAAE,GAAG,EAAE;IACtE,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,wDAAwD;QACxD,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,UAAU,CAAC;YACvB,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;iBACvD;aACF;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;YACpC,MAAM,EAAE;gBACN,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;aACnE;SACF,CAAC,CAAC;QAEH,iDAAiD;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACzC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,IAAI;gBAAE,MAAM;QAClB,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;QACtG,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBACnD;aACF;SACF,CAAC;QAEF,oBAAoB;QACpB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,wBAAwB;QACxB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,wBAAwB;QACxB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CACvC,gBAAgB,EAChB,gBAAgB,CACjB,CAAC;QAEF,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;iBACzD;aACF;SACF,CAAC;QAEF,0CAA0C;QAC1C,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YAClB,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,8DAA8D;AAC9D,8EAA8E;AAE9E,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,yBAAyB,CAAC;YACvC,SAAS,EAAE,eAAe;YAC1B,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,aAAa,IAAI,YAAY,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CACvE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,MAAM,GAAG,yBAAyB,CAAC,eAAe,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,aAAa,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,yBAAyB,CAAC;YACvC,SAAS,EAAE,oCAAoC;YAC/C,MAAM,EAAE,aAAa;YACrB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,aAAa,IAAI,YAAY,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CACvE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,iCAAiC;AACjC,8EAA8E;AAE9E,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,SAAS,GAAa,EAAE,CAAC;QAE/B,MAAM,gBAAgB,GAA8B;YAClD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;gBACrC,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;gBAChC,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;gBAClC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;gBAC1C,OAAO,MAAM,CAAC;YAChB,CAAC;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,gBAAgB;SAC7B,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,UAAU,CAAC;YACvB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SACzE,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,MAAM,iBAAiB,GAA8B;YACnD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;gBACrC,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;gBAClC,UAAU,IAAI,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;gBAC5C,WAAW,IAAI,MAAM,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;gBAC9C,OAAO,MAAM,CAAC;YAChB,CAAC;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,iBAAiB;SAC9B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACjD;aACF;SACF,CAAC;QAEF,MAAM;QACN,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,UAAU;QACV,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,UAAU;QACV,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5B,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;IAC1D,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;QAElD,MAAM,kBAAkB,GAA8B;YACpD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5C,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBACjD,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC9D,OAAO,UAAU,EAAE,CAAC;YACtB,CAAC;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,uBAAuB,CACpC,0BAA0B,EAC1B,oBAAoB,CACrB,CAAC;QAEF,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,MAAM;YACb,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,MAAM;YACb,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACjD;aACF;SACF,CAAC;QAEF,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEpC,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CACJ,eAAe,CAAC,GAAG,CAAC,6CAA6C,CAAC,CACnE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACV,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,uBAAuB,GAA8B;YACzD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;gBACrC,IAAI,CAAC;oBACH,OAAO,MAAM,UAAU,EAAE,CAAC;gBAC5B,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,CAAC,GAAY,CAAC,CAAC;oBAC1B,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC;SACF,CAAC;QAEF,MAAM,YAAY,GAAoB;YACpC,oBAAoB,EAAE,IAAI;YAC1B,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,QAAQ;YACjB,2BAA2B,EAAE,MAAM;YACnC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACvE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SACY,CAAC;QAEhC,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,uBAAuB;SACpC,CAAC,CAAC;QAEH,MAAM,MAAM,CACV,OAAO,CAAC,UAAU,CAAC;YACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;SACtE,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAEzC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oDAAoD,EAAE,GAAG,EAAE;IAClE,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,MAAM,aAAa,GAA8B;YAC/C,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACzB,OAAO,UAAU,EAAE,CAAC;YACtB,CAAC;SACF,CAAC;QAEF,MAAM,KAAK,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,uBAAuB,CACnC,0BAA0B,EAC1B,oBAAoB,CACrB,CAAC;QAEF,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,aAAa;SAC1B,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,aAAa;SAC1B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACjD;aACF;SACF,CAAC;QAEF,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEpC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,0BAA0B,EAAE,QAAQ,CAAC,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,aAAa;AACb,8EAA8E;AAE9E,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,MAAM,iBAAiB,GAA8B;YACnD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5C,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAClC,OAAO,UAAU,EAAE,CAAC;YACtB,CAAC;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,iBAAiB;SAC9B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;YACpC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;SACxE,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAEpC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;YACtC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;SACtE,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YAC3B,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,EAAE;SAChB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport type {\n  LanguageModelV2,\n  LanguageModelV2Middleware,\n} from \"@ai-sdk/provider\";\nimport {\n  getAISDKLanguageModel,\n  LLMProvider,\n} from \"../../lib/v3/llm/LLMProvider.js\";\nimport { resolveModelConfiguration } from \"../../lib/v3/v3.js\";\n\n/**\n * Creates a recording middleware that captures every doGenerate / doStream\n * invocation along with the model identity and returned usage data.\n */\nfunction createRecordingMiddleware() {\n  const calls: {\n    type: \"generate\" | \"stream\";\n    modelId: string;\n    provider: string;\n    usage?: { inputTokens?: number; outputTokens?: number };\n  }[] = [];\n\n  const middleware: LanguageModelV2Middleware = {\n    wrapGenerate: async ({ doGenerate, model }) => {\n      const result = await doGenerate();\n      calls.push({\n        type: \"generate\",\n        modelId: model.modelId,\n        provider: model.provider,\n        usage: {\n          inputTokens: result.usage.inputTokens ?? undefined,\n          outputTokens: result.usage.outputTokens ?? undefined,\n        },\n      });\n      return result;\n    },\n    wrapStream: async ({ doStream, model }) => {\n      const result = await doStream();\n      calls.push({\n        type: \"stream\",\n        modelId: model.modelId,\n        provider: model.provider,\n      });\n      return result;\n    },\n  };\n\n  return { middleware, calls };\n}\n\n/**\n * Creates a minimal mock LanguageModelV2 that returns canned results\n * without hitting any real provider. Useful for testing the wrapping\n * mechanics in isolation.\n */\nfunction createMockLanguageModel(\n  modelId: string,\n  provider: string,\n): LanguageModelV2 {\n  return {\n    specificationVersion: \"v2\",\n    provider,\n    modelId,\n    defaultObjectGenerationMode: \"json\",\n    doGenerate: vi.fn().mockResolvedValue({\n      content: [{ type: \"text\", text: \"mock response\" }],\n      finishReason: \"stop\",\n      usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },\n      warnings: [],\n    }),\n    doStream: vi.fn().mockResolvedValue({\n      stream: new ReadableStream({\n        start(controller) {\n          controller.enqueue({\n            type: \"finish\",\n            finishReason: \"stop\",\n            usage: { inputTokens: 8, outputTokens: 3, totalTokens: 11 },\n          });\n          controller.close();\n        },\n      }),\n    }),\n  } as unknown as LanguageModelV2;\n}\n\n// ---------------------------------------------------------------------------\n// getAISDKLanguageModel with middleware\n// ---------------------------------------------------------------------------\n\ndescribe(\"getAISDKLanguageModel with middleware\", () => {\n  it(\"returns a model when no middleware is provided\", () => {\n    const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\");\n    expect(model).toBeDefined();\n    expect(model.modelId).toBe(\"llama3.2\");\n  });\n\n  it(\"returns a wrapped model when middleware is provided\", () => {\n    const { middleware } = createRecordingMiddleware();\n    const model = getAISDKLanguageModel(\n      \"ollama\",\n      \"llama3.2\",\n      undefined,\n      middleware,\n    );\n    expect(model).toBeDefined();\n    // wrapLanguageModel preserves modelId\n    expect(model.modelId).toBe(\"llama3.2\");\n  });\n\n  it(\"wrapped model preserves doGenerate and doStream methods\", () => {\n    const { middleware } = createRecordingMiddleware();\n    const model = getAISDKLanguageModel(\n      \"ollama\",\n      \"llama3.2\",\n      undefined,\n      middleware,\n    );\n\n    expect(typeof model.doGenerate).toBe(\"function\");\n    expect(typeof model.doStream).toBe(\"function\");\n    expect(model.provider).toContain(\"ollama\");\n  });\n});\n\n// ---------------------------------------------------------------------------\n// LLMProvider with middleware\n// ---------------------------------------------------------------------------\n\ndescribe(\"LLMProvider with middleware\", () => {\n  const noop = () => {};\n\n  it(\"creates an AISdkClient without middleware\", () => {\n    const provider = new LLMProvider(noop);\n    const client = provider.getClient(\"openai/gpt-4o\" as never);\n    expect(client).toBeDefined();\n    expect(client.type).toBe(\"aisdk\");\n  });\n\n  it(\"creates an AISdkClient that carries middleware-wrapped model\", () => {\n    const { middleware } = createRecordingMiddleware();\n    const provider = new LLMProvider(noop, middleware);\n    const client = provider.getClient(\"ollama/llama3.2\" as never);\n    expect(client).toBeDefined();\n    expect(client.type).toBe(\"aisdk\");\n\n    const languageModel = client.getLanguageModel();\n    expect(languageModel).toBeDefined();\n    // Wrapped models should still expose the original modelId\n    expect(languageModel.modelId).toBe(\"llama3.2\");\n  });\n\n  it(\"applies the same middleware to different models from getClient\", () => {\n    const { middleware } = createRecordingMiddleware();\n    const provider = new LLMProvider(noop, middleware);\n\n    const clientA = provider.getClient(\"ollama/llama3.2\" as never);\n    const clientB = provider.getClient(\"ollama/mistral\" as never);\n\n    expect(clientA.getLanguageModel()).toBeDefined();\n    expect(clientB.getLanguageModel()).toBeDefined();\n    expect(clientA.getLanguageModel().modelId).toBe(\"llama3.2\");\n    expect(clientB.getLanguageModel().modelId).toBe(\"mistral\");\n  });\n});\n\n// ---------------------------------------------------------------------------\n// Middleware captures usage across act/extract/observe/agent code paths\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware captures usage from doGenerate and doStream\", () => {\n  it(\"wrapGenerate fires and captures usage on doGenerate\", async () => {\n    const { middleware, calls } = createRecordingMiddleware();\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n    // Simulate what wrapLanguageModel does: import and wrap\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n    await wrapped.doGenerate({\n      prompt: [\n        {\n          role: \"user\",\n          content: [{ type: \"text\", text: \"act: click button\" }],\n        },\n      ],\n    });\n\n    expect(calls).toHaveLength(1);\n    expect(calls[0].type).toBe(\"generate\");\n    expect(calls[0].modelId).toBe(\"gpt-4o\");\n    expect(calls[0].usage).toEqual({ inputTokens: 10, outputTokens: 5 });\n  });\n\n  it(\"wrapStream fires on doStream\", async () => {\n    const { middleware, calls } = createRecordingMiddleware();\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n    const result = await wrapped.doStream({\n      prompt: [\n        { role: \"user\", content: [{ type: \"text\", text: \"stream this\" }] },\n      ],\n    });\n\n    // Consume the stream to trigger the finish chunk\n    const reader = result.stream.getReader();\n    while (true) {\n      const { done } = await reader.read();\n      if (done) break;\n    }\n\n    expect(calls).toHaveLength(1);\n    expect(calls[0].type).toBe(\"stream\");\n    expect(calls[0].modelId).toBe(\"gpt-4o\");\n  });\n\n  it(\"middleware fires for each separate doGenerate call (simulates act + extract + observe)\", async () => {\n    const { middleware, calls } = createRecordingMiddleware();\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n    const callOpts = {\n      prompt: [\n        {\n          role: \"user\" as const,\n          content: [{ type: \"text\" as const, text: \"test\" }],\n        },\n      ],\n    };\n\n    // Simulate act call\n    await wrapped.doGenerate(callOpts);\n    // Simulate extract call\n    await wrapped.doGenerate(callOpts);\n    // Simulate observe call\n    await wrapped.doGenerate(callOpts);\n\n    expect(calls).toHaveLength(3);\n    expect(calls.every((c) => c.type === \"generate\")).toBe(true);\n    expect(calls.every((c) => c.modelId === \"gpt-4o\")).toBe(true);\n  });\n\n  it(\"middleware fires for agent multi-step calls (sequential doGenerate)\", async () => {\n    const { middleware, calls } = createRecordingMiddleware();\n    const mockModel = createMockLanguageModel(\n      \"openai/gpt-4.1\",\n      \"gateway.openai\",\n    );\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n    const callOpts = {\n      prompt: [\n        {\n          role: \"user\" as const,\n          content: [{ type: \"text\" as const, text: \"agent step\" }],\n        },\n      ],\n    };\n\n    // Simulate multiple agent reasoning steps\n    await wrapped.doGenerate(callOpts);\n    await wrapped.doGenerate(callOpts);\n    await wrapped.doGenerate(callOpts);\n    await wrapped.doGenerate(callOpts);\n    await wrapped.doGenerate(callOpts);\n\n    expect(calls).toHaveLength(5);\n    calls.forEach((c) => {\n      expect(c.usage).toEqual({ inputTokens: 10, outputTokens: 5 });\n    });\n  });\n});\n\n// ---------------------------------------------------------------------------\n// ModelConfiguration-level middleware (the user-facing shape)\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware inside ModelConfiguration\", () => {\n  it(\"resolveModelConfiguration extracts middleware from object config\", () => {\n    const { middleware: mw } = createRecordingMiddleware();\n    const result = resolveModelConfiguration({\n      modelName: \"openai/gpt-4o\",\n      apiKey: \"sk-test\",\n      middleware: mw,\n    });\n\n    expect(result.modelName).toBe(\"openai/gpt-4o\");\n    expect(result.middleware).toBe(mw);\n    expect(result.clientOptions).toEqual({ apiKey: \"sk-test\" });\n    expect(result.clientOptions && \"middleware\" in result.clientOptions).toBe(\n      false,\n    );\n  });\n\n  it(\"string ModelConfiguration has no middleware\", () => {\n    const result = resolveModelConfiguration(\"openai/gpt-4o\");\n    expect(result.modelName).toBe(\"openai/gpt-4o\");\n    expect(result.middleware).toBeUndefined();\n    expect(result.clientOptions).toBeUndefined();\n  });\n\n  it(\"middleware is separated from clientOptions when resolving per-method overrides\", () => {\n    const { middleware: mw } = createRecordingMiddleware();\n    const result = resolveModelConfiguration({\n      modelName: \"anthropic/claude-sonnet-4-20250514\",\n      apiKey: \"sk-ant-test\",\n      middleware: mw,\n    });\n\n    expect(result.modelName).toBe(\"anthropic/claude-sonnet-4-20250514\");\n    expect(result.middleware).toBe(mw);\n    expect(result.clientOptions).toEqual({ apiKey: \"sk-ant-test\" });\n    expect(result.clientOptions && \"middleware\" in result.clientOptions).toBe(\n      false,\n    );\n  });\n});\n\n// ---------------------------------------------------------------------------\n// Practical middleware behaviors\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware that tracks duration\", () => {\n  it(\"measures wall-clock time of doGenerate\", async () => {\n    const durations: number[] = [];\n\n    const timingMiddleware: LanguageModelV2Middleware = {\n      wrapGenerate: async ({ doGenerate }) => {\n        const start = performance.now();\n        const result = await doGenerate();\n        durations.push(performance.now() - start);\n        return result;\n      },\n    };\n\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({\n      model: mockModel,\n      middleware: timingMiddleware,\n    });\n\n    await wrapped.doGenerate({\n      prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"time me\" }] }],\n    });\n\n    expect(durations).toHaveLength(1);\n    expect(durations[0]).toBeGreaterThanOrEqual(0);\n  });\n});\n\ndescribe(\"middleware that aggregates token usage\", () => {\n  it(\"sums tokens across multiple calls like a billing tracker\", async () => {\n    let totalInput = 0;\n    let totalOutput = 0;\n\n    const billingMiddleware: LanguageModelV2Middleware = {\n      wrapGenerate: async ({ doGenerate }) => {\n        const result = await doGenerate();\n        totalInput += result.usage.inputTokens ?? 0;\n        totalOutput += result.usage.outputTokens ?? 0;\n        return result;\n      },\n    };\n\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({\n      model: mockModel,\n      middleware: billingMiddleware,\n    });\n\n    const callOpts = {\n      prompt: [\n        {\n          role: \"user\" as const,\n          content: [{ type: \"text\" as const, text: \"go\" }],\n        },\n      ],\n    };\n\n    // act\n    await wrapped.doGenerate(callOpts);\n    // extract\n    await wrapped.doGenerate(callOpts);\n    // observe\n    await wrapped.doGenerate(callOpts);\n\n    expect(totalInput).toBe(30);\n    expect(totalOutput).toBe(15);\n  });\n});\n\ndescribe(\"middleware that logs per-model call counts\", () => {\n  it(\"tracks which models were called and how many times\", async () => {\n    const modelCallCounts = new Map<string, number>();\n\n    const countingMiddleware: LanguageModelV2Middleware = {\n      wrapGenerate: async ({ doGenerate, model }) => {\n        const key = `${model.provider}/${model.modelId}`;\n        modelCallCounts.set(key, (modelCallCounts.get(key) ?? 0) + 1);\n        return doGenerate();\n      },\n    };\n\n    const modelA = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n    const modelB = createMockLanguageModel(\n      \"claude-sonnet-4-20250514\",\n      \"anthropic.messages\",\n    );\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrappedA = wrapLanguageModel({\n      model: modelA,\n      middleware: countingMiddleware,\n    });\n    const wrappedB = wrapLanguageModel({\n      model: modelB,\n      middleware: countingMiddleware,\n    });\n\n    const callOpts = {\n      prompt: [\n        {\n          role: \"user\" as const,\n          content: [{ type: \"text\" as const, text: \"go\" }],\n        },\n      ],\n    };\n\n    await wrappedA.doGenerate(callOpts);\n    await wrappedA.doGenerate(callOpts);\n    await wrappedB.doGenerate(callOpts);\n\n    expect(modelCallCounts.get(\"openai.chat/gpt-4o\")).toBe(2);\n    expect(\n      modelCallCounts.get(\"anthropic.messages/claude-sonnet-4-20250514\"),\n    ).toBe(1);\n    expect(modelCallCounts.size).toBe(2);\n  });\n});\n\ndescribe(\"middleware that detects errors\", () => {\n  it(\"catches and re-throws errors from doGenerate while still recording the failure\", async () => {\n    const errors: Error[] = [];\n\n    const errorTrackingMiddleware: LanguageModelV2Middleware = {\n      wrapGenerate: async ({ doGenerate }) => {\n        try {\n          return await doGenerate();\n        } catch (err) {\n          errors.push(err as Error);\n          throw err;\n        }\n      },\n    };\n\n    const failingModel: LanguageModelV2 = {\n      specificationVersion: \"v2\",\n      provider: \"openai.chat\",\n      modelId: \"gpt-4o\",\n      defaultObjectGenerationMode: \"json\",\n      doGenerate: vi.fn().mockRejectedValue(new Error(\"rate limit exceeded\")),\n      doStream: vi.fn(),\n    } as unknown as LanguageModelV2;\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({\n      model: failingModel,\n      middleware: errorTrackingMiddleware,\n    });\n\n    await expect(\n      wrapped.doGenerate({\n        prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"fail\" }] }],\n      }),\n    ).rejects.toThrow(\"rate limit exceeded\");\n\n    expect(errors).toHaveLength(1);\n    expect(errors[0].message).toBe(\"rate limit exceeded\");\n  });\n});\n\ndescribe(\"chaining middleware across multiple wrapped models\", () => {\n  it(\"same middleware instance sees calls from different models\", async () => {\n    const seen: string[] = [];\n\n    const spyMiddleware: LanguageModelV2Middleware = {\n      wrapGenerate: async ({ doGenerate, model }) => {\n        seen.push(model.modelId);\n        return doGenerate();\n      },\n    };\n\n    const mockA = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n    const mockB = createMockLanguageModel(\n      \"claude-sonnet-4-20250514\",\n      \"anthropic.messages\",\n    );\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrappedA = wrapLanguageModel({\n      model: mockA,\n      middleware: spyMiddleware,\n    });\n    const wrappedB = wrapLanguageModel({\n      model: mockB,\n      middleware: spyMiddleware,\n    });\n\n    const callOpts = {\n      prompt: [\n        {\n          role: \"user\" as const,\n          content: [{ type: \"text\" as const, text: \"go\" }],\n        },\n      ],\n    };\n\n    await wrappedA.doGenerate(callOpts);\n    await wrappedB.doGenerate(callOpts);\n    await wrappedA.doGenerate(callOpts);\n\n    expect(seen).toEqual([\"gpt-4o\", \"claude-sonnet-4-20250514\", \"gpt-4o\"]);\n  });\n});\n\n// ---------------------------------------------------------------------------\n// Edge cases\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware edge cases\", () => {\n  it(\"middleware that only implements wrapGenerate still works for doStream\", async () => {\n    const generateCalls: string[] = [];\n    const partialMiddleware: LanguageModelV2Middleware = {\n      wrapGenerate: async ({ doGenerate, model }) => {\n        generateCalls.push(model.modelId);\n        return doGenerate();\n      },\n    };\n\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({\n      model: mockModel,\n      middleware: partialMiddleware,\n    });\n\n    const result = await wrapped.doStream({\n      prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"stream\" }] }],\n    });\n    expect(result.stream).toBeDefined();\n\n    expect(generateCalls).toHaveLength(0);\n  });\n\n  it(\"middleware does not alter the response data\", async () => {\n    const { middleware } = createRecordingMiddleware();\n    const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n    const { wrapLanguageModel } = await import(\"ai\");\n    const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n    const result = await wrapped.doGenerate({\n      prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"test\" }] }],\n    });\n\n    expect(result.content).toEqual([{ type: \"text\", text: \"mock response\" }]);\n    expect(result.usage).toEqual({\n      inputTokens: 10,\n      outputTokens: 5,\n      totalTokens: 15,\n    });\n    expect(result.finishReason).toBe(\"stop\");\n  });\n});\n"]}

package/dist/esm/tests/unit/model-deprecation.test.js

@@ -6,10 +6,7 @@ const mockClientOptions = { apiKey: "test-api-key-for-testing" };
 describe("Model format deprecation", () => {
     describe("UnsupportedModelError", () => {
         it("includes guidance to use provider/model format for unknown model names", () => {
-            const error = new UnsupportedModelError([
-                "gpt-4o",
-                "claude-3-5-sonnet-latest",
-            ]);
+            const error = new UnsupportedModelError(["gpt-4o", "gemini-2.0-flash"]);
             // Should mention the new format
             expect(error.message).toContain("provider/model");
             // Should include link to docs
@@ -68,15 +65,15 @@ describe("Model format deprecation", () => {
             // The client should be an OpenAIClient (check constructor name)
             expect(client.constructor.name).toBe("OpenAIClient");
         });
-        it("returns AnthropicClient for legacy Anthropic model names", () => {
+        it("returns GoogleClient for legacy Google model names", () => {
             const logs = [];
             const logger = (line) => logs.push(line);
             const provider = new LLMProvider(logger);
-            const client = provider.getClient("claude-3-5-sonnet-latest", mockClientOptions);
+            const client = provider.getClient("gemini-2.0-flash", mockClientOptions);
             // Should return a client
             expect(client).toBeDefined();
-            // The client should be an AnthropicClient
-            expect(client.constructor.name).toBe("AnthropicClient");
+            // The client should be a GoogleClient
+            expect(client.constructor.name).toBe("GoogleClient");
         });
     });
     describe("LLMProvider.getClient error handling", () => {

package/dist/esm/tests/unit/model-deprecation.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-deprecation.test.js","sourceRoot":"","sources":["../../../../tests/unit/model-deprecation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EACL,qBAAqB,EACrB,kCAAkC,GACnC,MAAM,wCAAwC,CAAC;AAGhD,qDAAqD;AACrD,MAAM,iBAAiB,GAAG,EAAE,MAAM,EAAE,0BAA0B,EAAE,CAAC;AAEjE,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC;gBACtC,QAAQ;gBACR,0BAA0B;aAC3B,CAAC,CAAC;YAEH,gCAAgC;YAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAClD,8BAA8B;YAC9B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAC7B,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;YAEpD,6CAA6C;YAC7C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;YAE/D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAC7B,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,qEAAqE;YACrE,yDAAyD;YACzD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAE/D,qCAAqC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,sDAAsD;YACtD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,kBAAmB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAEhD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YAEF,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,kBAAmB,CAAC,OAAO,CAAC;YAC5C,2CAA2C;YAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC5C,yBAAyB;YACzB,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAE/D,yBAAyB;YACzB,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAC/B,0BAA0B,EAC1B,iBAAiB,CAClB,CAAC;YAEF,yBAAyB;YACzB,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,0CAA0C;YAC1C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,iEAAiE;YACjE,MAAM,CAAC,GAAG,EAAE;gBACV,QAAQ,CAAC,SAAS,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,IAAI,CAAC;gBACH,QAAQ,CAAC,SAAS,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAAC;YAC9D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAE,KAAe,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yFAAyF,EAAE,GAAG,EAAE;YACjG,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,sCAAsC;YACtC,MAAM,CAAC,GAAG,EAAE;gBACV,QAAQ,CAAC,SAAS,CAAC,6BAA6B,EAAE,iBAAiB,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,IAAI,CAAC;gBACH,QAAQ,CAAC,SAAS,CAAC,6BAA6B,EAAE,iBAAiB,CAAC,CAAC;YACvE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAI,KAAe,CAAC,OAAO,CAAC;gBACzC,8BAA8B;gBAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACpC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,uBAAuB;YACvB,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,wCAAwC;YACxC,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,CAAC,aAAa,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { LLMProvider } from \"../../lib/v3/llm/LLMProvider.js\";\nimport {\n  UnsupportedModelError,\n  UnsupportedAISDKModelProviderError,\n} from \"../../lib/v3/types/public/sdkErrors.js\";\nimport type { LogLine } from \"../../lib/v3/types/public/logs.js\";\n\n// Mock client options with fake API keys for testing\nconst mockClientOptions = { apiKey: \"test-api-key-for-testing\" };\n\ndescribe(\"Model format deprecation\", () => {\n  describe(\"UnsupportedModelError\", () => {\n    it(\"includes guidance to use provider/model format for unknown model names\", () => {\n      const error = new UnsupportedModelError([\n        \"gpt-4o\",\n        \"claude-3-5-sonnet-latest\",\n      ]);\n\n      // Should mention the new format\n      expect(error.message).toContain(\"provider/model\");\n      // Should include link to docs\n      expect(error.message).toContain(\n        \"https://docs.stagehand.dev/v3/configuration/models\",\n      );\n    });\n\n    it(\"includes example of provider/model format\", () => {\n      const error = new UnsupportedModelError([\"gpt-4o\"]);\n\n      // Should provide examples like openai/gpt-4o\n      expect(error.message).toContain(\"openai/gpt-4o\");\n      expect(error.message).toContain(\"anthropic/claude-sonnet-4\");\n    });\n\n    it(\"works with feature parameter\", () => {\n      const error = new UnsupportedModelError([\"gpt-4o\"], \"extract\");\n\n      expect(error.message).toContain(\"extract\");\n      expect(error.message).toContain(\"provider/model\");\n      expect(error.message).toContain(\n        \"https://docs.stagehand.dev/v3/configuration/models\",\n      );\n    });\n  });\n\n  describe(\"LLMProvider.getClient deprecation warning\", () => {\n    it(\"logs deprecation warning for legacy model names\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Using a legacy model name like \"gpt-4o\" instead of \"openai/gpt-4o\"\n      // Should not throw, but should log a deprecation warning\n      const client = provider.getClient(\"gpt-4o\", mockClientOptions);\n\n      // Should return a client (not throw)\n      expect(client).toBeDefined();\n\n      // Should have logged a deprecation warning at level 0\n      const deprecationWarning = logs.find(\n        (log) =>\n          log.message.toLowerCase().includes(\"deprecated\") ||\n          log.message.toLowerCase().includes(\"deprecation\"),\n      );\n      expect(deprecationWarning).toBeDefined();\n      expect(deprecationWarning!.level).toBe(0);\n    });\n\n    it(\"deprecation warning mentions provider/model format\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      provider.getClient(\"gpt-4o\", mockClientOptions);\n\n      const deprecationWarning = logs.find(\n        (log) =>\n          log.message.toLowerCase().includes(\"deprecated\") ||\n          log.message.toLowerCase().includes(\"deprecation\"),\n      );\n\n      expect(deprecationWarning).toBeDefined();\n      const message = deprecationWarning!.message;\n      // Should mention the provider/model format\n      expect(message).toContain(\"provider/model\");\n      // Should give an example\n      expect(message).toContain(\"openai/gpt-5\");\n    });\n\n    it(\"returns OpenAIClient for legacy OpenAI model names\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      const client = provider.getClient(\"gpt-4o\", mockClientOptions);\n\n      // Should return a client\n      expect(client).toBeDefined();\n      // The client should be an OpenAIClient (check constructor name)\n      expect(client.constructor.name).toBe(\"OpenAIClient\");\n    });\n\n    it(\"returns AnthropicClient for legacy Anthropic model names\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      const client = provider.getClient(\n        \"claude-3-5-sonnet-latest\",\n        mockClientOptions,\n      );\n\n      // Should return a client\n      expect(client).toBeDefined();\n      // The client should be an AnthropicClient\n      expect(client.constructor.name).toBe(\"AnthropicClient\");\n    });\n  });\n\n  describe(\"LLMProvider.getClient error handling\", () => {\n    it(\"throws UnsupportedModelError for unknown model without slash\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Unknown model without slash should throw UnsupportedModelError\n      expect(() => {\n        provider.getClient(\"some-unknown-model\", mockClientOptions);\n      }).toThrow(UnsupportedModelError);\n    });\n\n    it(\"UnsupportedModelError includes provider/model format guidance\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      try {\n        provider.getClient(\"some-unknown-model\", mockClientOptions);\n      } catch (error) {\n        expect((error as Error).message).toContain(\"provider/model\");\n      }\n    });\n\n    it(\"throws UnsupportedAISDKModelProviderError for invalid provider in provider/model format\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Invalid provider but correct format\n      expect(() => {\n        provider.getClient(\"invalid-provider/some-model\", mockClientOptions);\n      }).toThrow(UnsupportedAISDKModelProviderError);\n    });\n\n    it(\"UnsupportedAISDKModelProviderError lists valid providers\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      try {\n        provider.getClient(\"invalid-provider/some-model\", mockClientOptions);\n      } catch (error) {\n        const message = (error as Error).message;\n        // Should list valid providers\n        expect(message).toContain(\"openai\");\n        expect(message).toContain(\"anthropic\");\n        expect(message).toContain(\"google\");\n      }\n    });\n  });\n\n  describe(\"new provider/model format\", () => {\n    it(\"does not log deprecation warning for provider/model format\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Using the new format\n      const client = provider.getClient(\"openai/gpt-4o\", mockClientOptions);\n\n      expect(client).toBeDefined();\n\n      // Should NOT have a deprecation warning\n      const deprecationWarning = logs.find(\n        (log) =>\n          log.message.toLowerCase().includes(\"deprecated\") ||\n          log.message.toLowerCase().includes(\"deprecation\"),\n      );\n      expect(deprecationWarning).toBeUndefined();\n    });\n  });\n});\n"]}
+{"version":3,"file":"model-deprecation.test.js","sourceRoot":"","sources":["../../../../tests/unit/model-deprecation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EACL,qBAAqB,EACrB,kCAAkC,GACnC,MAAM,wCAAwC,CAAC;AAGhD,qDAAqD;AACrD,MAAM,iBAAiB,GAAG,EAAE,MAAM,EAAE,0BAA0B,EAAE,CAAC;AAEjE,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC;YAExE,gCAAgC;YAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAClD,8BAA8B;YAC9B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAC7B,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;YAEpD,6CAA6C;YAC7C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;YAE/D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAC7B,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,qEAAqE;YACrE,yDAAyD;YACzD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAE/D,qCAAqC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,sDAAsD;YACtD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,kBAAmB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAEhD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YAEF,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,kBAAmB,CAAC,OAAO,CAAC;YAC5C,2CAA2C;YAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC5C,yBAAyB;YACzB,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAE/D,yBAAyB;YACzB,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;YAEzE,yBAAyB;YACzB,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,sCAAsC;YACtC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,iEAAiE;YACjE,MAAM,CAAC,GAAG,EAAE;gBACV,QAAQ,CAAC,SAAS,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,IAAI,CAAC;gBACH,QAAQ,CAAC,SAAS,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAAC;YAC9D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAE,KAAe,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yFAAyF,EAAE,GAAG,EAAE;YACjG,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,sCAAsC;YACtC,MAAM,CAAC,GAAG,EAAE;gBACV,QAAQ,CAAC,SAAS,CAAC,6BAA6B,EAAE,iBAAiB,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,IAAI,CAAC;gBACH,QAAQ,CAAC,SAAS,CAAC,6BAA6B,EAAE,iBAAiB,CAAC,CAAC;YACvE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAI,KAAe,CAAC,OAAO,CAAC;gBACzC,8BAA8B;gBAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACpC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,uBAAuB;YACvB,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,wCAAwC;YACxC,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,CAAC,aAAa,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { LLMProvider } from \"../../lib/v3/llm/LLMProvider.js\";\nimport {\n  UnsupportedModelError,\n  UnsupportedAISDKModelProviderError,\n} from \"../../lib/v3/types/public/sdkErrors.js\";\nimport type { LogLine } from \"../../lib/v3/types/public/logs.js\";\n\n// Mock client options with fake API keys for testing\nconst mockClientOptions = { apiKey: \"test-api-key-for-testing\" };\n\ndescribe(\"Model format deprecation\", () => {\n  describe(\"UnsupportedModelError\", () => {\n    it(\"includes guidance to use provider/model format for unknown model names\", () => {\n      const error = new UnsupportedModelError([\"gpt-4o\", \"gemini-2.0-flash\"]);\n\n      // Should mention the new format\n      expect(error.message).toContain(\"provider/model\");\n      // Should include link to docs\n      expect(error.message).toContain(\n        \"https://docs.stagehand.dev/v3/configuration/models\",\n      );\n    });\n\n    it(\"includes example of provider/model format\", () => {\n      const error = new UnsupportedModelError([\"gpt-4o\"]);\n\n      // Should provide examples like openai/gpt-4o\n      expect(error.message).toContain(\"openai/gpt-4o\");\n      expect(error.message).toContain(\"anthropic/claude-sonnet-4\");\n    });\n\n    it(\"works with feature parameter\", () => {\n      const error = new UnsupportedModelError([\"gpt-4o\"], \"extract\");\n\n      expect(error.message).toContain(\"extract\");\n      expect(error.message).toContain(\"provider/model\");\n      expect(error.message).toContain(\n        \"https://docs.stagehand.dev/v3/configuration/models\",\n      );\n    });\n  });\n\n  describe(\"LLMProvider.getClient deprecation warning\", () => {\n    it(\"logs deprecation warning for legacy model names\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Using a legacy model name like \"gpt-4o\" instead of \"openai/gpt-4o\"\n      // Should not throw, but should log a deprecation warning\n      const client = provider.getClient(\"gpt-4o\", mockClientOptions);\n\n      // Should return a client (not throw)\n      expect(client).toBeDefined();\n\n      // Should have logged a deprecation warning at level 0\n      const deprecationWarning = logs.find(\n        (log) =>\n          log.message.toLowerCase().includes(\"deprecated\") ||\n          log.message.toLowerCase().includes(\"deprecation\"),\n      );\n      expect(deprecationWarning).toBeDefined();\n      expect(deprecationWarning!.level).toBe(0);\n    });\n\n    it(\"deprecation warning mentions provider/model format\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      provider.getClient(\"gpt-4o\", mockClientOptions);\n\n      const deprecationWarning = logs.find(\n        (log) =>\n          log.message.toLowerCase().includes(\"deprecated\") ||\n          log.message.toLowerCase().includes(\"deprecation\"),\n      );\n\n      expect(deprecationWarning).toBeDefined();\n      const message = deprecationWarning!.message;\n      // Should mention the provider/model format\n      expect(message).toContain(\"provider/model\");\n      // Should give an example\n      expect(message).toContain(\"openai/gpt-5\");\n    });\n\n    it(\"returns OpenAIClient for legacy OpenAI model names\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      const client = provider.getClient(\"gpt-4o\", mockClientOptions);\n\n      // Should return a client\n      expect(client).toBeDefined();\n      // The client should be an OpenAIClient (check constructor name)\n      expect(client.constructor.name).toBe(\"OpenAIClient\");\n    });\n\n    it(\"returns GoogleClient for legacy Google model names\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      const client = provider.getClient(\"gemini-2.0-flash\", mockClientOptions);\n\n      // Should return a client\n      expect(client).toBeDefined();\n      // The client should be a GoogleClient\n      expect(client.constructor.name).toBe(\"GoogleClient\");\n    });\n  });\n\n  describe(\"LLMProvider.getClient error handling\", () => {\n    it(\"throws UnsupportedModelError for unknown model without slash\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Unknown model without slash should throw UnsupportedModelError\n      expect(() => {\n        provider.getClient(\"some-unknown-model\", mockClientOptions);\n      }).toThrow(UnsupportedModelError);\n    });\n\n    it(\"UnsupportedModelError includes provider/model format guidance\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      try {\n        provider.getClient(\"some-unknown-model\", mockClientOptions);\n      } catch (error) {\n        expect((error as Error).message).toContain(\"provider/model\");\n      }\n    });\n\n    it(\"throws UnsupportedAISDKModelProviderError for invalid provider in provider/model format\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Invalid provider but correct format\n      expect(() => {\n        provider.getClient(\"invalid-provider/some-model\", mockClientOptions);\n      }).toThrow(UnsupportedAISDKModelProviderError);\n    });\n\n    it(\"UnsupportedAISDKModelProviderError lists valid providers\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      try {\n        provider.getClient(\"invalid-provider/some-model\", mockClientOptions);\n      } catch (error) {\n        const message = (error as Error).message;\n        // Should list valid providers\n        expect(message).toContain(\"openai\");\n        expect(message).toContain(\"anthropic\");\n        expect(message).toContain(\"google\");\n      }\n    });\n  });\n\n  describe(\"new provider/model format\", () => {\n    it(\"does not log deprecation warning for provider/model format\", () => {\n      const logs: LogLine[] = [];\n      const logger = (line: LogLine) => logs.push(line);\n      const provider = new LLMProvider(logger);\n\n      // Using the new format\n      const client = provider.getClient(\"openai/gpt-4o\", mockClientOptions);\n\n      expect(client).toBeDefined();\n\n      // Should NOT have a deprecation warning\n      const deprecationWarning = logs.find(\n        (log) =>\n          log.message.toLowerCase().includes(\"deprecated\") ||\n          log.message.toLowerCase().includes(\"deprecation\"),\n      );\n      expect(deprecationWarning).toBeUndefined();\n    });\n  });\n});\n"]}

package/dist/esm/tests/unit/openai-cua-client.test.d.ts

@@ -0,0 +1 @@
+export {};