AI - Ask a question closes #21.

Income metrics updates.
Added Dow Jones Industrial Average to Performance chart.

Author: Aymeric

README.md

@@ -208,6 +208,14 @@ Functions will automatically deploy with hosting deployments with `firebase depl
 firebase deploy --only functions
 ```
 
+The following secrets should be configured to prevent storing sensitive passwords and tokens in source control.
+
+```bash
+# Change the value of an existing secret
+firebase functions:secrets:set GEMINI_API_KEY
+# View the value of a secret
+firebase functions:secrets:access GEMINI_API_KEY
+```
 
 ### Android Play Store
 

src/robinhood_options_mobile/functions/package-lock.json

@@ -11,10 +11,12 @@
     "logs": "firebase functions:log"
   },
   "engines": {
-    "node": "18"
+    "node": "22"
   },
   "main": "lib/index.js",
   "dependencies": {
+    "@google-cloud/vertexai": "^1.9.3",
+    "@google/generative-ai": "^0.24.0",
     "firebase-admin": "^12.6.0",
     "firebase-functions": "^6.1.1",
     "plaid": "^29.0.0"

src/robinhood_options_mobile/functions/package.json

@@ -0,0 +1,66 @@
+// import * as functions from 'firebase-functions';
+import * as https from "firebase-functions/v2/https";
+import * as logger from "firebase-functions/logger";
+import {
+  // DynamicRetrievalMode,
+  GoogleGenerativeAI,
+} from "@google/generative-ai";
+import { VertexAI, type Tool } from "@google-cloud/vertexai";
+
+export const generateContent = https.onCall({ secrets: ["GEMINI_API_KEY"] },
+  async (request) => {
+    logger.info(request.data, { structuredData: true });
+    if (process.env.GEMINI_API_KEY == null) {
+      throw new https.HttpsError(
+        "unavailable", "GEMINI_API_KEY not found.");
+    }
+    const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY);
+    const model = genAI.getGenerativeModel(
+      {
+        model: "models/gemini-1.5-flash",
+        tools: [
+          {
+            googleSearchRetrieval: {
+              // dynamicRetrievalConfig: {
+              //   mode: DynamicRetrievalMode.MODE_DYNAMIC,
+              //   dynamicThreshold: 0.7,
+              // },
+            },
+          },
+        ],
+      },
+      { apiVersion: "v1beta" },
+    );
+
+    // const prompt = "Who won Wimbledon this year?";
+    const result = await model.generateContent(request.data.prompt);
+    console.log(result);
+    return result;
+  });
+
+export const generateContent2 = https.onCall({ secrets: ["GEMINI_API_KEY"] },
+  async (request) => {
+    logger.info(request.data, { structuredData: true });
+    if (process.env.GEMINI_API_KEY == null) {
+      throw new https.HttpsError(
+        "unavailable", "GEMINI_API_KEY not found.");
+    }
+    const vertexAI = new VertexAI({
+      project: "realizealpha", // process.env.GOOGLE_PROJECT_ID,
+      location: "us-central1", // process.env.GOOGLE_VERTEXAI_LOCATION,
+    });
+
+    const googleSearchTool = {
+      googleSearch: {},
+    } as Tool;
+
+    const model = vertexAI.getGenerativeModel({
+      model: "gemini-2.0-flash-001",
+      tools: [googleSearchTool],
+    });
+
+    const { response } = await model.generateContent({
+      contents: [{ role: "user", parts: [{ text: request.data.prompt }] }],
+    });
+    return response;
+  });

src/robinhood_options_mobile/functions/src/gemini.ts

@@ -16,6 +16,7 @@ initializeApp();
 import * as plaidfunc from "./plaid";
 import * as authfunc from "./auth";
 import * as messagingfunc from "./messaging";
+import * as gemini from "./gemini";
 
 // Start writing functions
 // https://firebase.google.com/docs/functions/typescript
@@ -34,3 +35,5 @@ export const getInvestmentsTransactions =
   plaidfunc.getInvestmentsTransactions;
 export const changeUserRole = authfunc.changeUserRole;
 export const sendEachForMulticast = messagingfunc.sendEachForMulticast;
+export const generateContent = gemini.generateContent;
+export const generateContent2 = gemini.generateContent2;