Update

langchain-ai · Jan 25, 2025 · c21b9a6 · c21b9a6
1 parent 21bf2bf
commit c21b9a6
Show file tree

Hide file tree

Showing 10 changed files with 157 additions and 231 deletions.
diff --git a/examples/src/chains/multi_prompt.ts b/examples/src/chains/multi_prompt.ts
diff --git a/examples/src/chains/multi_retrieval_qa.ts b/examples/src/chains/multi_retrieval_qa.ts
diff --git a/examples/src/llms/openai-chat.ts b/examples/src/llms/openai-chat.ts
diff --git a/langchain/langchain.config.js b/langchain/langchain.config.js
@@ -225,16 +225,31 @@ export const config = {
       alias: ["chat_models", "openai"],
       path: "@langchain/openai",
     },
+    {
+      modules: ["AzureChatOpenAI"],
+      alias: ["chat_models", "azure_openai"],
+      path: "@langchain/openai",
+    },
     {
       modules: ["OpenAI"],
       alias: ["llms", "openai"],
       path: "@langchain/openai",
     },
+    {
+      modules: ["AzureOpenAI"],
+      alias: ["llms", "azure_openai"],
+      path: "@langchain/openai",
+    },
     {
       modules: ["OpenAIEmbeddings"],
       alias: ["embeddings", "openai"],
       path: "@langchain/openai",
     },
+    {
+      modules: ["AzureOpenAIEmbeddings"],
+      alias: ["embeddings", "azure_openai"],
+      path: "@langchain/openai",
+    },
     {
       modules: ["PromptTemplate"],
       alias: ["prompts", "prompt"],

diff --git a/langchain/src/chains/router/tests/multi_prompt.int.test.ts b/langchain/src/chains/router/tests/multi_prompt.int.test.ts
diff --git a/langchain/src/chains/router/tests/multi_retrieval_qa.int.test.ts b/langchain/src/chains/router/tests/multi_retrieval_qa.int.test.ts
@@ -1,5 +1,5 @@
 import { test } from "@jest/globals";
-import { OpenAIEmbeddings, OpenAIChat } from "@langchain/openai";
+import { OpenAIEmbeddings, ChatOpenAI } from "@langchain/openai";
 import { ConsoleCallbackHandler } from "@langchain/core/tracers/console";
 import { MultiRetrievalQAChain } from "../multi_retrieval_qa.js";
 import { MemoryVectorStore } from "../../../vectorstores/memory.js";
@@ -44,7 +44,7 @@ test("Test MultiPromptChain", async () => {
     embeddings
   );
 
-  const llm = new OpenAIChat({
+  const llm = new ChatOpenAI({
     callbacks: [new ConsoleCallbackHandler()],
   });
 

diff --git a/langchain/src/load/import_map.ts b/langchain/src/load/import_map.ts
@@ -68,8 +68,11 @@ export * as schema__query_constructor from "../schema/query_constructor.js";
 export * as schema__prompt_template from "../schema/prompt_template.js";
 import {
   ChatOpenAI,
+  AzureChatOpenAI,
   OpenAI,
-  OpenAIEmbeddings
+  AzureOpenAI,
+  OpenAIEmbeddings,
+  AzureOpenAIEmbeddings
 } from "@langchain/openai";
 import {
   PromptTemplate,
@@ -127,14 +130,26 @@ const chat_models__openai = {
   ChatOpenAI
 };
 export { chat_models__openai };
+const chat_models__azure_openai = {
+  AzureChatOpenAI
+};
+export { chat_models__azure_openai };
 const llms__openai = {
   OpenAI
 };
 export { llms__openai };
+const llms__azure_openai = {
+  AzureOpenAI
+};
+export { llms__azure_openai };
 const embeddings__openai = {
   OpenAIEmbeddings
 };
 export { embeddings__openai };
+const embeddings__azure_openai = {
+  AzureOpenAIEmbeddings
+};
+export { embeddings__azure_openai };
 const prompts__prompt = {
   PromptTemplate
 };