|
|
@@ -138,6 +138,14 @@ function useLanguageModel(sdk: any) {
|
|
|
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> {
|
|
|
return {
|
|
|
anthropic: () =>
|
|
|
@@ -222,12 +230,7 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
|
|
|
return {
|
|
|
autoload: false,
|
|
|
async getModel(sdk: any, modelID: string, options?: Record<string, any>) {
|
|
|
- if (useLanguageModel(sdk)) return sdk.languageModel(modelID)
|
|
|
- if (options?.["useCompletionUrls"]) {
|
|
|
- return sdk.chat(modelID)
|
|
|
- } else {
|
|
|
- return sdk.responses(modelID)
|
|
|
- }
|
|
|
+ return selectAzureLanguageModel(sdk, modelID, Boolean(options?.["useCompletionUrls"]))
|
|
|
},
|
|
|
options: {
|
|
|
resourceName: resource,
|
|
|
@@ -247,12 +250,7 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
|
|
|
return {
|
|
|
autoload: false,
|
|
|
async getModel(sdk: any, modelID: string, options?: Record<string, any>) {
|
|
|
- if (useLanguageModel(sdk)) return sdk.languageModel(modelID)
|
|
|
- if (options?.["useCompletionUrls"]) {
|
|
|
- return sdk.chat(modelID)
|
|
|
- } else {
|
|
|
- return sdk.responses(modelID)
|
|
|
- }
|
|
|
+ return selectAzureLanguageModel(sdk, modelID, Boolean(options?.["useCompletionUrls"]))
|
|
|
},
|
|
|
options: {
|
|
|
baseURL: resourceName ? `https://${resourceName}.cognitiveservices.azure.com/openai` : undefined,
|