feat: schema updates (#104)

IBM · May 28, 2024 · b89924a · b89924a
1 parent a8b3b5c
commit b89924a
Show file tree

Hide file tree

Showing 15 changed files with 5,048 additions and 2,969 deletions.
diff --git a/README.md b/README.md
@@ -203,7 +203,7 @@ import { GenAIChatModel } from '@ibm-generative-ai/node-sdk/langchain';
 import { SystemMessage, HumanMessage } from '@langchain/core/messages';
 
 const client = new GenAIChatModel({
-  model_id: 'meta-llama/llama-2-70b-chat',
+  model_id: 'meta-llama/llama-3-70b-instruct',
   client: new Client({
     endpoint: process.env.ENDPOINT,
     apiKey: process.env.API_KEY,

diff --git a/examples/langchain/llm-chat.ts b/examples/langchain/llm-chat.ts
@@ -5,7 +5,7 @@ import { Client } from '../../src/index.js';
 
 const makeClient = () =>
   new GenAIChatModel({
-    model_id: 'meta-llama/llama-2-70b-chat',
+    model_id: 'meta-llama/llama-3-70b-instruct',
     client: new Client({
       endpoint: process.env.ENDPOINT,
       apiKey: process.env.API_KEY,

diff --git a/examples/langchain/llm.ts b/examples/langchain/llm.ts
@@ -15,7 +15,16 @@ const makeClient = () =>
       repetition_penalty: 1.5,
     },
     moderations: {
-      hap: true,
+      hap: {
+        input: {
+          enabled: true,
+          threshold: 0.75,
+        },
+        output: {
+          enabled: true,
+          threshold: 0.75,
+        },
+      },
     },
   });
 

diff --git a/examples/shared/constants.ts b/examples/shared/constants.ts
@@ -1,2 +1,2 @@
 export const MODEL = 'google/flan-ul2';
-export const CHAT_MODEL = 'meta-llama/llama-2-70b-chat';
+export const CHAT_MODEL = 'meta-llama/llama-3-70b-instruct';