feat(*): use genkit where possible (#547)

* Release firestore-vector-search, firestore-genai-chatbot, firestore-multimodel-genai,storage-reverse-image-search,firestore-semantic-search (#540) * fix(*): singapore location value (#528) * fix: update default model name to `gemini-1.0-pro` (#509) * fix: bump deps, fix backfill (#539) * chore(*): bump extension versions --------- Co-authored-by: Mais Alheraki <[email protected]> * feat(firestore-genai-chatbot): add genkit client --------- Co-authored-by: Mais Alheraki <[email protected]>
GoogleCloudPlatform · Nov 26, 2024 · 4a4eb88 · 4a4eb88
1 parent 3921a22
commit 4a4eb88
Show file tree

Hide file tree

Showing 10 changed files with 2,260 additions and 564 deletions.
diff --git a/firestore-genai-chatbot/functions/__tests__/google_ai/index.test.ts b/firestore-genai-chatbot/functions/__tests__/google_ai/index.test.ts
@@ -32,6 +32,8 @@ jest.mock('../../src/config', () => ({
     provider: 'google-ai',
     model: 'gemini-pro',
     apiKey: 'test-api-key',
+    candidateCount: 2,
+    candidateField: 'candidates',
   },
 }));
 

diff --git a/firestore-genai-chatbot/functions/__tests__/util.ts b/firestore-genai-chatbot/functions/__tests__/util.ts
@@ -27,6 +27,7 @@ export const expectToProcessCorrectly = (
   expect(firestoreCallData[2]).toEqual({
     ...message,
     response: mockResponse,
+    candidates: expect.any(Array),
     createTime: expect.any(Timestamp),
     status: {
       state: 'COMPLETED',

diff --git a/firestore-genai-chatbot/functions/__tests__/vertex_ai/index.test.ts b/firestore-genai-chatbot/functions/__tests__/vertex_ai/index.test.ts
@@ -16,11 +16,11 @@ process.env.FIRESTORE_EMULATOR_HOST = '127.0.0.1:8080';
 jest.mock('../../src/config', () => ({
   default: {
     googleAi: {
-      model: 'gemini-pro',
+      model: 'gemini-1.0-pro',
       apiKey: 'test-api-key',
     },
     vertex: {
-      model: 'gemini-pro',
+      model: 'gemini-1.0-pro',
     },
     collectionName: 'discussionsTestGenerative/{discussionId}/messages',
     location: 'us-central1',
@@ -30,8 +30,10 @@ jest.mock('../../src/config', () => ({
     enableDiscussionOptionOverrides: true,
     candidatesField: 'candidates',
     provider: 'vertex-ai',
-    model: 'gemini-pro',
+    model: 'gemini-1.0-pro',
     apiKey: 'test-api-key',
+    candidateCount: 2,
+    candidateField: 'candidates',
   },
 }));