@@ -3,6 +3,7 @@ import type { Model } from "../llm/types.js";
|
3 | 3 | |
4 | 4 | const mocks = vi.hoisted(() => ({ |
5 | 5 | ensureCustomApiRegistered: vi.fn(), |
| 6 | +createBoundaryAwareStreamFnForModel: vi.fn(), |
6 | 7 | createTransportAwareStreamFnForModel: vi.fn(), |
7 | 8 | resolveProviderStreamFn: vi.fn(), |
8 | 9 | wrapProviderStreamFn: vi.fn(), |
@@ -16,6 +17,7 @@ vi.mock("./custom-api-registry.js", () => ({
|
16 | 17 | ensureCustomApiRegistered: mocks.ensureCustomApiRegistered, |
17 | 18 | })); |
18 | 19 | vi.mock("./provider-transport-stream.js", () => ({ |
| 20 | +createBoundaryAwareStreamFnForModel: mocks.createBoundaryAwareStreamFnForModel, |
19 | 21 | createTransportAwareStreamFnForModel: mocks.createTransportAwareStreamFnForModel, |
20 | 22 | })); |
21 | 23 | |
@@ -39,10 +41,10 @@ describe("registerProviderStreamForModel", () => {
|
39 | 41 | vi.clearAllMocks(); |
40 | 42 | }); |
41 | 43 | |
42 | | -it("applies an opted-in provider wrapper after selecting the managed transport", () => { |
| 44 | +it("applies an opted-in provider wrapper after selecting the boundary transport", () => { |
43 | 45 | const baseStream = vi.fn(); |
44 | 46 | const wrappedStream = vi.fn(); |
45 | | -mocks.createTransportAwareStreamFnForModel.mockReturnValue(baseStream); |
| 47 | +mocks.createBoundaryAwareStreamFnForModel.mockReturnValue(baseStream); |
46 | 48 | mocks.wrapProviderStreamFn.mockReturnValue(wrappedStream); |
47 | 49 | |
48 | 50 | expect( |
@@ -54,6 +56,12 @@ describe("registerProviderStreamForModel", () => {
|
54 | 56 | applyProviderWrapper: true, |
55 | 57 | }), |
56 | 58 | ).toBe(wrappedStream); |
| 59 | +expect(mocks.createBoundaryAwareStreamFnForModel).toHaveBeenCalledWith(MODEL, { |
| 60 | +cfg: { models: {} }, |
| 61 | +agentDir: "/agent", |
| 62 | +workspaceDir: "/workspace", |
| 63 | +env: undefined, |
| 64 | +}); |
57 | 65 | expect(mocks.wrapProviderStreamFn).toHaveBeenCalledWith({ |
58 | 66 | provider: "clawrouter", |
59 | 67 | config: { models: {} }, |
|
