Implementing last requested changes

Ultrahi · Dec 20, 2023 · 1324031 · 1324031
1 parent 4fede08
commit 1324031
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git 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
@@ -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({