Quick Fix for Respones API Only (for Hannes)

DangerMouseUK · DangerMouseUK · commit 64f1c80aa3fb · 2026-02-08T21:44:15.000Z
diff --git a/src/api/providers/__tests__/azure.spec.ts b/src/api/providers/__tests__/azure.spec.ts
@@ -3,11 +3,16 @@ const { mockStreamText, mockGenerateText, mockCreateAzure } = vi.hoisted(() => (
 	mockStreamText: vi.fn(),
 	mockGenerateText: vi.fn(),
 	mockCreateAzure: vi.fn(() => {
-		// Return a function that returns a mock language model
-		return vi.fn(() => ({
+		// Return a provider function that supports Responses API model creation
+		const mockProvider = vi.fn(() => ({
 			modelId: "gpt-4o",
 			provider: "azure",
 		}))
+		;(mockProvider as any).responses = vi.fn(() => ({
+			modelId: "gpt-4o",
+			provider: "azure.responses",
+		}))
+		return mockProvider
 	}),
 }))
 
@@ -139,6 +144,27 @@ describe("AzureHandler", () => {
 			},
 		]
 
+		it("should use the Responses API language model", async () => {
+			async function* mockFullStream() {
+				yield { type: "text-delta", text: "Test response" }
+			}
+
+			mockStreamText.mockReturnValue({
+				fullStream: mockFullStream(),
+				usage: Promise.resolve({ inputTokens: 1, outputTokens: 1 }),
+				providerMetadata: Promise.resolve({}),
+			})
+
+			const stream = handler.createMessage(systemPrompt, messages)
+			for await (const _chunk of stream) {
+				// exhaust stream
+			}
+
+			expect(mockStreamText).toHaveBeenCalled()
+			const requestOptions = mockStreamText.mock.calls[0][0]
+			expect((requestOptions.model as any).provider).toBe("azure.responses")
+		})
+
 		it("should handle streaming responses", async () => {
 			// Mock the fullStream async generator
 			async function* mockFullStream() {
diff --git a/src/api/providers/azure.ts b/src/api/providers/azure.ts
@@ -72,10 +72,11 @@ export class AzureHandler extends BaseProvider implements SingleCompletionHandle
 
 	/**
 	 * Get the language model for the configured deployment name.
+	 * Azure provider is wired to use the Responses API endpoint.
 	 */
 	protected getLanguageModel() {
 		const { id } = this.getModel()
-		return this.provider(id)
+		return this.provider.responses(id)
 	}
 
 	/**
