add functions

functions/index.js

@@ -1,39 +1,39 @@
 const admin = require("firebase-admin");
-
-const serviceAccount = require("./serviceAccountProd.json");
+const serviceAccount = require("./serviceAccountKey.json");
+const { googleAI, gemini25FlashLite } = require("@genkit-ai/googleai");
+const { GEMINI_API_KEY } = require("./config/keys");
+const { genkit } = require("genkit");
 
 admin.initializeApp({
   credential: admin.credential.cert(serviceAccount),
+  storageBucket: "musicland-d33f9.firebasestorage.app",
 });
 
-let db = admin.firestore();
-exports.db = db;
+exports.generateAI = async ({ system = "", prompt = "", schema }) => {
+  const genkitInstance = genkit({
+    plugins: [
+      googleAI({
+        apiKey: GEMINI_API_KEY,
+      }),
+    ],
+    model: gemini25FlashLite,
+  });
 
+  if (prompt?.length < 1) {
+    throw new Error(
+      "Vous devez spécifier un prompt pour effectuer cette action.",
+    );
+  }
 
-const { vertexAI, gemini15Flash } = require("@genkit-ai/vertexai");
-const { genkit } = require("genkit");
+  const { output } = await genkitInstance.generate({
+    system,
+    prompt,
+    output: { schema },
+  });
 
-require("dotenv").config();
-
-require("@babel/register")({
-  presets: ["@babel/preset-env", "@babel/preset-react"],
-});
-
-// configure a Genkit instance
-exports.genkitInstance = genkit({
-  plugins: [vertexAI({ projectId: "starsclick", location: "us-central1" })],
-  logLevel: "debug",
-  enableTracingAndMetrics: true,
-  model: gemini15Flash, // set default model
-});
-
-
-exports.refsList = {
-  users: db.collection("users"),
-  documents: db.collection("documents"),
-  logs: db.collection("logs"),
+  return output;
 };
 
-exports.users = require("./src/users");
-exports.payments = require("./src/payments");
-exports.documents = require("./src/documents");
+// Exporter toutes les fonctions
+exports.music = require("./src/music");
+exports.lyrics = require("./src/lyrics");