Moving createVectorIndexer Firebase Functions to V2 (#2281)

* Moving createVectorIndexer firebase functions to V2

* Enable the vector indexing trigger functions

Author: Mephistic

functions/src/bills/index.ts

@@ -3,3 +3,4 @@ export * from "./bills"
 export * from "./search"
 export * from "./updateBillReferences"
 export * from "./updateBillSearchIndex"
+export * from "./vector"

functions/src/index.ts

@@ -7,7 +7,8 @@ export {
   syncBillToSearchIndex,
   updateBillReferences,
   updateBillSearchIndex,
-  upgradeBillSearchIndex
+  upgradeBillSearchIndex,
+  syncBillToVectorIndex
 } from "./bills"
 export { updateBillTracker } from "./analysis"
 export { fetchCityBatch, startCityBatches } from "./cities"
@@ -41,7 +42,8 @@ export {
   runTestimonyBackfillChunk,
   syncTestimonyToSearchIndex,
   upgradeTestimonySearchIndex,
-  resolveReportV2 as adminResolveReportV2
+  resolveReportV2 as adminResolveReportV2,
+  syncTestimonyToVectorIndex
 } from "./testimony"
 export {
   publishNotifications,
@@ -67,6 +69,8 @@ export { transcriptionV2 } from "./webhooks"
 export { matchOcpfMembersV2 } from "./ocpf/matchOcpfMembers"
 export { scrapeOcpfFinanceV2 } from "./ocpf/scrapeOcpfFinance"
 
+export { syncBallotQuestionToVectorIndex } from "./ballotQuestions/vector"
+
 export * from "./triggerPubsubFunction"
 
 export { mcpProxyV2 } from "./mcp/proxy"

functions/src/search/createVectorIndexer.ts

@@ -1,4 +1,4 @@
-import { runWith } from "firebase-functions"
+import { onDocumentWritten } from "firebase-functions/v2/firestore"
 import * as admin from "firebase-admin"
 import { FieldValue } from "firebase-admin/firestore"
 import { PredictionServiceClient, helpers } from "@google-cloud/aiplatform"
@@ -16,12 +16,15 @@ export function createVectorIndexer(config: VectorIndexerConfig) {
   const publisher = "google"
   const model = "text-embedding-005"
 
-  return runWith({
-    timeoutSeconds: 60,
-    memory: "512MB"
-  })
-    .firestore.document(config.documentTrigger)
-    .onWrite(async change => {
+  return onDocumentWritten(
+    {
+      document: config.documentTrigger,
+      timeoutSeconds: 60,
+      memory: "512MiB"
+    },
+    async event => {
+      const change = event.data
+      if (!change) return
       const data = change.after.exists ? change.after.data() : null
       if (!data) return // Deleted
 
@@ -103,5 +106,6 @@ export function createVectorIndexer(config: VectorIndexerConfig) {
         [config.vectorField]: fieldValue.vector(embedding),
         [`${config.vectorField}_hash`]: textHash // Store hash to track changes
       })
-    })
+    }
+  )
 }

functions/src/testimony/index.ts

@@ -2,3 +2,4 @@ export * from "./deleteTestimony"
 export * from "./publishTestimony"
 export * from "./search"
 export * from "./resolveReport"
+export * from "./vector"