langchain-openai[patch]: fix #5520

langchain-ai · May 23, 2024 · 74ecce6 · 74ecce6
1 parent 4839804
commit 74ecce6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/libs/langchain-openai/src/azure/llms.ts b/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);

diff --git a/libs/langchain-openai/src/utils/azure.ts b/libs/langchain-openai/src/utils/azure.ts
@@ -40,7 +40,7 @@ export function getEndpoint(config: OpenAIEndpointConfig) {
   } = config;
 
   if (
-    azureOpenAIApiKey &&
+    (azureOpenAIApiKey || azureADTokenProvider) &&
     azureOpenAIBasePath &&
     azureOpenAIApiDeploymentName
   ) {