@langchain/openai 0.0.28 → 0.0.29

package/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/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/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/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/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/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;
 }

package/dist/llms.js

@@ -208,6 +208,12 @@ export class OpenAI extends 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,
@@ -253,8 +259,9 @@ export class OpenAI extends BaseLLM {
         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 ??
@@ -293,7 +300,7 @@ export class OpenAI extends 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/dist/tests/azure/chat_models.int.test.d.ts

@@ -0,0 +1 @@
+export {};