diff --git a/langchain/src/chat_models/azure_ml.ts b/langchain/src/chat_models/azure_ml.ts index b8b3254875c6..1e16297916b7 100644 --- a/langchain/src/chat_models/azure_ml.ts +++ b/langchain/src/chat_models/azure_ml.ts @@ -76,7 +76,7 @@ export class AzureMLChatOnlineEndpoint implements AzureMLChatParams { static lc_name() { - return "AzureMLChat"; + return "AzureMLChatOnlineEndpoint"; } static lc_description() { return "A class for interacting with AzureML Chat models."; diff --git a/langchain/src/llms/tests/azure_ml.int.test.ts b/langchain/src/llms/tests/azure_ml.int.test.ts index 4726755b15b9..1d1e935bb857 100644 --- a/langchain/src/llms/tests/azure_ml.int.test.ts +++ b/langchain/src/llms/tests/azure_ml.int.test.ts @@ -6,9 +6,9 @@ import { HFContentFormatter, LlamaContentFormatter, } from "../azure_ml.js"; -*/ -/* LLama Test + +// LLama Test test("Test AzureML LLama Call", async () => { const prompt = "What is the meaning of Foo?"; const model = new AzureMLOnlineEndpoint({ @@ -20,9 +20,9 @@ test("Test AzureML LLama Call", async () => { console.log(res); }); -*/ -/* GPT2 Test + +// GPT2 Test test("Test AzureML GPT2 Call", async () => { const prompt = "What is the meaning of Foo?"; const model = new AzureMLOnlineEndpoint({ @@ -34,9 +34,9 @@ test("Test AzureML GPT2 Call", async () => { console.log(res); }); -*/ -/* HF Test + +// HF Test test("Test AzureML HF Call", async () => { const prompt = "What is the meaning of Foo?"; const model = new AzureMLOnlineEndpoint({ @@ -48,9 +48,9 @@ test("Test AzureML HF Call", async () => { console.log(res); }); -*/ -/* Dolly Test + +// Dolly Test test("Test AzureML Dolly Call", async () => { const prompt = "What is the meaning of Foo?"; const model = new AzureMLOnlineEndpoint({