@langchain/openai 0.0.28 → 0.0.29

package/dist/azure/chat_models.cjs

@@ -1,7 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AzureChatOpenAI = void 0;
+const openai_1 = require("openai");
 const chat_models_js_1 = require("../chat_models.cjs");
+const azure_js_1 = require("../utils/azure.cjs");
 class AzureChatOpenAI extends chat_models_js_1.ChatOpenAI {
     _llmType() {
         return "azure_openai";
@@ -14,21 +16,59 @@ class AzureChatOpenAI extends chat_models_js_1.ChatOpenAI {
         };
     }
     constructor(fields) {
-        // assume the base URL does not contain "openai" nor "deployments" prefix
-        let basePath = fields?.openAIBasePath ?? "";
-        if (!basePath.endsWith("/"))
-            basePath += "/";
-        if (!basePath.endsWith("openai/deployments"))
-            basePath += "openai/deployments";
         const newFields = fields ? { ...fields } : fields;
         if (newFields) {
-            newFields.azureOpenAIBasePath = basePath;
             newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
             newFields.azureOpenAIApiKey = newFields.openAIApiKey;
             newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
         }
         super(newFields);
     }
+    _getClientOptions(options) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                baseURL: this.clientConfig.baseURL,
+            };
+            const endpoint = (0, azure_js_1.getEndpoint)(openAIEndpointConfig);
+            const params = {
+                ...this.clientConfig,
+                baseURL: endpoint,
+                timeout: this.timeout,
+                maxRetries: 0,
+            };
+            if (!this.azureADTokenProvider) {
+                params.apiKey = openAIEndpointConfig.azureOpenAIApiKey;
+            }
+            if (!params.baseURL) {
+                delete params.baseURL;
+            }
+            this.client = new openai_1.AzureOpenAI({
+                apiVersion: this.azureOpenAIApiVersion,
+                azureADTokenProvider: this.azureADTokenProvider,
+                deployment: this.azureOpenAIApiDeploymentName,
+                ...params,
+            });
+        }
+        const requestOptions = {
+            ...this.clientConfig,
+            ...options,
+        };
+        if (this.azureOpenAIApiKey) {
+            requestOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...requestOptions.headers,
+            };
+            requestOptions.query = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...requestOptions.query,
+            };
+        }
+        return requestOptions;
+    }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     toJSON() {
         const json = super.toJSON();

package/dist/azure/chat_models.d.ts

@@ -1,7 +1,7 @@
 import { type ClientOptions } from "openai";
 import { type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
 import { ChatOpenAI } from "../chat_models.js";
-import { AzureOpenAIInput, LegacyOpenAIInput, OpenAIChatInput } from "../types.js";
+import { AzureOpenAIInput, LegacyOpenAIInput, OpenAIChatInput, OpenAICoreRequestOptions } from "../types.js";
 export declare class AzureChatOpenAI extends ChatOpenAI {
     _llmType(): string;
     get lc_aliases(): Record<string, string>;
@@ -13,5 +13,6 @@ export declare class AzureChatOpenAI extends ChatOpenAI {
     } & BaseChatModelParams & {
         configuration?: ClientOptions & LegacyOpenAIInput;
     });
+    protected _getClientOptions(options: OpenAICoreRequestOptions | undefined): OpenAICoreRequestOptions;
     toJSON(): any;
 }

package/dist/azure/chat_models.js

@@ -1,4 +1,6 @@
+import { AzureOpenAI as AzureOpenAIClient } from "openai";
 import { ChatOpenAI } from "../chat_models.js";
+import { getEndpoint } from "../utils/azure.js";
 export class AzureChatOpenAI extends ChatOpenAI {
     _llmType() {
         return "azure_openai";
@@ -11,21 +13,59 @@ export class AzureChatOpenAI extends ChatOpenAI {
         };
     }
     constructor(fields) {
-        // assume the base URL does not contain "openai" nor "deployments" prefix
-        let basePath = fields?.openAIBasePath ?? "";
-        if (!basePath.endsWith("/"))
-            basePath += "/";
-        if (!basePath.endsWith("openai/deployments"))
-            basePath += "openai/deployments";
         const newFields = fields ? { ...fields } : fields;
         if (newFields) {
-            newFields.azureOpenAIBasePath = basePath;
             newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
             newFields.azureOpenAIApiKey = newFields.openAIApiKey;
             newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
         }
         super(newFields);
     }
+    _getClientOptions(options) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                baseURL: this.clientConfig.baseURL,
+            };
+            const endpoint = getEndpoint(openAIEndpointConfig);
+            const params = {
+                ...this.clientConfig,
+                baseURL: endpoint,
+                timeout: this.timeout,
+                maxRetries: 0,
+            };
+            if (!this.azureADTokenProvider) {
+                params.apiKey = openAIEndpointConfig.azureOpenAIApiKey;
+            }
+            if (!params.baseURL) {
+                delete params.baseURL;
+            }
+            this.client = new AzureOpenAIClient({
+                apiVersion: this.azureOpenAIApiVersion,
+                azureADTokenProvider: this.azureADTokenProvider,
+                deployment: this.azureOpenAIApiDeploymentName,
+                ...params,
+            });
+        }
+        const requestOptions = {
+            ...this.clientConfig,
+            ...options,
+        };
+        if (this.azureOpenAIApiKey) {
+            requestOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...requestOptions.headers,
+            };
+            requestOptions.query = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...requestOptions.query,
+            };
+        }
+        return requestOptions;
+    }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     toJSON() {
         const json = super.toJSON();

package/dist/azure/embeddings.cjs

@@ -0,0 +1,70 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AzureOpenAIEmbeddings = void 0;
+const openai_1 = require("openai");
+const embeddings_js_1 = require("../embeddings.cjs");
+const azure_js_1 = require("../utils/azure.cjs");
+const openai_js_1 = require("../utils/openai.cjs");
+class AzureOpenAIEmbeddings extends embeddings_js_1.OpenAIEmbeddings {
+    constructor(fields, configuration) {
+        const newFields = { ...fields };
+        if (Object.entries(newFields).length) {
+            newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
+            newFields.azureOpenAIApiKey = newFields.apiKey;
+            newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
+        }
+        super(newFields, configuration);
+    }
+    async embeddingWithRetry(request) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                baseURL: this.clientConfig.baseURL,
+            };
+            const endpoint = (0, azure_js_1.getEndpoint)(openAIEndpointConfig);
+            const params = {
+                ...this.clientConfig,
+                baseURL: endpoint,
+                timeout: this.timeout,
+                maxRetries: 0,
+            };
+            if (!this.azureADTokenProvider) {
+                params.apiKey = openAIEndpointConfig.azureOpenAIApiKey;
+            }
+            if (!params.baseURL) {
+                delete params.baseURL;
+            }
+            this.client = new openai_1.AzureOpenAI({
+                apiVersion: this.azureOpenAIApiVersion,
+                azureADTokenProvider: this.azureADTokenProvider,
+                deployment: this.azureOpenAIApiDeploymentName,
+                ...params,
+            });
+        }
+        const requestOptions = {};
+        if (this.azureOpenAIApiKey) {
+            requestOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...requestOptions.headers,
+            };
+            requestOptions.query = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...requestOptions.query,
+            };
+        }
+        return this.caller.call(async () => {
+            try {
+                const res = await this.client.embeddings.create(request, requestOptions);
+                return res;
+            }
+            catch (e) {
+                const error = (0, openai_js_1.wrapOpenAIClientError)(e);
+                throw error;
+            }
+        });
+    }
+}
+exports.AzureOpenAIEmbeddings = AzureOpenAIEmbeddings;

package/dist/azure/embeddings.d.ts

@@ -0,0 +1,14 @@
+import { type ClientOptions, OpenAI as OpenAIClient } from "openai";
+import { OpenAIEmbeddings, OpenAIEmbeddingsParams } from "../embeddings.js";
+import { AzureOpenAIInput, LegacyOpenAIInput } from "../types.js";
+export declare class AzureOpenAIEmbeddings extends OpenAIEmbeddings {
+    constructor(fields?: Partial<OpenAIEmbeddingsParams> & Partial<AzureOpenAIInput> & {
+        verbose?: boolean;
+        /** The OpenAI API key to use. */
+        apiKey?: string;
+        configuration?: ClientOptions;
+        deploymentName?: string;
+        openAIApiVersion?: string;
+    }, configuration?: ClientOptions & LegacyOpenAIInput);
+    protected embeddingWithRetry(request: OpenAIClient.EmbeddingCreateParams): Promise<OpenAIClient.Embeddings.CreateEmbeddingResponse>;
+}

package/dist/azure/embeddings.js

@@ -0,0 +1,66 @@
+import { AzureOpenAI as AzureOpenAIClient, } from "openai";
+import { OpenAIEmbeddings } from "../embeddings.js";
+import { getEndpoint } from "../utils/azure.js";
+import { wrapOpenAIClientError } from "../utils/openai.js";
+export class AzureOpenAIEmbeddings extends OpenAIEmbeddings {
+    constructor(fields, configuration) {
+        const newFields = { ...fields };
+        if (Object.entries(newFields).length) {
+            newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
+            newFields.azureOpenAIApiKey = newFields.apiKey;
+            newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
+        }
+        super(newFields, configuration);
+    }
+    async embeddingWithRetry(request) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                baseURL: this.clientConfig.baseURL,
+            };
+            const endpoint = getEndpoint(openAIEndpointConfig);
+            const params = {
+                ...this.clientConfig,
+                baseURL: endpoint,
+                timeout: this.timeout,
+                maxRetries: 0,
+            };
+            if (!this.azureADTokenProvider) {
+                params.apiKey = openAIEndpointConfig.azureOpenAIApiKey;
+            }
+            if (!params.baseURL) {
+                delete params.baseURL;
+            }
+            this.client = new AzureOpenAIClient({
+                apiVersion: this.azureOpenAIApiVersion,
+                azureADTokenProvider: this.azureADTokenProvider,
+                deployment: this.azureOpenAIApiDeploymentName,
+                ...params,
+            });
+        }
+        const requestOptions = {};
+        if (this.azureOpenAIApiKey) {
+            requestOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...requestOptions.headers,
+            };
+            requestOptions.query = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...requestOptions.query,
+            };
+        }
+        return this.caller.call(async () => {
+            try {
+                const res = await this.client.embeddings.create(request, requestOptions);
+                return res;
+            }
+            catch (e) {
+                const error = wrapOpenAIClientError(e);
+                throw error;
+            }
+        });
+    }
+}

package/dist/azure/llms.cjs

@@ -1,7 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AzureOpenAI = void 0;
+const openai_1 = require("openai");
 const llms_js_1 = require("../llms.cjs");
+const azure_js_1 = require("../utils/azure.cjs");
 class AzureOpenAI extends llms_js_1.OpenAI {
     get lc_aliases() {
         return {
@@ -11,21 +13,58 @@ class AzureOpenAI extends llms_js_1.OpenAI {
         };
     }
     constructor(fields) {
-        // assume the base URL does not contain "openai" nor "deployments" prefix
-        let basePath = fields?.openAIBasePath ?? "";
-        if (!basePath.endsWith("/"))
-            basePath += "/";
-        if (!basePath.endsWith("openai/deployments"))
-            basePath += "openai/deployments";
         const newFields = fields ? { ...fields } : fields;
         if (newFields) {
-            newFields.azureOpenAIBasePath = basePath;
             newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
             newFields.azureOpenAIApiKey = newFields.openAIApiKey;
             newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
         }
         super(newFields);
     }
+    _getClientOptions(options) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                baseURL: this.clientConfig.baseURL,
+            };
+            const endpoint = (0, azure_js_1.getEndpoint)(openAIEndpointConfig);
+            const params = {
+                ...this.clientConfig,
+                baseURL: endpoint,
+                timeout: this.timeout,
+                maxRetries: 0,
+            };
+            if (!this.azureADTokenProvider) {
+                params.apiKey = openAIEndpointConfig.azureOpenAIApiKey;
+            }
+            if (!params.baseURL) {
+                delete params.baseURL;
+            }
+            this.client = new openai_1.AzureOpenAI({
+                apiVersion: this.azureOpenAIApiVersion,
+                azureADTokenProvider: this.azureADTokenProvider,
+                ...params,
+            });
+        }
+        const requestOptions = {
+            ...this.clientConfig,
+            ...options,
+        };
+        if (this.azureOpenAIApiKey) {
+            requestOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...requestOptions.headers,
+            };
+            requestOptions.query = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...requestOptions.query,
+            };
+        }
+        return requestOptions;
+    }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     toJSON() {
         const json = super.toJSON();

package/dist/azure/llms.d.ts

@@ -1,7 +1,7 @@
 import { type ClientOptions } from "openai";
 import { type BaseLLMParams } from "@langchain/core/language_models/llms";
 import { OpenAI } from "../llms.js";
-import type { OpenAIInput, AzureOpenAIInput, LegacyOpenAIInput } from "../types.js";
+import type { OpenAIInput, AzureOpenAIInput, OpenAICoreRequestOptions, LegacyOpenAIInput } from "../types.js";
 export declare class AzureOpenAI extends OpenAI {
     get lc_aliases(): Record<string, string>;
     constructor(fields?: Partial<OpenAIInput> & {
@@ -12,5 +12,6 @@ export declare class AzureOpenAI extends OpenAI {
     } & Partial<AzureOpenAIInput> & BaseLLMParams & {
         configuration?: ClientOptions & LegacyOpenAIInput;
     });
+    protected _getClientOptions(options: OpenAICoreRequestOptions | undefined): OpenAICoreRequestOptions;
     toJSON(): any;
 }

package/dist/azure/llms.js

@@ -1,4 +1,6 @@
+import { AzureOpenAI as AzureOpenAIClient } from "openai";
 import { OpenAI } from "../llms.js";
+import { getEndpoint } from "../utils/azure.js";
 export class AzureOpenAI extends OpenAI {
     get lc_aliases() {
         return {
@@ -8,21 +10,58 @@ export class AzureOpenAI extends OpenAI {
         };
     }
     constructor(fields) {
-        // assume the base URL does not contain "openai" nor "deployments" prefix
-        let basePath = fields?.openAIBasePath ?? "";
-        if (!basePath.endsWith("/"))
-            basePath += "/";
-        if (!basePath.endsWith("openai/deployments"))
-            basePath += "openai/deployments";
         const newFields = fields ? { ...fields } : fields;
         if (newFields) {
-            newFields.azureOpenAIBasePath = basePath;
             newFields.azureOpenAIApiDeploymentName = newFields.deploymentName;
             newFields.azureOpenAIApiKey = newFields.openAIApiKey;
             newFields.azureOpenAIApiVersion = newFields.openAIApiVersion;
         }
         super(newFields);
     }
+    _getClientOptions(options) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                baseURL: this.clientConfig.baseURL,
+            };
+            const endpoint = getEndpoint(openAIEndpointConfig);
+            const params = {
+                ...this.clientConfig,
+                baseURL: endpoint,
+                timeout: this.timeout,
+                maxRetries: 0,
+            };
+            if (!this.azureADTokenProvider) {
+                params.apiKey = openAIEndpointConfig.azureOpenAIApiKey;
+            }
+            if (!params.baseURL) {
+                delete params.baseURL;
+            }
+            this.client = new AzureOpenAIClient({
+                apiVersion: this.azureOpenAIApiVersion,
+                azureADTokenProvider: this.azureADTokenProvider,
+                ...params,
+            });
+        }
+        const requestOptions = {
+            ...this.clientConfig,
+            ...options,
+        };
+        if (this.azureOpenAIApiKey) {
+            requestOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...requestOptions.headers,
+            };
+            requestOptions.query = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...requestOptions.query,
+            };
+        }
+        return requestOptions;
+    }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     toJSON() {
         const json = super.toJSON();

package/dist/chat_models.cjs

@@ -371,6 +371,12 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "azureADTokenProvider", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "azureOpenAIApiInstanceName", {
             enumerable: true,
             configurable: true,
@@ -415,8 +421,9 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
         this.azureOpenAIApiKey =
             fields?.azureOpenAIApiKey ??
                 (0, env_1.getEnvironmentVariable)("AZURE_OPENAI_API_KEY");
-        if (!this.azureOpenAIApiKey && !this.apiKey) {
-            throw new Error("OpenAI or Azure OpenAI API key not found");
+        this.azureADTokenProvider = fields?.azureADTokenProvider ?? undefined;
+        if (!this.azureOpenAIApiKey && !this.apiKey && !this.azureADTokenProvider) {
+            throw new Error("OpenAI or Azure OpenAI API key or Token Provider not found");
         }
         this.azureOpenAIApiInstanceName =
             fields?.azureOpenAIApiInstanceName ??
@@ -450,7 +457,7 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
         this.stopSequences = this?.stop;
         this.user = fields?.user;
         this.streaming = fields?.streaming ?? false;
-        if (this.azureOpenAIApiKey) {
+        if (this.azureOpenAIApiKey || this.azureADTokenProvider) {
             if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
                 throw new Error("Azure OpenAI API instance name not found");
             }

package/dist/chat_models.d.ts

@@ -91,12 +91,13 @@ export declare class ChatOpenAI<CallOptions extends ChatOpenAICallOptions = Chat
     apiKey?: string;
     azureOpenAIApiVersion?: string;
     azureOpenAIApiKey?: string;
+    azureADTokenProvider?: () => Promise<string>;
     azureOpenAIApiInstanceName?: string;
     azureOpenAIApiDeploymentName?: string;
     azureOpenAIBasePath?: string;
     organization?: string;
-    private client;
-    private clientConfig;
+    protected client: OpenAIClient;
+    protected clientConfig: ClientOptions;
     constructor(fields?: Partial<OpenAIChatInput> & Partial<AzureOpenAIInput> & BaseChatModelParams & {
         configuration?: ClientOptions & LegacyOpenAIInput;
     }, 
@@ -142,7 +143,7 @@ export declare class ChatOpenAI<CallOptions extends ChatOpenAICallOptions = Chat
      */
     completionWithRetry(request: OpenAIClient.Chat.ChatCompletionCreateParamsStreaming, options?: OpenAICoreRequestOptions): Promise<AsyncIterable<OpenAIClient.Chat.Completions.ChatCompletionChunk>>;
     completionWithRetry(request: OpenAIClient.Chat.ChatCompletionCreateParamsNonStreaming, options?: OpenAICoreRequestOptions): Promise<OpenAIClient.Chat.Completions.ChatCompletion>;
-    private _getClientOptions;
+    protected _getClientOptions(options: OpenAICoreRequestOptions | undefined): OpenAICoreRequestOptions;
     _llmType(): string;
     /** @ignore */
     _combineLLMOutput(...llmOutputs: OpenAILLMOutput[]): OpenAILLMOutput;

package/dist/chat_models.js

@@ -367,6 +367,12 @@ export class ChatOpenAI extends BaseChatModel {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "azureADTokenProvider", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "azureOpenAIApiInstanceName", {
             enumerable: true,
             configurable: true,
@@ -411,8 +417,9 @@ export class ChatOpenAI extends BaseChatModel {
         this.azureOpenAIApiKey =
             fields?.azureOpenAIApiKey ??
                 getEnvironmentVariable("AZURE_OPENAI_API_KEY");
-        if (!this.azureOpenAIApiKey && !this.apiKey) {
-            throw new Error("OpenAI or Azure OpenAI API key not found");
+        this.azureADTokenProvider = fields?.azureADTokenProvider ?? undefined;
+        if (!this.azureOpenAIApiKey && !this.apiKey && !this.azureADTokenProvider) {
+            throw new Error("OpenAI or Azure OpenAI API key or Token Provider not found");
         }
         this.azureOpenAIApiInstanceName =
             fields?.azureOpenAIApiInstanceName ??
@@ -446,7 +453,7 @@ export class ChatOpenAI extends BaseChatModel {
         this.stopSequences = this?.stop;
         this.user = fields?.user;
         this.streaming = fields?.streaming ?? false;
-        if (this.azureOpenAIApiKey) {
+        if (this.azureOpenAIApiKey || this.azureADTokenProvider) {
             if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
                 throw new Error("Azure OpenAI API instance name not found");
             }

package/dist/embeddings.cjs

@@ -79,6 +79,12 @@ class OpenAIEmbeddings extends embeddings_1.Embeddings {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "azureADTokenProvider", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "azureOpenAIApiInstanceName", {
             enumerable: true,
             configurable: true,
@@ -120,8 +126,9 @@ class OpenAIEmbeddings extends embeddings_1.Embeddings {
             (0, env_1.getEnvironmentVariable)("OPENAI_API_KEY");
         const azureApiKey = fieldsWithDefaults?.azureOpenAIApiKey ??
             (0, env_1.getEnvironmentVariable)("AZURE_OPENAI_API_KEY");
-        if (!azureApiKey && !apiKey) {
-            throw new Error("OpenAI or Azure OpenAI API key not found");
+        this.azureADTokenProvider = fields?.azureADTokenProvider ?? undefined;
+        if (!azureApiKey && !apiKey && !this.azureADTokenProvider) {
+            throw new Error("OpenAI or Azure OpenAI API key or Token Provider not found");
         }
         const azureApiInstanceName = fieldsWithDefaults?.azureOpenAIApiInstanceName ??
             (0, env_1.getEnvironmentVariable)("AZURE_OPENAI_API_INSTANCE_NAME");
@@ -150,7 +157,7 @@ class OpenAIEmbeddings extends embeddings_1.Embeddings {
         this.azureOpenAIApiKey = azureApiKey;
         this.azureOpenAIApiInstanceName = azureApiInstanceName;
         this.azureOpenAIApiDeploymentName = azureApiDeploymentName;
-        if (this.azureOpenAIApiKey) {
+        if (this.azureOpenAIApiKey || this.azureADTokenProvider) {
             if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
                 throw new Error("Azure OpenAI API instance name not found");
             }

package/dist/embeddings.d.ts

@@ -1,4 +1,4 @@
-import { type ClientOptions } from "openai";
+import { type ClientOptions, OpenAI as OpenAIClient } from "openai";
 import { Embeddings, type EmbeddingsParams } from "@langchain/core/embeddings";
 import { AzureOpenAIInput, LegacyOpenAIInput } from "./types.js";
 /**
@@ -62,12 +62,13 @@ export declare class OpenAIEmbeddings extends Embeddings implements OpenAIEmbedd
     timeout?: number;
     azureOpenAIApiVersion?: string;
     azureOpenAIApiKey?: string;
+    azureADTokenProvider?: () => Promise<string>;
     azureOpenAIApiInstanceName?: string;
     azureOpenAIApiDeploymentName?: string;
     azureOpenAIBasePath?: string;
     organization?: string;
-    private client;
-    private clientConfig;
+    protected client: OpenAIClient;
+    protected clientConfig: ClientOptions;
     constructor(fields?: Partial<OpenAIEmbeddingsParams> & Partial<AzureOpenAIInput> & {
         verbose?: boolean;
         /**
@@ -101,5 +102,5 @@ export declare class OpenAIEmbeddings extends Embeddings implements OpenAIEmbedd
      * @param request Request to send to the OpenAI API.
      * @returns Promise that resolves to the response from the API.
      */
-    private embeddingWithRetry;
+    protected embeddingWithRetry(request: OpenAIClient.EmbeddingCreateParams): Promise<OpenAIClient.Embeddings.CreateEmbeddingResponse>;
 }