feat(sdk-js): add palm2 llm

sdk/embedbase-js/src/EmbedbaseClient.ts

@@ -765,6 +765,29 @@ export default class EmbedbaseClient {
     const data: { results: Document[] } = await res.json()
     return data.results
   }
+
+  /**
+   * Execute a prompt using Google PaLM2 model
+   * @param prompt 
+   * @returns 
+   */
+  public async complete(prompt: string): Promise<string> {
+    const url = 'https://llm-usx5gpslaq-uc.a.run.app'
+
+    const res = await fetch(url, {
+      method: 'POST',
+      body: JSON.stringify({
+        prompt: prompt,
+      }),
+      headers: {
+        'Content-Type': 'application/json',
+      },
+    })
+
+    await this.handleError(res);
+    const data: { answer: string } = await res.json()
+    return data.answer
+  }
 }
 
 

sdk/embedbase-js/src/EmbedbaseExperimentalClient.ts

@@ -771,7 +771,27 @@ export default class EmbedbaseExperimentalClient {
     const data: { results: Document[] } = await res.json()
     return data.results
   }
-}
 
+  /**
+   * Execute a prompt using Google PaLM2 model
+   * @param prompt 
+   * @returns 
+   */
+  public async complete(prompt: string): Promise<string> {
+    const url = 'https://llm-usx5gpslaq-uc.a.run.app'
 
+    const res = await fetch(url, {
+      method: 'POST',
+      body: JSON.stringify({
+        prompt: prompt,
+      }),
+      headers: {
+        'Content-Type': 'application/json',
+      },
+    })
 
+    await this.handleError(res);
+    const data: { answer: string } = await res.json()
+    return data.answer
+  }
+}

sdk/embedbase-js/test/client.test.ts

@@ -1,9 +1,9 @@
 import { describe, expect, jest, test } from '@jest/globals';
+import fs from 'fs';
+import glob from 'glob';
+import path from 'path';
 import { createClient, createExperimentalClient } from '../src/index';
 import { stream as originalStream } from '../src/utils';
-import glob from 'glob'
-import fs from 'fs'
-import path from 'path';
 
 try {
   require('dotenv').config({ path: './.env' })
@@ -524,4 +524,10 @@ test('should return a list of strings when using createContext with experimental
 
   expect(data).toBeDefined()
   expect(data).toHaveLength(2)
-}, TIMEOUT)
+}, TIMEOUT)
+
+test('should be able to complete with experimental api', async () => {
+  const embedbase = createExperimentalClient(URL, KEY)
+  const data = await embedbase.complete('the tiger is the biggest cat')
+  expect(data).toBeDefined()
+}, TIMEOUT)