diff --git a/src/api/providers/__tests__/azure.spec.ts b/src/api/providers/__tests__/azure.spec.ts index 1164e40f28..e7e4ed028b 100644 --- a/src/api/providers/__tests__/azure.spec.ts +++ b/src/api/providers/__tests__/azure.spec.ts @@ -91,6 +91,12 @@ describe("AzureHandler", () => { }) }) + describe("isAiSdkProvider", () => { + it("should return true", () => { + expect(handler.isAiSdkProvider()).toBe(true) + }) + }) + describe("createMessage", () => { const systemPrompt = "You are a helpful assistant." const messages: Anthropic.Messages.MessageParam[] = [ diff --git a/src/api/providers/azure.ts b/src/api/providers/azure.ts index 85d1ec5a14..b10d4c78d7 100644 --- a/src/api/providers/azure.ts +++ b/src/api/providers/azure.ts @@ -184,4 +184,8 @@ export class AzureHandler extends BaseProvider implements SingleCompletionHandle return text } + + override isAiSdkProvider(): boolean { + return true + } }