feat: vertexAI package support

.eslintignore

@@ -8,4 +8,6 @@ docs
 packages/template/project/**
 app.playground.js
 type-test.ts
-packages/**/modular/dist/**
+packages/**/modular/dist/**
+packages/vertexai/__tests__/test-utils
+packages/vertexai/dist

.github/workflows/scripts/functions/src/index.ts

@@ -24,3 +24,5 @@ export { testFunctionDefaultRegionV2 } from './testFunctionDefaultRegion';
 export { testFunctionRemoteConfigUpdateV2 } from './testFunctionRemoteConfigUpdate';
 export { fetchAppCheckTokenV2 } from './fetchAppCheckToken';
 export { sendFCM } from './sendFCM';
+
+export { testFetchStream, testFetch } from './vertexaiFunctions';

.github/workflows/scripts/functions/src/vertexaiFunctions.ts

@@ -0,0 +1,62 @@
+import * as functions from 'firebase-functions/v2';
+
+const poem = [
+  'The wind whispers secrets through the trees,',
+  'Rustling leaves in a gentle breeze.',
+  'Sunlight dances on the grass,',
+  'A fleeting moment, sure to pass.',
+  'Birdsong fills the air so bright,',
+  'A symphony of pure delight.',
+  'Time stands still, a peaceful pause,',
+  "In nature's beauty, no flaws.",
+];
+
+const response = {
+  candidates: [
+    {
+      content: {
+        parts: [
+          {
+            text: 'Google\'s mission is to "organize the world\'s information and make it universally accessible and useful."',
+          },
+        ],
+        role: 'model',
+      },
+      finishReason: 'STOP',
+      index: 0,
+      safetyRatings: [
+        { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', probability: 'NEGLIGIBLE' },
+        { category: 'HARM_CATEGORY_HATE_SPEECH', probability: 'NEGLIGIBLE' },
+        { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },
+        { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', probability: 'NEGLIGIBLE' },
+      ],
+    },
+  ],
+  promptFeedback: {
+    safetyRatings: [
+      { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', probability: 'NEGLIGIBLE' },
+      { category: 'HARM_CATEGORY_HATE_SPEECH', probability: 'NEGLIGIBLE' },
+      { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },
+      { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', probability: 'NEGLIGIBLE' },
+    ],
+  },
+};
+
+export const testFetchStream = functions.https.onRequest(async (req, res) => {
+  res.setHeader('Content-Type', 'text/event-stream');
+  res.setHeader('Cache-Control', 'no-cache');
+  res.setHeader('Connection', 'keep-alive');
+
+  const updatedResponse = { ...response };
+
+  for (let i = 0; i < poem.length; i++) {
+    updatedResponse.candidates[0].content.parts[0].text = poem[i];
+    res.write(`data: ${JSON.stringify(updatedResponse)}\n\n`);
+    await new Promise(resolve => setTimeout(resolve, 100));
+  }
+  res.end();
+});
+
+export const testFetch = functions.https.onRequest(async (req, res) => {
+  res.json(response);
+});

.gitignore

@@ -535,6 +535,7 @@ Pods/*
 **/Pods/**
 **/dist/
 packages/**/version.js
+packages/**/version.ts
 typedoc.raw.json
 tests/ios/Firebase
 tests/ios/resetXcode.sh
@@ -579,3 +580,7 @@ website/public
 
 # Typescript items
 *.tsbuildinfo
+
+# vertexai test data
+vertexai-sdk-test-data
+mocks-lookup.ts

.spellcheck.dict.txt

@@ -76,6 +76,7 @@ firebase-js-sdk
 FirebaseApp
 firestore
 Firestore
+freeform
 GDPR
 GDPR-compliant
 getIdToken
@@ -214,6 +215,7 @@ v6
 v7
 v9
 Ventura
+VertexAI
 VPN
 VSCode
 Wix

docs/sidebar.yaml

@@ -127,6 +127,10 @@
     - - KY Integration
       - '/perf/ky-integration'
   - '//static.invertase.io/assets/firebase/performance-monitoring.svg'
+- - VertexAi
+  - - - Usage
+      - '/vertexai/usage'
+  - '//static.invertase.io/assets/social/firebase-logo.png'
 - - Legacy docs
   - - - Migrating to v6
       - '/migrating-to-v6'