From a56aa6ccbe5b290979a2c3f6438670954da2fa1c Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Sun, 19 Apr 2026 03:10:47 +0100 Subject: [PATCH] test: share model compat streaming fixture --- src/agents/model-compat.test.ts | 39 +++++++++++++-------------------- 1 file changed, 15 insertions(+), 24 deletions(-) diff --git a/src/agents/model-compat.test.ts b/src/agents/model-compat.test.ts index e2aed179fbe..4a6dfb1e986 100644 --- a/src/agents/model-compat.test.ts +++ b/src/agents/model-compat.test.ts @@ -72,6 +72,15 @@ function expectSupportsStrictModeForcedOff(overrides?: Partial>): voi expect(supportsStrictMode(normalized)).toBe(false); } +function expectNativeStreamingSupported(overrides: Partial>): void { + const model = { ...baseModel(), ...overrides }; + delete (model as { compat?: unknown }).compat; + const normalized = normalizeModelCompat(model as Model); + expect(supportsDeveloperRole(normalized)).toBe(false); + expect(supportsUsageInStreaming(normalized)).toBe(true); + expect(supportsStrictMode(normalized)).toBe(false); +} + beforeEach(() => { providerRuntimeMocks.resolveProviderModernModelRef.mockReset(); providerRuntimeMocks.resolveProviderModernModelRef.mockReturnValue(undefined); @@ -170,42 +179,24 @@ describe("normalizeModelCompat", () => { }); it("keeps supportsUsageInStreaming on for native Qwen endpoints", () => { - const model = { - ...baseModel(), + expectNativeStreamingSupported({ provider: "qwen", baseUrl: "https://dashscope-intl.aliyuncs.com/compatible-mode/v1", - }; - delete (model as { compat?: unknown }).compat; - const normalized = normalizeModelCompat(model); - expect(supportsDeveloperRole(normalized)).toBe(false); - expect(supportsUsageInStreaming(normalized)).toBe(true); - expect(supportsStrictMode(normalized)).toBe(false); + }); }); it("keeps supportsUsageInStreaming on for DashScope-compatible endpoints regardless of provider id", () => { - const model = { - ...baseModel(), + expectNativeStreamingSupported({ provider: "custom-qwen", baseUrl: "https://dashscope-intl.aliyuncs.com/compatible-mode/v1", - }; - delete (model as { compat?: unknown }).compat; - const normalized = normalizeModelCompat(model); - expect(supportsDeveloperRole(normalized)).toBe(false); - expect(supportsUsageInStreaming(normalized)).toBe(true); - expect(supportsStrictMode(normalized)).toBe(false); + }); }); it("keeps supportsUsageInStreaming on for Moonshot-native endpoints regardless of provider id", () => { - const model = { - ...baseModel(), + expectNativeStreamingSupported({ provider: "custom-kimi", baseUrl: "https://api.moonshot.ai/v1", - }; - delete (model as { compat?: unknown }).compat; - const normalized = normalizeModelCompat(model); - expect(supportsDeveloperRole(normalized)).toBe(false); - expect(supportsUsageInStreaming(normalized)).toBe(true); - expect(supportsStrictMode(normalized)).toBe(false); + }); }); it("leaves native api.openai.com model untouched", () => {