@blinkk/root-cms 3.0.1-alpha.1 → 3.0.1-beta.2

package/dist/chunk-T5UK2H24.js

@@ -1,419 +0,0 @@
-// 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 = "gemini-2.5-flash";
-  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 = [
-    "Summarize CMS document changes in 2-4 bullet points.",
-    "Focus on content changes only. Ignore metadata like timestamps.",
-    'If no meaningful changes, say "No significant changes."'
-  ].join("\n");
-  const diffPrompt = [
-    "Before:",
-    beforeJson,
-    "",
-    "After:",
-    afterJson,
-    "",
-    "What changed?"
-  ].join("\n");
-  const res = await generate(ai, model, {
-    messages: [
-      {
-        role: "system",
-        content: [{ text: systemPrompt }]
-      }
-    ],
-    prompt: [{ text: diffPrompt }],
-    config: {
-      // Respond more quickly with less creativity.
-      temperature: 0.3
-    }
-  });
-  return res.text?.trim() || "";
-}
-async function generatePublishMessage(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 generates concise commit-style messages for CMS document changes.",
-    "Generate a single short sentence (maximum 60 characters) describing the most important change.",
-    'Use imperative mood like "Add feature" or "Update content" or "Fix typo".',
-    "Focus on the key content change, ignore structural metadata changes.",
-    "Do not use punctuation at the end.",
-    'Examples: "Add new hero image", "Update pricing details", "Fix typo in headline"',
-    "Include "
-  ].join("\n");
-  const diffPrompt = [
-    "Previous version JSON:",
-    "```json",
-    beforeJson,
-    "```",
-    "",
-    "Updated version JSON:",
-    "```json",
-    afterJson,
-    "```",
-    "",
-    "Generate a commit message for these changes."
-  ].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;
-}
-function extractJsonFromResponse(responseText) {
-  let jsonText = responseText.trim();
-  if (jsonText.startsWith("```")) {
-    const lines = jsonText.split("\n");
-    jsonText = lines.slice(1, -1).join("\n");
-    if (jsonText.startsWith("json")) {
-      jsonText = jsonText.substring(4).trim();
-    }
-  }
-  return jsonText;
-}
-async function translateString(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 systemPrompt = [
-    "You are a professional translator assistant.",
-    "Translate the given source text into the requested target languages.",
-    "Maintain the tone, style, and intent of the original text.",
-    "Return ONLY a valid JSON object with locale codes as keys and translations as values.",
-    "Do not include any markdown formatting, code blocks, or explanatory text."
-  ].join("\n");
-  const userPromptParts = [
-    `Source text: "${options.sourceText}"`,
-    ""
-  ];
-  if (options.description) {
-    userPromptParts.push(`Context/Description: ${options.description}`, "");
-  }
-  if (options.existingTranslations && Object.keys(options.existingTranslations).length > 0) {
-    userPromptParts.push("Existing translations for reference:");
-    Object.entries(options.existingTranslations).forEach(
-      ([locale, translation]) => {
-        if (translation) {
-          userPromptParts.push(`- ${locale}: "${translation}"`);
-        }
-      }
-    );
-    userPromptParts.push("");
-  }
-  userPromptParts.push(
-    `Target locales: ${options.targetLocales.join(", ")}`,
-    "",
-    "Provide translations as a JSON object with locale codes as keys."
-  );
-  const userPrompt = userPromptParts.join("\n");
-  const res = await generate(ai, model, {
-    messages: [
-      {
-        role: "system",
-        content: [{ text: systemPrompt }]
-      },
-      {
-        role: "user",
-        content: [{ text: userPrompt }]
-      }
-    ]
-  });
-  const responseText = res.text || "{}";
-  const jsonText = extractJsonFromResponse(responseText);
-  try {
-    const translations = JSON.parse(jsonText);
-    return translations;
-  } catch (err) {
-    console.error("Failed to parse AI translation response:", responseText);
-    throw new Error("Invalid response format from AI translation");
-  }
-}
-
-export {
-  summarizeDiff,
-  generatePublishMessage,
-  generateImage,
-  Chat,
-  ChatClient,
-  extractJsonFromResponse,
-  translateString
-};