@@ -138,6 +138,14 @@ function useLanguageModel(sdk: any) {
138138 return sdk . responses === undefined && sdk . chat === undefined
139139}
140140
141+ function selectAzureLanguageModel ( sdk : any , modelID : string , useChat : boolean ) {
142+ if ( useChat && sdk . chat ) return sdk . chat ( modelID )
143+ if ( sdk . responses ) return sdk . responses ( modelID )
144+ if ( sdk . messages ) return sdk . messages ( modelID )
145+ if ( sdk . chat ) return sdk . chat ( modelID )
146+ return sdk . languageModel ( modelID )
147+ }
148+
141149function custom ( dep : CustomDep ) : Record < string , CustomLoader > {
142150 return {
143151 anthropic : ( ) =>
@@ -222,12 +230,7 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
222230 return {
223231 autoload : false ,
224232 async getModel ( sdk : any , modelID : string , options ?: Record < string , any > ) {
225- if ( useLanguageModel ( sdk ) ) return sdk . languageModel ( modelID )
226- if ( options ?. [ "useCompletionUrls" ] ) {
227- return sdk . chat ( modelID )
228- } else {
229- return sdk . responses ( modelID )
230- }
233+ return selectAzureLanguageModel ( sdk , modelID , Boolean ( options ?. [ "useCompletionUrls" ] ) )
231234 } ,
232235 options : {
233236 resourceName : resource ,
@@ -247,12 +250,7 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
247250 return {
248251 autoload : false ,
249252 async getModel ( sdk : any , modelID : string , options ?: Record < string , any > ) {
250- if ( useLanguageModel ( sdk ) ) return sdk . languageModel ( modelID )
251- if ( options ?. [ "useCompletionUrls" ] ) {
252- return sdk . chat ( modelID )
253- } else {
254- return sdk . responses ( modelID )
255- }
253+ return selectAzureLanguageModel ( sdk , modelID , Boolean ( options ?. [ "useCompletionUrls" ] ) )
256254 } ,
257255 options : {
258256 baseURL : resourceName ? `https://${ resourceName } .cognitiveservices.azure.com/openai` : undefined ,
0 commit comments