fix(deepseek): add Azure AI Inference path handling for DeepSeek via Azure

roomote · roomote · commit 6da8e4225d84 · 2025-12-15T20:16:44.000Z
- Change _isAzureAiInference and _getUrlHost from private to protected in OpenAiHandler
- Add OPENAI_AZURE_AI_INFERENCE_PATH import and path handling in DeepSeekHandler.createMessage()
- Update test to expect path options argument
diff --git a/src/api/providers/__tests__/deepseek.spec.ts b/src/api/providers/__tests__/deepseek.spec.ts
@@ -455,10 +455,12 @@ describe("DeepSeekHandler", () => {
 			}
 
 			// Verify that the thinking parameter was passed to the API
+			// Note: mockCreate receives two arguments - request options and path options
 			expect(mockCreate).toHaveBeenCalledWith(
 				expect.objectContaining({
 					thinking: { type: "enabled" },
 				}),
+				{}, // Empty path options for non-Azure URLs
 			)
 		})
 
diff --git a/src/api/providers/deepseek.ts b/src/api/providers/deepseek.ts
@@ -1,7 +1,12 @@
 import { Anthropic } from "@anthropic-ai/sdk"
 import OpenAI from "openai"
 
-import { deepSeekModels, deepSeekDefaultModelId, DEEP_SEEK_DEFAULT_TEMPERATURE } from "@roo-code/types"
+import {
+	deepSeekModels,
+	deepSeekDefaultModelId,
+	DEEP_SEEK_DEFAULT_TEMPERATURE,
+	OPENAI_AZURE_AI_INFERENCE_PATH,
+} from "@roo-code/types"
 
 import type { ApiHandlerOptions } from "../../shared/api"
 
@@ -84,9 +89,15 @@ export class DeepSeekHandler extends OpenAiHandler {
 		// Add max_tokens if needed
 		this.addMaxTokensIfNeeded(requestOptions, modelInfo)
 
+		// Check if base URL is Azure AI Inference (for DeepSeek via Azure)
+		const isAzureAiInference = this._isAzureAiInference(this.options.deepSeekBaseUrl)
+
 		let stream
 		try {
-			stream = await this.client.chat.completions.create(requestOptions)
+			stream = await this.client.chat.completions.create(
+				requestOptions,
+				isAzureAiInference ? { path: OPENAI_AZURE_AI_INFERENCE_PATH } : {},
+			)
 		} catch (error) {
 			const { handleOpenAIError } = await import("./utils/openai-error-handler")
 			throw handleOpenAIError(error, "DeepSeek")
diff --git a/src/api/providers/openai.ts b/src/api/providers/openai.ts
@@ -478,7 +478,7 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl
 		}
 	}
 
-	private _getUrlHost(baseUrl?: string): string {
+	protected _getUrlHost(baseUrl?: string): string {
 		try {
 			return new URL(baseUrl ?? "").host
 		} catch (error) {
@@ -491,7 +491,7 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl
 		return urlHost.includes("x.ai")
 	}
 
-	private _isAzureAiInference(baseUrl?: string): boolean {
+	protected _isAzureAiInference(baseUrl?: string): boolean {
 		const urlHost = this._getUrlHost(baseUrl)
 		return urlHost.endsWith(".services.ai.azure.com")
 	}

Original file line number	Diff line number	Diff line change
`@@ -455,10 +455,12 @@ describe("DeepSeekHandler", () => {`
`455`	`455`	`}`
`456`	`456`
`457`	`457`	`// Verify that the thinking parameter was passed to the API`
	`458`	`+ // Note: mockCreate receives two arguments - request options and path options`
`458`	`459`	`expect(mockCreate).toHaveBeenCalledWith(`
`459`	`460`	`expect.objectContaining({`
`460`	`461`	`thinking: { type: "enabled" },`
`461`	`462`	`}),`
	`463`	`+ {}, // Empty path options for non-Azure URLs`
`462`	`464`	`)`
`463`	`465`	`})`
`464`	`466`
Original file line number	Diff line number	Diff line change
`@@ -478,7 +478,7 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl`
`478`	`478`	`}`
`479`	`479`	`}`
`480`	`480`
`481`		`- private _getUrlHost(baseUrl?: string): string {`
	`481`	`+ protected _getUrlHost(baseUrl?: string): string {`
`482`	`482`	`try {`
`483`	`483`	`return new URL(baseUrl ?? "").host`
`484`	`484`	`} catch (error) {`
`@@ -491,7 +491,7 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl`
`491`	`491`	`return urlHost.includes("x.ai")`
`492`	`492`	`}`
`493`	`493`
`494`		`- private _isAzureAiInference(baseUrl?: string): boolean {`
	`494`	`+ protected _isAzureAiInference(baseUrl?: string): boolean {`
`495`	`495`	`const urlHost = this._getUrlHost(baseUrl)`
`496`	`496`	`return urlHost.endsWith(".services.ai.azure.com")`
`497`	`497`	`}`