openai[patch]: fix #5520 (#5521)

* langchain-openai[patch]: fix #5520

* fix: don't rewrite config if they are already set in embedding and chat_model

* Format

---------

Co-authored-by: jeasonnow <[email protected]>
Co-authored-by: jacoblee93 <[email protected]>

Author: 3 people

libs/langchain-openai/src/azure/chat_models.ts

@@ -38,9 +38,13 @@ export class AzureChatOpenAI extends ChatOpenAI {
   ) {
     const newFields = fields ? { ...fields } : fields;
     if (newFields) {
-      newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
-      newFields.azureOpenAIApiKey = newFields.openAIApiKey;
-      newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
+      // don't rewrite the fields if they are already set
+      newFields.azureOpenAIApiDeploymentName =
+        newFields.azureOpenAIApiDeploymentName ?? newFields.deploymentName;
+      newFields.azureOpenAIApiKey =
+        newFields.azureOpenAIApiKey ?? newFields.openAIApiKey;
+      newFields.azureOpenAIApiVersion =
+        newFields.azureOpenAIApiVersion ?? newFields.openAIApiVersion;
     }
 
     super(newFields);

libs/langchain-openai/src/azure/embeddings.ts

@@ -27,9 +27,13 @@ export class AzureOpenAIEmbeddings extends OpenAIEmbeddings {
   ) {
     const newFields = { ...fields };
     if (Object.entries(newFields).length) {
-      newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
-      newFields.azureOpenAIApiKey = newFields.apiKey;
-      newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
+      // don't rewrite the fields if they are already set
+      newFields.azureOpenAIApiDeploymentName =
+        newFields.azureOpenAIApiDeploymentName ?? newFields.deploymentName;
+      newFields.azureOpenAIApiKey =
+        newFields.azureOpenAIApiKey ?? newFields.apiKey;
+      newFields.azureOpenAIApiVersion =
+        newFields.azureOpenAIApiVersion ?? newFields.openAIApiVersion;
     }
 
     super(newFields, configuration);

libs/langchain-openai/src/azure/llms.ts

@@ -31,9 +31,13 @@ export class AzureOpenAI extends OpenAI {
   ) {
     const newFields = fields ? { ...fields } : fields;
     if (newFields) {
-      newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
-      newFields.azureOpenAIApiKey = newFields.openAIApiKey;
-      newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
+      // don't rewrite the fields if they are already set
+      newFields.azureOpenAIApiDeploymentName =
+        newFields.azureOpenAIApiDeploymentName ?? newFields.deploymentName;
+      newFields.azureOpenAIApiKey =
+        newFields.azureOpenAIApiKey ?? newFields.openAIApiKey;
+      newFields.azureOpenAIApiVersion =
+        newFields.azureOpenAIApiVersion ?? newFields.openAIApiVersion;
     }
 
     super(newFields);

libs/langchain-openai/src/utils/azure.ts

@@ -40,7 +40,7 @@ export function getEndpoint(config: OpenAIEndpointConfig) {
   } = config;
 
   if (
-    azureOpenAIApiKey &&
+    (azureOpenAIApiKey || azureADTokenProvider) &&
     azureOpenAIBasePath &&
     azureOpenAIApiDeploymentName
   ) {