feat(sdk-js): add bigscience/bloomz-7b1

different-ai · Jun 27, 2023 · 672344d · 672344d · vercel · Jun 27, 2023
1 parent 4b6c309
commit 672344d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdk/embedbase-js/src/types.ts b/sdk/embedbase-js/src/types.ts
@@ -68,7 +68,7 @@ export interface ClientDatasets {
  public: boolean
 }
 
-export type LLM = 'openai/gpt-4' | 'openai/gpt-3.5-turbo-16k' | 'google/bison' // | 'tiiuae/falcon-7b' | 'bigscience/bloomz-7b1'
+export type LLM = 'openai/gpt-4' | 'openai/gpt-3.5-turbo-16k' | 'google/bison' | 'bigscience/bloomz-7b1' //| 'tiiuae/falcon-7b'
 
 export interface LLMDescription {
  name: LLM

diff --git a/sdk/embedbase-js/test/client.test.ts b/sdk/embedbase-js/test/client.test.ts
@@ -219,7 +219,7 @@ test('should be able to filter by metadata using where', async () => {
 }, TIMEOUT)
 
 test('should be able to chat', async () => {
- for await (const res of embedbase.useModel('openai/gpt-3.5-turbo-16k').streamText('tell a story', {
+ for await (const res of embedbase.useModel('bigscience/bloomz-7b1').streamText('1+1=', {
  // maxNewTokens: 10,
  })) {
  expect(res).toBeDefined()
@@ -390,7 +390,7 @@ test('should provide clear error message', async () => {
 }, TIMEOUT)
 
 test('should be able to generate text sync', async () => {
- const res = await embedbase.useModel('tiiuae/falcon-7b').generateText('1+1=', {
+ const res = await embedbase.useModel('bigscience/bloomz-7b1').generateText('1+1=', {
  maxNewTokens: 10,
  })
  expect(res).toBeDefined()