@blinkk/root-cms 2.4.9 → 2.4.10

package/dist/chunk-ZOEPOKGE.js

@@ -0,0 +1,294 @@
+// core/ai.ts
+import crypto from "crypto";
+import fs from "fs";
+import path from "path";
+import { vertexAI } from "@genkit-ai/google-genai";
+import { Timestamp } from "firebase-admin/firestore";
+import { genkit } from "genkit";
+import { logger } from "genkit/logging";
+logger.setLogLevel("warn");
+var DEFAULT_MODEL = "gemini-3-flash-preview";
+var DEFAULT_IMAGEGEN_MODEL = "gemini-2.5-flash-image";
+var LEGACY_MODEL_RENAME = {
+  "vertexai/gemini-2.5-flash": "gemini-2.5-flash",
+  "vertexai/gemini-2.0-pro": "gemini-2.0-pro"
+};
+async function summarizeDiff(cmsClient, options) {
+  const cmsPluginOptions = cmsClient.cmsPlugin.getConfig();
+  const firebaseConfig = cmsPluginOptions.firebaseConfig;
+  const model = (typeof cmsPluginOptions.experiments?.ai === "object" ? cmsPluginOptions.experiments.ai.model : void 0) || DEFAULT_MODEL;
+  const ai = genkit({
+    plugins: [
+      vertexAI({
+        projectId: firebaseConfig.projectId,
+        location: firebaseConfig.location || "us-central1"
+      })
+    ]
+  });
+  const beforeJson = JSON.stringify(options.before ?? null, null, 2);
+  const afterJson = JSON.stringify(options.after ?? null, null, 2);
+  const systemPrompt = [
+    "You are an assistant that summarizes changes made to CMS documents stored as JSON.",
+    "Provide a concise description of the most important updates using short bullet points.",
+    'If there are no meaningful differences, respond with "No significant changes."',
+    `Focus on just the content changes, ignore insignificant changes to richtext blocks and structure, such as updates to the richtext block's "timestamp" and "version" fields.`
+  ].join("\n");
+  const diffPrompt = [
+    "Previous version JSON:",
+    "```json",
+    beforeJson,
+    "```",
+    "",
+    "Updated version JSON:",
+    "```json",
+    afterJson,
+    "```",
+    "",
+    "Summarize the differences between the two payloads."
+  ].join("\n");
+  const res = await generate(ai, model, {
+    messages: [
+      {
+        role: "system",
+        content: [{ text: systemPrompt }]
+      }
+    ],
+    prompt: [{ text: diffPrompt }]
+  });
+  return res.text?.trim() || "";
+}
+async function generateImage(cmsClient, options) {
+  const cmsPluginOptions = cmsClient.cmsPlugin.getConfig();
+  const firebaseConfig = cmsPluginOptions.firebaseConfig;
+  const model = options.model || DEFAULT_IMAGEGEN_MODEL;
+  const ai = genkit({
+    plugins: [
+      vertexAI({
+        projectId: firebaseConfig.projectId,
+        location: firebaseConfig.location || "us-central1"
+      })
+    ]
+  });
+  const res = await generate(ai, model, {
+    prompt: [
+      {
+        text: options.prompt
+      }
+    ],
+    config: {
+      imageConfig: {
+        aspectRatio: options.aspectRatio
+      }
+    },
+    output: {
+      format: "media"
+    }
+  });
+  if (!res.media) {
+    throw new Error("No image generated");
+  }
+  return res.media.url;
+}
+async function generate(ai, model, options) {
+  const generateOptions = {
+    ...options,
+    model: vertexAI.model(model)
+  };
+  if (model.startsWith("gemini-3")) {
+    generateOptions.config = {
+      ...generateOptions.config,
+      location: "global"
+    };
+  }
+  return ai.generate(generateOptions);
+}
+var Chat = class {
+  constructor(chatClient, id, options) {
+    this.chatClient = chatClient;
+    this.cmsClient = chatClient.cmsClient;
+    this.cmsPluginOptions = this.cmsClient.cmsPlugin.getConfig();
+    this.id = id;
+    this.history = options?.history ?? [];
+    this.model = cleanModelName(
+      options?.model || (typeof this.cmsPluginOptions.experiments?.ai === "object" ? this.cmsPluginOptions.experiments.ai.model : void 0) || DEFAULT_MODEL
+    );
+    const firebaseConfig = this.cmsPluginOptions.firebaseConfig;
+    this.ai = genkit({
+      plugins: [
+        vertexAI({
+          projectId: firebaseConfig.projectId,
+          location: firebaseConfig.location || "us-central1"
+        })
+      ]
+    });
+  }
+  /** Builds the messages for the AI request. */
+  async buildMessages(options) {
+    const messages = this.history;
+    const hasSystemPrompt = messages.some(
+      (msg) => msg.role === "system" && msg.content.length > 0
+    );
+    if (!hasSystemPrompt) {
+      messages.push({
+        role: "system",
+        content: [{ text: await this.buildSystemPrompt(options) }]
+      });
+    }
+    if (options.mode === "edit") {
+      messages.push({
+        role: "user",
+        content: [
+          {
+            text: [
+              "The JSON you must edit is:",
+              "",
+              JSON.stringify(options.editData || {}, null, 2)
+            ].join("")
+          }
+        ]
+      });
+    }
+    return messages;
+  }
+  /** Builds the request sent to the AI based on the `ChatMode`. */
+  async buildGenerateRequest(prompt, options) {
+    if (options.mode === "edit") {
+      return {
+        messages: await this.buildMessages(options),
+        model: this.model,
+        prompt
+      };
+    }
+    return {
+      messages: await this.buildMessages(options),
+      model: this.model,
+      prompt
+    };
+  }
+  /** Sends the request to the AI and stores the history in the session and the database. */
+  async sendPrompt(prompt, options = {}) {
+    const chatRequest = await this.buildGenerateRequest(prompt, options);
+    const res = await generate(this.ai, chatRequest.model, {
+      messages: chatRequest.messages,
+      prompt: Array.isArray(prompt) ? prompt.flat() : prompt
+    });
+    this.history = res.messages;
+    await this.dbDoc().update({
+      history: this.history,
+      modifiedAt: Timestamp.now()
+    });
+    if (options.mode === "edit") {
+      const aiResponse = res.output;
+      if (aiResponse?.data && !aiResponse.message) {
+        aiResponse.message = "";
+      }
+      return aiResponse;
+    }
+    return { message: res.text, data: null };
+  }
+  dbDoc() {
+    return this.chatClient.dbCollection().doc(this.id);
+  }
+  /**
+   * Builds the system prompt sent to the AI, based on the `ChatMode` and
+   * supplied `SendPromptOptions`. `SendPromptOptions` may contain data or
+   * references to information needed to construct the prompt.
+   */
+  async buildSystemPrompt(options) {
+    const serializedRootConfig = JSON.stringify(
+      this.cmsClient.rootConfig,
+      null,
+      2
+    );
+    if (options.mode === "edit") {
+      const rootDir = process.cwd();
+      const rootCmsDefsPath = path.resolve(rootDir, "root-cms.d.ts");
+      const rootCmsDefs = fs.existsSync(rootCmsDefsPath) ? fs.readFileSync(rootCmsDefsPath, {
+        encoding: "utf8"
+      }) : null;
+      const text = [(await import("./edit-XX3LAGK6.js")).default];
+      if (rootCmsDefs) {
+        text.push(
+          "Here is the `root-cms.d.ts` file for this project:",
+          "```",
+          rootCmsDefs,
+          "```"
+        );
+      }
+      return text.join("\n");
+    }
+    if (options.mode === "altText") {
+      return (await import("./altText-RDKJNVGH.js")).default;
+    }
+    const systemText = [
+      `You are an assistant for a headless CMS called Root CMS which is used on a website called ${this.cmsPluginOptions.name || this.cmsPluginOptions.id}. Your job is to answer questions about the docs in the system, and if requested, help suggest changes to the JSON data in the docs. If you don't know the answer, just say that you don't know, don't try to make up an answer. Be friendly and playful with your messaging.`,
+      "",
+      "Here is the root.config.ts file for the site:",
+      "```",
+      serializedRootConfig,
+      "```",
+      "",
+      "Here are the docs that exist in the system:"
+    ];
+    const pages = await this.cmsClient.listDocs("Pages", { mode: "draft" });
+    pages.docs.forEach((doc) => {
+      systemText.push(JSON.stringify(doc));
+    });
+    return systemText.join("\n");
+  }
+};
+var ChatClient = class {
+  constructor(cmsClient, user) {
+    this.cmsClient = cmsClient;
+    this.user = user;
+  }
+  async createChat() {
+    const chatId = crypto.randomUUID();
+    const docRef = this.dbCollection().doc(chatId);
+    const chat = new Chat(this, chatId);
+    await docRef.set({
+      id: chatId,
+      createdBy: this.user,
+      createdAt: Timestamp.now(),
+      modifiedAt: Timestamp.now(),
+      model: chat.model
+    });
+    return chat;
+  }
+  async getOrCreateChat(chatId) {
+    return chatId ? this.getChat(chatId) : this.createChat();
+  }
+  async getChat(chatId) {
+    const docRef = this.dbCollection().doc(chatId);
+    const chatDoc = await docRef.get();
+    if (!chatDoc.exists) {
+      throw new Error(`${chatId} does not exist`);
+    }
+    const chatData = chatDoc.data() || {};
+    return new Chat(this, chatId, {
+      history: chatData.history,
+      model: chatData.model
+    });
+  }
+  async listChats(options) {
+    const limit = options?.limit || 20;
+    const query = this.dbCollection().where("createdBy", "==", this.user).limit(limit).orderBy("createdAt", "desc");
+    const res = await query.get();
+    return res.docs.map((doc) => doc.data());
+  }
+  dbCollection() {
+    return this.cmsClient.db.collection(
+      `Projects/${this.cmsClient.projectId}/Experiments/ai/Chat`
+    );
+  }
+};
+function cleanModelName(model) {
+  return LEGACY_MODEL_RENAME[model] || model;
+}
+
+export {
+  summarizeDiff,
+  generateImage,
+  Chat,
+  ChatClient
+};