fix: ensure anthropic sdk properly resolves when using azure (#25721)

This commit is contained in:
Aiden Cline
2026-05-04 09:58:21 -05:00
committed by GitHub
parent 2c819f290f
commit c1f607d206
+10 -12
View File
@@ -138,6 +138,14 @@ function useLanguageModel(sdk: any) {
return sdk.responses === undefined && sdk.chat === undefined return sdk.responses === undefined && sdk.chat === undefined
} }
function selectAzureLanguageModel(sdk: any, modelID: string, useChat: boolean) {
if (useChat && sdk.chat) return sdk.chat(modelID)
if (sdk.responses) return sdk.responses(modelID)
if (sdk.messages) return sdk.messages(modelID)
if (sdk.chat) return sdk.chat(modelID)
return sdk.languageModel(modelID)
}
function custom(dep: CustomDep): Record<string, CustomLoader> { function custom(dep: CustomDep): Record<string, CustomLoader> {
return { return {
anthropic: () => anthropic: () =>
@@ -222,12 +230,7 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
return { return {
autoload: false, autoload: false,
async getModel(sdk: any, modelID: string, options?: Record<string, any>) { async getModel(sdk: any, modelID: string, options?: Record<string, any>) {
if (useLanguageModel(sdk)) return sdk.languageModel(modelID) return selectAzureLanguageModel(sdk, modelID, Boolean(options?.["useCompletionUrls"]))
if (options?.["useCompletionUrls"]) {
return sdk.chat(modelID)
} else {
return sdk.responses(modelID)
}
}, },
options: { options: {
resourceName: resource, resourceName: resource,
@@ -247,12 +250,7 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
return { return {
autoload: false, autoload: false,
async getModel(sdk: any, modelID: string, options?: Record<string, any>) { async getModel(sdk: any, modelID: string, options?: Record<string, any>) {
if (useLanguageModel(sdk)) return sdk.languageModel(modelID) return selectAzureLanguageModel(sdk, modelID, Boolean(options?.["useCompletionUrls"]))
if (options?.["useCompletionUrls"]) {
return sdk.chat(modelID)
} else {
return sdk.responses(modelID)
}
}, },
options: { options: {
baseURL: resourceName ? `https://${resourceName}.cognitiveservices.azure.com/openai` : undefined, baseURL: resourceName ? `https://${resourceName}.cognitiveservices.azure.com/openai` : undefined,