openlayer 0.1.45 → 0.1.46

package/examples/node_modules/@langchain/openai/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,69 @@ 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;
+            // 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);
     }
+    _getClientOptions(options) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                azureADTokenProvider: this.azureADTokenProvider,
+                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;
+            }
+            params.defaultHeaders = {
+                ...params.defaultHeaders,
+                "User-Agent": params.defaultHeaders?.["User-Agent"]
+                    ? `${params.defaultHeaders["User-Agent"]}: langchainjs-azure-openai-v2`
+                    : `langchainjs-azure-openai-v2`,
+            };
+            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/examples/node_modules/@langchain/openai/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/examples/node_modules/@langchain/openai/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,69 @@ 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;
+            // 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);
     }
+    _getClientOptions(options) {
+        if (!this.client) {
+            const openAIEndpointConfig = {
+                azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
+                azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
+                azureOpenAIApiKey: this.azureOpenAIApiKey,
+                azureOpenAIBasePath: this.azureOpenAIBasePath,
+                azureADTokenProvider: this.azureADTokenProvider,
+                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;
+            }
+            params.defaultHeaders = {
+                ...params.defaultHeaders,
+                "User-Agent": params.defaultHeaders?.["User-Agent"]
+                    ? `${params.defaultHeaders["User-Agent"]}: langchainjs-azure-openai-v2`
+                    : `langchainjs-azure-openai-v2`,
+            };
+            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/examples/node_modules/@langchain/openai/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");
             }
@@ -475,6 +482,17 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             ...fields?.configuration,
         };
     }
+    getLsParams(options) {
+        const params = this.invocationParams(options);
+        return {
+            ls_provider: "openai",
+            ls_model_name: this.model,
+            ls_model_type: "chat",
+            ls_temperature: params.temperature ?? undefined,
+            ls_max_tokens: params.max_tokens ?? undefined,
+            ls_stop: options.stop,
+        };
+    }
     bindTools(tools, kwargs) {
         return this.bind({
             tools: tools.map(function_calling_1.convertToOpenAITool),
@@ -511,6 +529,9 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             tool_choice: options?.tool_choice,
             response_format: options?.response_format,
             seed: options?.seed,
+            ...(options?.stream_options !== undefined
+                ? { stream_options: options.stream_options }
+                : {}),
             ...this.modelKwargs,
         };
         return params;
@@ -532,8 +553,12 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
         };
         let defaultRole;
         const streamIterable = await this.completionWithRetry(params, options);
+        let usage;
         for await (const data of streamIterable) {
             const choice = data?.choices[0];
+            if (data.usage) {
+                usage = data.usage;
+            }
             if (!choice) {
                 continue;
             }
@@ -568,6 +593,20 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
             // eslint-disable-next-line no-void
             void runManager?.handleLLMNewToken(generationChunk.text ?? "", newTokenIndices, undefined, undefined, undefined, { chunk: generationChunk });
         }
+        if (usage) {
+            const generationChunk = new outputs_1.ChatGenerationChunk({
+                message: new messages_1.AIMessageChunk({
+                    content: "",
+                    usage_metadata: {
+                        input_tokens: usage.prompt_tokens,
+                        output_tokens: usage.completion_tokens,
+                        total_tokens: usage.total_tokens,
+                    },
+                }),
+                text: "",
+            });
+            yield generationChunk;
+        }
         if (options.signal?.aborted) {
             throw new Error("AbortError");
         }
@@ -644,6 +683,13 @@ class ChatOpenAI extends chat_models_1.BaseChatModel {
                     ...(part.finish_reason ? { finish_reason: part.finish_reason } : {}),
                     ...(part.logprobs ? { logprobs: part.logprobs } : {}),
                 };
+                if ((0, messages_1.isAIMessage)(generation.message)) {
+                    generation.message.usage_metadata = {
+                        input_tokens: tokenUsage.promptTokens ?? 0,
+                        output_tokens: tokenUsage.completionTokens ?? 0,
+                        total_tokens: tokenUsage.totalTokens ?? 0,
+                    };
+                }
                 generations.push(generation);
             }
             return {

package/examples/node_modules/@langchain/openai/dist/chat_models.d.ts

@@ -3,10 +3,10 @@ import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 import { AIMessageChunk, type BaseMessage } from "@langchain/core/messages";
 import { ChatGenerationChunk, type ChatResult } from "@langchain/core/outputs";
 import { type StructuredToolInterface } from "@langchain/core/tools";
-import { BaseChatModel, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
+import { BaseChatModel, LangSmithParams, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
 import type { BaseFunctionCallOptions, BaseLanguageModelInput, StructuredOutputMethodOptions, StructuredOutputMethodParams } from "@langchain/core/language_models/base";
 import { z } from "zod";
-import { Runnable, RunnableInterface } from "@langchain/core/runnables";
+import { Runnable } from "@langchain/core/runnables";
 import type { AzureOpenAIInput, OpenAICallOptions, OpenAIChatInput, OpenAICoreRequestOptions, LegacyOpenAIInput } from "./types.js";
 export type { AzureOpenAIInput, OpenAICallOptions, OpenAIChatInput };
 interface TokenUsage {
@@ -27,6 +27,9 @@ export interface ChatOpenAICallOptions extends OpenAICallOptions, BaseFunctionCa
         type: "json_object";
     };
     seed?: number;
+    stream_options?: {
+        include_usage: boolean;
+    };
 }
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
@@ -91,18 +94,20 @@ 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;
     }, 
     /** @deprecated */
     configuration?: ClientOptions & LegacyOpenAIInput);
-    bindTools(tools: (Record<string, unknown> | StructuredToolInterface)[], kwargs?: Partial<CallOptions>): RunnableInterface<BaseLanguageModelInput, AIMessageChunk, CallOptions>;
+    getLsParams(options: this["ParsedCallOptions"]): LangSmithParams;
+    bindTools(tools: (Record<string, unknown> | StructuredToolInterface)[], kwargs?: Partial<CallOptions>): Runnable<BaseLanguageModelInput, AIMessageChunk, CallOptions>;
     /**
      * Get the parameters used to invoke the model
      */
@@ -142,7 +147,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/examples/node_modules/@langchain/openai/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");
             }
@@ -471,6 +478,17 @@ export class ChatOpenAI extends BaseChatModel {
             ...fields?.configuration,
         };
     }
+    getLsParams(options) {
+        const params = this.invocationParams(options);
+        return {
+            ls_provider: "openai",
+            ls_model_name: this.model,
+            ls_model_type: "chat",
+            ls_temperature: params.temperature ?? undefined,
+            ls_max_tokens: params.max_tokens ?? undefined,
+            ls_stop: options.stop,
+        };
+    }
     bindTools(tools, kwargs) {
         return this.bind({
             tools: tools.map(convertToOpenAITool),
@@ -507,6 +525,9 @@ export class ChatOpenAI extends BaseChatModel {
             tool_choice: options?.tool_choice,
             response_format: options?.response_format,
             seed: options?.seed,
+            ...(options?.stream_options !== undefined
+                ? { stream_options: options.stream_options }
+                : {}),
             ...this.modelKwargs,
         };
         return params;
@@ -528,8 +549,12 @@ export class ChatOpenAI extends BaseChatModel {
         };
         let defaultRole;
         const streamIterable = await this.completionWithRetry(params, options);
+        let usage;
         for await (const data of streamIterable) {
             const choice = data?.choices[0];
+            if (data.usage) {
+                usage = data.usage;
+            }
             if (!choice) {
                 continue;
             }
@@ -564,6 +589,20 @@ export class ChatOpenAI extends BaseChatModel {
             // eslint-disable-next-line no-void
             void runManager?.handleLLMNewToken(generationChunk.text ?? "", newTokenIndices, undefined, undefined, undefined, { chunk: generationChunk });
         }
+        if (usage) {
+            const generationChunk = new ChatGenerationChunk({
+                message: new AIMessageChunk({
+                    content: "",
+                    usage_metadata: {
+                        input_tokens: usage.prompt_tokens,
+                        output_tokens: usage.completion_tokens,
+                        total_tokens: usage.total_tokens,
+                    },
+                }),
+                text: "",
+            });
+            yield generationChunk;
+        }
         if (options.signal?.aborted) {
             throw new Error("AbortError");
         }
@@ -640,6 +679,13 @@ export class ChatOpenAI extends BaseChatModel {
                     ...(part.finish_reason ? { finish_reason: part.finish_reason } : {}),
                     ...(part.logprobs ? { logprobs: part.logprobs } : {}),
                 };
+                if (isAIMessage(generation.message)) {
+                    generation.message.usage_metadata = {
+                        input_tokens: tokenUsage.promptTokens ?? 0,
+                        output_tokens: tokenUsage.completionTokens ?? 0,
+                        total_tokens: tokenUsage.totalTokens ?? 0,
+                    };
+                }
                 generations.push(generation);
             }
             return {

package/examples/node_modules/@langchain/openai/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/examples/node_modules/@langchain/openai/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>;
 }

package/examples/node_modules/@langchain/openai/dist/embeddings.js

@@ -76,6 +76,12 @@ export class OpenAIEmbeddings extends 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,
@@ -117,8 +123,9 @@ export class OpenAIEmbeddings extends Embeddings {
             getEnvironmentVariable("OPENAI_API_KEY");
         const azureApiKey = fieldsWithDefaults?.azureOpenAIApiKey ??
             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 ??
             getEnvironmentVariable("AZURE_OPENAI_API_INSTANCE_NAME");
@@ -147,7 +154,7 @@ export class OpenAIEmbeddings extends 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/examples/node_modules/@langchain/openai/dist/index.cjs

@@ -22,6 +22,7 @@ __exportStar(require("./chat_models.cjs"), exports);
 __exportStar(require("./azure/chat_models.cjs"), exports);
 __exportStar(require("./llms.cjs"), exports);
 __exportStar(require("./azure/llms.cjs"), exports);
+__exportStar(require("./azure/embeddings.cjs"), exports);
 __exportStar(require("./embeddings.cjs"), exports);
 __exportStar(require("./types.cjs"), exports);
 __exportStar(require("./utils/openai.cjs"), exports);

package/examples/node_modules/@langchain/openai/dist/index.d.ts

@@ -3,6 +3,7 @@ export * from "./chat_models.js";
 export * from "./azure/chat_models.js";
 export * from "./llms.js";
 export * from "./azure/llms.js";
+export * from "./azure/embeddings.js";
 export * from "./embeddings.js";
 export * from "./types.js";
 export * from "./utils/openai.js";

package/examples/node_modules/@langchain/openai/dist/index.js

@@ -3,6 +3,7 @@ export * from "./chat_models.js";
 export * from "./azure/chat_models.js";
 export * from "./llms.js";
 export * from "./azure/llms.js";
+export * from "./azure/embeddings.js";
 export * from "./embeddings.js";
 export * from "./types.js";
 export * from "./utils/openai.js";

package/examples/node_modules/@langchain/openai/dist/llms.cjs

@@ -211,6 +211,12 @@ class OpenAI extends llms_1.BaseLLM {
             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,
@@ -256,8 +262,9 @@ class OpenAI extends llms_1.BaseLLM {
         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 ??
@@ -296,7 +303,7 @@ class OpenAI extends llms_1.BaseLLM {
         if (this.streaming && this.bestOf && this.bestOf > 1) {
             throw new Error("Cannot stream results when bestOf > 1");
         }
-        if (this.azureOpenAIApiKey) {
+        if (this.azureOpenAIApiKey || this.azureADTokenProvider) {
             if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
                 throw new Error("Azure OpenAI API instance name not found");
             }

package/examples/node_modules/@langchain/openai/dist/llms.d.ts

@@ -67,12 +67,13 @@ export declare class OpenAI<CallOptions extends OpenAICallOptions = OpenAICallOp
     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<OpenAIInput> & Partial<AzureOpenAIInput> & BaseLLMParams & {
         configuration?: ClientOptions & LegacyOpenAIInput;
     }, 
@@ -124,6 +125,6 @@ export declare class OpenAI<CallOptions extends OpenAICallOptions = OpenAICallOp
      * @param options Optional configuration for the API call.
      * @returns The response from the OpenAI API.
      */
-    private _getClientOptions;
+    protected _getClientOptions(options: OpenAICoreRequestOptions | undefined): OpenAICoreRequestOptions;
     _llmType(): string;
 }