strapi-plugin-ai-sdk 0.7.3 → 0.7.7

package/dist/_chunks/{App-DCV7o6Hc.mjs → App-DMdQymB3.mjs}

@@ -4,7 +4,7 @@ import { Link, useNavigate, Routes, Route } from "react-router-dom";
 import { Box, Typography, TextInput, Button, Main, SearchForm, Searchbar, Table, Thead, Tr, Th, Tbody, Td, Flex, Pagination, Modal, Field, Textarea, SingleSelect, SingleSelectOption } from "@strapi/design-system";
 import { useState, useEffect, useCallback, useMemo, useRef, forwardRef } from "react";
 import styled from "styled-components";
-import { P as PLUGIN_ID } from "./index-BMrDQVQl.mjs";
+import { P as PLUGIN_ID } from "./index-DrLcqX__.mjs";
 import { Plus, Trash, Sparkle, ArrowLeft, Pencil } from "@strapi/icons";
 import Markdown from "react-markdown";
 import remarkGfm from "remark-gfm";
@@ -1099,14 +1099,14 @@ function extractContentLinks(toolCall) {
     }
     return links;
   }
-  if (toolCall.toolName === "writeContent") {
+  if (toolCall.toolName === "createContent" || toolCall.toolName === "updateContent") {
     const doc = output.document;
     const docId = doc?.documentId;
     if (docId) {
       const title = doc?.title || doc?.name || docId;
       return [
         {
-          label: `${input.action === "create" ? "Created" : "Updated"}: ${title}`,
+          label: `${toolCall.toolName === "createContent" ? "Created" : "Updated"}: ${title}`,
           to: buildContentManagerUrl(contentType, docId)
         }
       ];

package/dist/_chunks/{App-CEEsJsKL.js → App-joEmdpxi.js}

@@ -6,7 +6,7 @@ const reactRouterDom = require("react-router-dom");
 const designSystem = require("@strapi/design-system");
 const react = require("react");
 const styled = require("styled-components");
-const index = require("./index-Cw2aiQ8K.js");
+const index = require("./index-B7qLITWV.js");
 const icons = require("@strapi/icons");
 const Markdown = require("react-markdown");
 const remarkGfm = require("remark-gfm");
@@ -1105,14 +1105,14 @@ function extractContentLinks(toolCall) {
     }
     return links;
   }
-  if (toolCall.toolName === "writeContent") {
+  if (toolCall.toolName === "createContent" || toolCall.toolName === "updateContent") {
     const doc = output.document;
     const docId = doc?.documentId;
     if (docId) {
       const title = doc?.title || doc?.name || docId;
       return [
         {
-          label: `${input.action === "create" ? "Created" : "Updated"}: ${title}`,
+          label: `${toolCall.toolName === "createContent" ? "Created" : "Updated"}: ${title}`,
           to: buildContentManagerUrl(contentType, docId)
         }
       ];

package/dist/_chunks/{index-Cw2aiQ8K.js → index-B7qLITWV.js}

@@ -37,7 +37,7 @@ const index = {
         defaultMessage: PLUGIN_ID
       },
       Component: async () => {
-        const { App } = await Promise.resolve().then(() => require("./App-CEEsJsKL.js"));
+        const { App } = await Promise.resolve().then(() => require("./App-joEmdpxi.js"));
         return App;
       }
     });

package/dist/_chunks/{index-BMrDQVQl.mjs → index-DrLcqX__.mjs}

@@ -36,7 +36,7 @@ const index = {
         defaultMessage: PLUGIN_ID
       },
       Component: async () => {
-        const { App } = await import("./App-DCV7o6Hc.mjs");
+        const { App } = await import("./App-DMdQymB3.mjs");
         return App;
       }
     });

package/dist/admin/index.js

@@ -1,3 +1,3 @@
 "use strict";
-const index = require("../_chunks/index-Cw2aiQ8K.js");
+const index = require("../_chunks/index-B7qLITWV.js");
 module.exports = index.index;

package/dist/admin/index.mjs

@@ -1,4 +1,4 @@
-import { i } from "../_chunks/index-BMrDQVQl.mjs";
+import { i } from "../_chunks/index-DrLcqX__.mjs";
 export {
   i as default
 };

package/dist/server/index.js

@@ -387,33 +387,41 @@ async function searchContent(strapi, params) {
     }
   };
 }
-const writeContentSchema = zod.z.object({
+const createContentSchema = zod.z.object({
   contentType: zod.z.string().describe('Content type UID, e.g. "api::article.article"'),
-  action: zod.z.enum(["create", "update"]).describe("Whether to create a new document or update an existing one"),
-  documentId: zod.z.string().optional().describe("Required for update — the document ID to update"),
   data: zod.z.record(zod.z.string(), zod.z.unknown()).describe("The field values to set. Must match the content type schema."),
   status: zod.z.enum(["draft", "published"]).optional().describe("Document status. Defaults to draft."),
   locale: zod.z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"')
 });
-const writeContentDescription = "Create or update a document in any Strapi content type. Use listContentTypes first to discover the schema, and searchContent to find existing documents for updates.";
-async function writeContent(strapi, params) {
-  const { contentType, action, documentId, data, status, locale } = params;
+const createContentDescription = "Create a new document in any Strapi content type (articles, blog posts, pages, etc.). Use listContentTypes first to discover available content types and their fields.";
+async function createContent(strapi, params) {
+  const { contentType, data, status, locale } = params;
   if (!strapi.contentTypes[contentType]) {
     throw new Error(`Content type "${contentType}" does not exist.`);
   }
-  if (action === "update" && !documentId) {
-    throw new Error("documentId is required for update actions.");
-  }
   const docs = strapi.documents(contentType);
-  if (action === "create") {
-    const document2 = await docs.create({
-      data,
-      ...status ? { status } : {},
-      ...locale ? { locale } : {},
-      populate: "*"
-    });
-    return { action: "create", document: document2 };
+  const document = await docs.create({
+    data,
+    ...status ? { status } : {},
+    ...locale ? { locale } : {},
+    populate: "*"
+  });
+  return { action: "create", document };
+}
+const updateContentSchema = zod.z.object({
+  contentType: zod.z.string().describe('Content type UID, e.g. "api::article.article"'),
+  documentId: zod.z.string().describe("The document ID to update"),
+  data: zod.z.record(zod.z.string(), zod.z.unknown()).describe("The field values to set. Must match the content type schema."),
+  status: zod.z.enum(["draft", "published"]).optional().describe("Document status. Defaults to draft."),
+  locale: zod.z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"')
+});
+const updateContentDescription = "Update an existing document in any Strapi content type. Use searchContent to find the document ID first.";
+async function updateContent(strapi, params) {
+  const { contentType, documentId, data, status, locale } = params;
+  if (!strapi.contentTypes[contentType]) {
+    throw new Error(`Content type "${contentType}" does not exist.`);
   }
+  const docs = strapi.documents(contentType);
   const existing = await docs.findOne({
     documentId,
     ...locale ? { locale } : {}
@@ -574,7 +582,7 @@ const uploadMediaSchema = zod.z.object({
   caption: zod.z.string().optional().describe("Caption for the media file"),
   alternativeText: zod.z.string().optional().describe("Alternative text for accessibility")
 });
-const uploadMediaDescription = "Upload a media file from a URL to the Strapi media library. Returns the uploaded file data. To link media to a content type field, use writeContent with the file ID.";
+const uploadMediaDescription = "Upload a media file from a URL to the Strapi media library. Returns the uploaded file data. To link media to a content type field, use createContent or updateContent with the file ID.";
 async function uploadMedia(strapi, params) {
   const { url, name, caption, alternativeText } = params;
   let parsedUrl;
@@ -642,7 +650,7 @@ async function uploadMedia(strapi, params) {
   return {
     file: uploadedFile,
     message: `File "${uploadedFile.name}" uploaded successfully (ID: ${uploadedFile.id}).`,
-    usage: `To link this file to a content type field, use writeContent with: { "fieldName": ${uploadedFile.id} }`
+    usage: `To link this file to a content type field, use createContent or updateContent with: { "fieldName": ${uploadedFile.id} }`
   };
 }
 const CONTENT_TYPE$5 = "plugin::ai-sdk.public-memory";
@@ -1143,13 +1151,24 @@ async function sanitizeInput(strapi, uid, data, auth) {
     throw error;
   }
 }
-const writeContentTool = {
-  name: "writeContent",
-  description: writeContentDescription,
-  schema: writeContentSchema,
+const createContentTool = {
+  name: "createContent",
+  description: createContentDescription,
+  schema: createContentSchema,
+  execute: async (args, strapi) => {
+    const sanitizedData = await sanitizeInput(strapi, args.contentType, args.data);
+    const result = await createContent(strapi, { ...args, data: sanitizedData });
+    const sanitizedDoc = await sanitizeOutput(strapi, args.contentType, result.document);
+    return { ...result, document: sanitizedDoc };
+  }
+};
+const updateContentTool = {
+  name: "updateContent",
+  description: updateContentDescription,
+  schema: updateContentSchema,
   execute: async (args, strapi) => {
     const sanitizedData = await sanitizeInput(strapi, args.contentType, args.data);
-    const result = await writeContent(strapi, { ...args, data: sanitizedData });
+    const result = await updateContent(strapi, { ...args, data: sanitizedData });
     const sanitizedDoc = await sanitizeOutput(strapi, args.contentType, result.document);
     return { ...result, document: sanitizedDoc };
   }
@@ -1217,7 +1236,8 @@ const manageTaskTool = {
 const builtInTools = [
   listContentTypesTool,
   searchContentTool,
-  writeContentTool,
+  createContentTool,
+  updateContentTool,
   findOneContentTool,
   uploadMediaTool,
   sendEmailTool,
@@ -1231,6 +1251,14 @@ const PLUGIN_ID$2 = "ai-sdk";
 const bootstrap = ({ strapi }) => {
   const plugin = strapi.plugin(PLUGIN_ID$2);
   const config2 = strapi.config.get(`plugin::${PLUGIN_ID$2}`);
+  initializeProvider(strapi, plugin, config2);
+  const registry = initializeToolRegistry(plugin);
+  discoverPluginTools(strapi, registry);
+  plugin.createMcpServer = () => createMcpServer(strapi);
+  plugin.mcpSessions = /* @__PURE__ */ new Map();
+  strapi.log.info(`[${PLUGIN_ID$2}] MCP endpoint available at: /api/${PLUGIN_ID$2}/mcp`);
+};
+function initializeProvider(strapi, plugin, config2) {
   AIProvider.registerProvider("anthropic", ({ apiKey, baseURL }) => {
     const provider = anthropic.createAnthropic({ apiKey, baseURL });
     return (modelId) => provider(modelId);
@@ -1243,27 +1271,22 @@ const bootstrap = ({ strapi }) => {
   } else {
     strapi.log.warn(`[${PLUGIN_ID$2}] anthropicApiKey not configured, AI provider will not be available`);
   }
+}
+function initializeToolRegistry(plugin) {
   const toolRegistry = new ToolRegistry();
   for (const tool of builtInTools) {
     toolRegistry.register(tool);
   }
   plugin.toolRegistry = toolRegistry;
+  return toolRegistry;
+}
+function discoverPluginTools(strapi, registry) {
   const pluginNames = Object.keys(strapi.plugins).filter((n) => n !== PLUGIN_ID$2);
   strapi.log.info(`[${PLUGIN_ID$2}] Scanning ${pluginNames.length} plugins for ai-tools: [${pluginNames.join(", ")}]`);
   for (const [pluginName, pluginInstance] of Object.entries(strapi.plugins)) {
     if (pluginName === PLUGIN_ID$2) continue;
     try {
-      let aiToolsService = null;
-      try {
-        aiToolsService = strapi.plugin(pluginName)?.service?.("ai-tools");
-      } catch {
-      }
-      if (!aiToolsService) {
-        try {
-          aiToolsService = pluginInstance.service?.("ai-tools");
-        } catch {
-        }
-      }
+      const aiToolsService = resolveAiToolsService(strapi, pluginName, pluginInstance);
       if (!aiToolsService?.getTools) {
         strapi.log.debug(`[${PLUGIN_ID$2}] No ai-tools service on plugin: ${pluginName}`);
         continue;
@@ -1271,21 +1294,7 @@ const bootstrap = ({ strapi }) => {
       strapi.log.info(`[${PLUGIN_ID$2}] Found ai-tools service on plugin: ${pluginName}`);
       const contributed = aiToolsService.getTools();
       if (!Array.isArray(contributed)) continue;
-      let count = 0;
-      for (const tool of contributed) {
-        if (!tool.name || !tool.execute || !tool.schema) {
-          strapi.log.warn(`[${PLUGIN_ID$2}] Invalid tool from ${pluginName}: ${tool.name || "unnamed"}`);
-          continue;
-        }
-        const safeName = pluginName.replace(/[^a-zA-Z0-9_-]/g, "_");
-        const namespacedName = `${safeName}__${tool.name}`;
-        if (toolRegistry.has(namespacedName)) {
-          strapi.log.warn(`[${PLUGIN_ID$2}] Duplicate tool: ${namespacedName}`);
-          continue;
-        }
-        toolRegistry.register({ ...tool, name: namespacedName });
-        count++;
-      }
+      const count = registerContributedTools(strapi, registry, pluginName, contributed);
       if (count > 0) {
         strapi.log.info(`[${PLUGIN_ID$2}] Registered ${count} tools from plugin: ${pluginName}`);
       }
@@ -1293,10 +1302,38 @@ const bootstrap = ({ strapi }) => {
       strapi.log.warn(`[${PLUGIN_ID$2}] Tool discovery failed for ${pluginName}: ${err}`);
     }
   }
-  plugin.createMcpServer = () => createMcpServer(strapi);
-  plugin.mcpSessions = /* @__PURE__ */ new Map();
-  strapi.log.info(`[${PLUGIN_ID$2}] MCP endpoint available at: /api/${PLUGIN_ID$2}/mcp`);
-};
+}
+function registerContributedTools(strapi, registry, pluginName, tools) {
+  const safeName = pluginName.replace(/[^a-zA-Z0-9_-]/g, "_");
+  let count = 0;
+  for (const tool of tools) {
+    if (!tool.name || !tool.execute || !tool.schema) {
+      strapi.log.warn(`[${PLUGIN_ID$2}] Invalid tool from ${pluginName}: ${tool.name || "unnamed"}`);
+      continue;
+    }
+    const namespacedName = `${safeName}__${tool.name}`;
+    if (registry.has(namespacedName)) {
+      strapi.log.warn(`[${PLUGIN_ID$2}] Duplicate tool: ${namespacedName}`);
+      continue;
+    }
+    registry.register({ ...tool, name: namespacedName });
+    count++;
+  }
+  return count;
+}
+function resolveAiToolsService(strapi, pluginName, pluginInstance) {
+  try {
+    const svc = strapi.plugin(pluginName)?.service?.("ai-tools");
+    if (svc) return svc;
+  } catch {
+  }
+  try {
+    const svc = pluginInstance.service?.("ai-tools");
+    if (svc) return svc;
+  } catch {
+  }
+  return null;
+}
 const PLUGIN_ID$1 = "ai-sdk";
 async function closeSession(strapi, sessionId, session) {
   try {
@@ -2644,15 +2681,21 @@ function createTools(strapi, context) {
   return tools;
 }
 const CONTENT_TOOLS = /* @__PURE__ */ new Set(["searchContent", "findOneContent", "aggregateContent"]);
-function createPublicTools(strapi, allowedContentTypes) {
+function createPublicTools(strapi, allowedContentTypes, publicToolSources) {
   const plugin = strapi.plugin("ai-sdk");
   const registry = plugin.toolRegistry;
   if (!registry) {
     throw new Error("Tool registry not initialized");
   }
   const allowed = new Set(allowedContentTypes);
+  const allowedSources = new Set(publicToolSources ?? []);
   const tools = {};
   for (const [name, def] of registry.getPublicSafe()) {
+    const sepIndex = name.indexOf("__");
+    if (sepIndex !== -1) {
+      const prefix = name.substring(0, sepIndex);
+      if (!allowedSources.has(prefix)) continue;
+    }
     if (CONTENT_TOOLS.has(name)) {
       tools[name] = ai.tool({
         description: def.description,
@@ -2800,9 +2843,10 @@ ${lines.join("\n")}`;
       const maxSteps = publicConfig?.maxSteps ?? DEFAULT_PUBLIC_MAX_STEPS;
       const publicModel = publicConfig?.chatModel ?? DEFAULT_PUBLIC_CHAT_MODEL;
       const allowedContentTypes = publicConfig?.allowedContentTypes ?? [];
+      const publicToolSources = publicConfig?.publicToolSources;
       const trimmedMessages = messages.length > maxMessages ? messages.slice(-maxMessages) : messages;
       const modelMessages = await ai.convertToModelMessages(trimmedMessages);
-      const tools = createPublicTools(strapi, allowedContentTypes);
+      const tools = createPublicTools(strapi, allowedContentTypes, publicToolSources);
       const toolsDescription = describeTools(tools);
       let system = composeSystemPrompt(config2, toolsDescription, options2?.system || DEFAULT_PUBLIC_PREAMBLE);
       try {

package/dist/server/index.mjs

@@ -367,33 +367,41 @@ async function searchContent(strapi, params) {
     }
   };
 }
-const writeContentSchema = z.object({
+const createContentSchema = z.object({
   contentType: z.string().describe('Content type UID, e.g. "api::article.article"'),
-  action: z.enum(["create", "update"]).describe("Whether to create a new document or update an existing one"),
-  documentId: z.string().optional().describe("Required for update — the document ID to update"),
   data: z.record(z.string(), z.unknown()).describe("The field values to set. Must match the content type schema."),
   status: z.enum(["draft", "published"]).optional().describe("Document status. Defaults to draft."),
   locale: z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"')
 });
-const writeContentDescription = "Create or update a document in any Strapi content type. Use listContentTypes first to discover the schema, and searchContent to find existing documents for updates.";
-async function writeContent(strapi, params) {
-  const { contentType, action, documentId, data, status, locale } = params;
+const createContentDescription = "Create a new document in any Strapi content type (articles, blog posts, pages, etc.). Use listContentTypes first to discover available content types and their fields.";
+async function createContent(strapi, params) {
+  const { contentType, data, status, locale } = params;
   if (!strapi.contentTypes[contentType]) {
     throw new Error(`Content type "${contentType}" does not exist.`);
   }
-  if (action === "update" && !documentId) {
-    throw new Error("documentId is required for update actions.");
-  }
   const docs = strapi.documents(contentType);
-  if (action === "create") {
-    const document2 = await docs.create({
-      data,
-      ...status ? { status } : {},
-      ...locale ? { locale } : {},
-      populate: "*"
-    });
-    return { action: "create", document: document2 };
+  const document = await docs.create({
+    data,
+    ...status ? { status } : {},
+    ...locale ? { locale } : {},
+    populate: "*"
+  });
+  return { action: "create", document };
+}
+const updateContentSchema = z.object({
+  contentType: z.string().describe('Content type UID, e.g. "api::article.article"'),
+  documentId: z.string().describe("The document ID to update"),
+  data: z.record(z.string(), z.unknown()).describe("The field values to set. Must match the content type schema."),
+  status: z.enum(["draft", "published"]).optional().describe("Document status. Defaults to draft."),
+  locale: z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"')
+});
+const updateContentDescription = "Update an existing document in any Strapi content type. Use searchContent to find the document ID first.";
+async function updateContent(strapi, params) {
+  const { contentType, documentId, data, status, locale } = params;
+  if (!strapi.contentTypes[contentType]) {
+    throw new Error(`Content type "${contentType}" does not exist.`);
   }
+  const docs = strapi.documents(contentType);
   const existing = await docs.findOne({
     documentId,
     ...locale ? { locale } : {}
@@ -554,7 +562,7 @@ const uploadMediaSchema = z.object({
   caption: z.string().optional().describe("Caption for the media file"),
   alternativeText: z.string().optional().describe("Alternative text for accessibility")
 });
-const uploadMediaDescription = "Upload a media file from a URL to the Strapi media library. Returns the uploaded file data. To link media to a content type field, use writeContent with the file ID.";
+const uploadMediaDescription = "Upload a media file from a URL to the Strapi media library. Returns the uploaded file data. To link media to a content type field, use createContent or updateContent with the file ID.";
 async function uploadMedia(strapi, params) {
   const { url, name, caption, alternativeText } = params;
   let parsedUrl;
@@ -622,7 +630,7 @@ async function uploadMedia(strapi, params) {
   return {
     file: uploadedFile,
     message: `File "${uploadedFile.name}" uploaded successfully (ID: ${uploadedFile.id}).`,
-    usage: `To link this file to a content type field, use writeContent with: { "fieldName": ${uploadedFile.id} }`
+    usage: `To link this file to a content type field, use createContent or updateContent with: { "fieldName": ${uploadedFile.id} }`
   };
 }
 const CONTENT_TYPE$5 = "plugin::ai-sdk.public-memory";
@@ -1123,13 +1131,24 @@ async function sanitizeInput(strapi, uid, data, auth) {
     throw error;
   }
 }
-const writeContentTool = {
-  name: "writeContent",
-  description: writeContentDescription,
-  schema: writeContentSchema,
+const createContentTool = {
+  name: "createContent",
+  description: createContentDescription,
+  schema: createContentSchema,
+  execute: async (args, strapi) => {
+    const sanitizedData = await sanitizeInput(strapi, args.contentType, args.data);
+    const result = await createContent(strapi, { ...args, data: sanitizedData });
+    const sanitizedDoc = await sanitizeOutput(strapi, args.contentType, result.document);
+    return { ...result, document: sanitizedDoc };
+  }
+};
+const updateContentTool = {
+  name: "updateContent",
+  description: updateContentDescription,
+  schema: updateContentSchema,
   execute: async (args, strapi) => {
     const sanitizedData = await sanitizeInput(strapi, args.contentType, args.data);
-    const result = await writeContent(strapi, { ...args, data: sanitizedData });
+    const result = await updateContent(strapi, { ...args, data: sanitizedData });
     const sanitizedDoc = await sanitizeOutput(strapi, args.contentType, result.document);
     return { ...result, document: sanitizedDoc };
   }
@@ -1197,7 +1216,8 @@ const manageTaskTool = {
 const builtInTools = [
   listContentTypesTool,
   searchContentTool,
-  writeContentTool,
+  createContentTool,
+  updateContentTool,
   findOneContentTool,
   uploadMediaTool,
   sendEmailTool,
@@ -1211,6 +1231,14 @@ const PLUGIN_ID$2 = "ai-sdk";
 const bootstrap = ({ strapi }) => {
   const plugin = strapi.plugin(PLUGIN_ID$2);
   const config2 = strapi.config.get(`plugin::${PLUGIN_ID$2}`);
+  initializeProvider(strapi, plugin, config2);
+  const registry = initializeToolRegistry(plugin);
+  discoverPluginTools(strapi, registry);
+  plugin.createMcpServer = () => createMcpServer(strapi);
+  plugin.mcpSessions = /* @__PURE__ */ new Map();
+  strapi.log.info(`[${PLUGIN_ID$2}] MCP endpoint available at: /api/${PLUGIN_ID$2}/mcp`);
+};
+function initializeProvider(strapi, plugin, config2) {
   AIProvider.registerProvider("anthropic", ({ apiKey, baseURL }) => {
     const provider = createAnthropic({ apiKey, baseURL });
     return (modelId) => provider(modelId);
@@ -1223,27 +1251,22 @@ const bootstrap = ({ strapi }) => {
   } else {
     strapi.log.warn(`[${PLUGIN_ID$2}] anthropicApiKey not configured, AI provider will not be available`);
   }
+}
+function initializeToolRegistry(plugin) {
   const toolRegistry = new ToolRegistry();
   for (const tool2 of builtInTools) {
     toolRegistry.register(tool2);
   }
   plugin.toolRegistry = toolRegistry;
+  return toolRegistry;
+}
+function discoverPluginTools(strapi, registry) {
   const pluginNames = Object.keys(strapi.plugins).filter((n) => n !== PLUGIN_ID$2);
   strapi.log.info(`[${PLUGIN_ID$2}] Scanning ${pluginNames.length} plugins for ai-tools: [${pluginNames.join(", ")}]`);
   for (const [pluginName, pluginInstance] of Object.entries(strapi.plugins)) {
     if (pluginName === PLUGIN_ID$2) continue;
     try {
-      let aiToolsService = null;
-      try {
-        aiToolsService = strapi.plugin(pluginName)?.service?.("ai-tools");
-      } catch {
-      }
-      if (!aiToolsService) {
-        try {
-          aiToolsService = pluginInstance.service?.("ai-tools");
-        } catch {
-        }
-      }
+      const aiToolsService = resolveAiToolsService(strapi, pluginName, pluginInstance);
       if (!aiToolsService?.getTools) {
         strapi.log.debug(`[${PLUGIN_ID$2}] No ai-tools service on plugin: ${pluginName}`);
         continue;
@@ -1251,21 +1274,7 @@ const bootstrap = ({ strapi }) => {
       strapi.log.info(`[${PLUGIN_ID$2}] Found ai-tools service on plugin: ${pluginName}`);
       const contributed = aiToolsService.getTools();
       if (!Array.isArray(contributed)) continue;
-      let count = 0;
-      for (const tool2 of contributed) {
-        if (!tool2.name || !tool2.execute || !tool2.schema) {
-          strapi.log.warn(`[${PLUGIN_ID$2}] Invalid tool from ${pluginName}: ${tool2.name || "unnamed"}`);
-          continue;
-        }
-        const safeName = pluginName.replace(/[^a-zA-Z0-9_-]/g, "_");
-        const namespacedName = `${safeName}__${tool2.name}`;
-        if (toolRegistry.has(namespacedName)) {
-          strapi.log.warn(`[${PLUGIN_ID$2}] Duplicate tool: ${namespacedName}`);
-          continue;
-        }
-        toolRegistry.register({ ...tool2, name: namespacedName });
-        count++;
-      }
+      const count = registerContributedTools(strapi, registry, pluginName, contributed);
       if (count > 0) {
         strapi.log.info(`[${PLUGIN_ID$2}] Registered ${count} tools from plugin: ${pluginName}`);
       }
@@ -1273,10 +1282,38 @@ const bootstrap = ({ strapi }) => {
       strapi.log.warn(`[${PLUGIN_ID$2}] Tool discovery failed for ${pluginName}: ${err}`);
     }
   }
-  plugin.createMcpServer = () => createMcpServer(strapi);
-  plugin.mcpSessions = /* @__PURE__ */ new Map();
-  strapi.log.info(`[${PLUGIN_ID$2}] MCP endpoint available at: /api/${PLUGIN_ID$2}/mcp`);
-};
+}
+function registerContributedTools(strapi, registry, pluginName, tools) {
+  const safeName = pluginName.replace(/[^a-zA-Z0-9_-]/g, "_");
+  let count = 0;
+  for (const tool2 of tools) {
+    if (!tool2.name || !tool2.execute || !tool2.schema) {
+      strapi.log.warn(`[${PLUGIN_ID$2}] Invalid tool from ${pluginName}: ${tool2.name || "unnamed"}`);
+      continue;
+    }
+    const namespacedName = `${safeName}__${tool2.name}`;
+    if (registry.has(namespacedName)) {
+      strapi.log.warn(`[${PLUGIN_ID$2}] Duplicate tool: ${namespacedName}`);
+      continue;
+    }
+    registry.register({ ...tool2, name: namespacedName });
+    count++;
+  }
+  return count;
+}
+function resolveAiToolsService(strapi, pluginName, pluginInstance) {
+  try {
+    const svc = strapi.plugin(pluginName)?.service?.("ai-tools");
+    if (svc) return svc;
+  } catch {
+  }
+  try {
+    const svc = pluginInstance.service?.("ai-tools");
+    if (svc) return svc;
+  } catch {
+  }
+  return null;
+}
 const PLUGIN_ID$1 = "ai-sdk";
 async function closeSession(strapi, sessionId, session) {
   try {
@@ -2624,15 +2661,21 @@ function createTools(strapi, context) {
   return tools;
 }
 const CONTENT_TOOLS = /* @__PURE__ */ new Set(["searchContent", "findOneContent", "aggregateContent"]);
-function createPublicTools(strapi, allowedContentTypes) {
+function createPublicTools(strapi, allowedContentTypes, publicToolSources) {
   const plugin = strapi.plugin("ai-sdk");
   const registry = plugin.toolRegistry;
   if (!registry) {
     throw new Error("Tool registry not initialized");
   }
   const allowed = new Set(allowedContentTypes);
+  const allowedSources = new Set(publicToolSources ?? []);
   const tools = {};
   for (const [name, def] of registry.getPublicSafe()) {
+    const sepIndex = name.indexOf("__");
+    if (sepIndex !== -1) {
+      const prefix = name.substring(0, sepIndex);
+      if (!allowedSources.has(prefix)) continue;
+    }
     if (CONTENT_TOOLS.has(name)) {
       tools[name] = tool({
         description: def.description,
@@ -2780,9 +2823,10 @@ ${lines.join("\n")}`;
       const maxSteps = publicConfig?.maxSteps ?? DEFAULT_PUBLIC_MAX_STEPS;
       const publicModel = publicConfig?.chatModel ?? DEFAULT_PUBLIC_CHAT_MODEL;
       const allowedContentTypes = publicConfig?.allowedContentTypes ?? [];
+      const publicToolSources = publicConfig?.publicToolSources;
       const trimmedMessages = messages.length > maxMessages ? messages.slice(-maxMessages) : messages;
       const modelMessages = await convertToModelMessages(trimmedMessages);
-      const tools = createPublicTools(strapi, allowedContentTypes);
+      const tools = createPublicTools(strapi, allowedContentTypes, publicToolSources);
       const toolsDescription = describeTools(tools);
       let system = composeSystemPrompt(config2, toolsDescription, options2?.system || DEFAULT_PUBLIC_PREAMBLE);
       try {

package/dist/server/src/lib/types.d.ts

@@ -23,6 +23,8 @@ export declare const DEFAULT_PUBLIC_CHAT_MODEL = "claude-haiku-4-5-20251001";
 export interface PublicChatConfig {
     /** Content type UIDs the public chat is allowed to query (e.g. ['api::article.article']) */
     allowedContentTypes?: string[];
+    /** Plugin tool source IDs allowed in public chat (e.g. ['yt-embeddings-strapi-plugin']). If omitted, no plugin tools are exposed. */
+    publicToolSources?: string[];
     /** Model to use for public chat (defaults to Haiku for lower cost & higher rate limits) */
     chatModel?: string;
     /** Max conversation messages for public chat (defaults to 10) */

package/dist/server/src/tool-logic/create-content.d.ts

@@ -0,0 +1,27 @@
+import type { Core } from '@strapi/strapi';
+import { z } from 'zod';
+export declare const createContentSchema: z.ZodObject<{
+    contentType: z.ZodString;
+    data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+    status: z.ZodOptional<z.ZodEnum<{
+        draft: "draft";
+        published: "published";
+    }>>;
+    locale: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export declare const createContentDescription = "Create a new document in any Strapi content type (articles, blog posts, pages, etc.). Use listContentTypes first to discover available content types and their fields.";
+export interface CreateContentParams {
+    contentType: string;
+    data: Record<string, unknown>;
+    status?: 'draft' | 'published';
+    locale?: string;
+}
+export interface CreateContentResult {
+    action: 'create';
+    document: any;
+}
+/**
+ * Core logic for creating content.
+ * Shared between AI SDK tool and MCP tool.
+ */
+export declare function createContent(strapi: Core.Strapi, params: CreateContentParams): Promise<CreateContentResult>;

package/dist/server/src/tool-logic/index.d.ts

@@ -2,8 +2,10 @@ export { listContentTypes, listContentTypesSchema, listContentTypesDescription }
 export type { ContentTypeSummary, ComponentSummary, RelationSummary, ListContentTypesResult } from './list-content-types';
 export { searchContent, searchContentSchema, searchContentDescription } from './search-content';
 export type { SearchContentParams, SearchContentResult } from './search-content';
-export { writeContent, writeContentSchema, writeContentDescription } from './write-content';
-export type { WriteContentParams, WriteContentResult } from './write-content';
+export { createContent, createContentSchema, createContentDescription } from './create-content';
+export type { CreateContentParams, CreateContentResult } from './create-content';
+export { updateContent, updateContentSchema, updateContentDescription } from './update-content';
+export type { UpdateContentParams, UpdateContentResult } from './update-content';
 export { sendEmail, sendEmailSchema, sendEmailDescription } from './send-email';
 export type { SendEmailParams, SendEmailResult } from './send-email';
 export { saveMemory, saveMemorySchema, saveMemoryDescription } from './save-memory';

package/dist/server/src/tool-logic/update-content.d.ts

@@ -0,0 +1,29 @@
+import type { Core } from '@strapi/strapi';
+import { z } from 'zod';
+export declare const updateContentSchema: z.ZodObject<{
+    contentType: z.ZodString;
+    documentId: z.ZodString;
+    data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+    status: z.ZodOptional<z.ZodEnum<{
+        draft: "draft";
+        published: "published";
+    }>>;
+    locale: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export declare const updateContentDescription = "Update an existing document in any Strapi content type. Use searchContent to find the document ID first.";
+export interface UpdateContentParams {
+    contentType: string;
+    documentId: string;
+    data: Record<string, unknown>;
+    status?: 'draft' | 'published';
+    locale?: string;
+}
+export interface UpdateContentResult {
+    action: 'update';
+    document: any;
+}
+/**
+ * Core logic for updating content.
+ * Shared between AI SDK tool and MCP tool.
+ */
+export declare function updateContent(strapi: Core.Strapi, params: UpdateContentParams): Promise<UpdateContentResult>;

package/dist/server/src/tool-logic/upload-media.d.ts

@@ -6,7 +6,7 @@ export declare const uploadMediaSchema: z.ZodObject<{
     caption: z.ZodOptional<z.ZodString>;
     alternativeText: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
-export declare const uploadMediaDescription = "Upload a media file from a URL to the Strapi media library. Returns the uploaded file data. To link media to a content type field, use writeContent with the file ID.";
+export declare const uploadMediaDescription = "Upload a media file from a URL to the Strapi media library. Returns the uploaded file data. To link media to a content type field, use createContent or updateContent with the file ID.";
 export interface UploadMediaParams {
     url: string;
     name?: string;

package/dist/server/src/tools/definitions/{write-content.d.ts → create-content.d.ts}

@@ -1,2 +1,2 @@
 import type { ToolDefinition } from '../../lib/tool-registry';
-export declare const writeContentTool: ToolDefinition;
+export declare const createContentTool: ToolDefinition;

package/dist/server/src/tools/definitions/update-content.d.ts

@@ -0,0 +1,2 @@
+import type { ToolDefinition } from '../../lib/tool-registry';
+export declare const updateContentTool: ToolDefinition;

package/dist/server/src/tools/index.d.ts

@@ -6,8 +6,9 @@ export declare function createTools(strapi: Core.Strapi, context?: ToolContext):
  * Create a restricted tool set for the public chat endpoint.
  * Only includes tools marked as publicSafe.
  * Content tools are wrapped to enforce allowedContentTypes.
+ * Plugin tools are only included if their source is in publicToolSources.
  */
-export declare function createPublicTools(strapi: Core.Strapi, allowedContentTypes: string[]): ToolSet;
+export declare function createPublicTools(strapi: Core.Strapi, allowedContentTypes: string[], publicToolSources?: string[]): ToolSet;
 /**
  * Build a system prompt section describing all available tools.
  * Reads the `description` from each tool definition so it stays in sync automatically.

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.7.3",
+  "version": "0.7.7",
   "keywords": [
     "strapi",
     "strapi-plugin",

package/dist/server/src/tool-logic/write-content.d.ts

@@ -1,34 +0,0 @@
-import type { Core } from '@strapi/strapi';
-import { z } from 'zod';
-export declare const writeContentSchema: z.ZodObject<{
-    contentType: z.ZodString;
-    action: z.ZodEnum<{
-        create: "create";
-        update: "update";
-    }>;
-    documentId: z.ZodOptional<z.ZodString>;
-    data: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-    status: z.ZodOptional<z.ZodEnum<{
-        draft: "draft";
-        published: "published";
-    }>>;
-    locale: z.ZodOptional<z.ZodString>;
-}, z.core.$strip>;
-export declare const writeContentDescription = "Create or update a document in any Strapi content type. Use listContentTypes first to discover the schema, and searchContent to find existing documents for updates.";
-export interface WriteContentParams {
-    contentType: string;
-    action: 'create' | 'update';
-    documentId?: string;
-    data: Record<string, unknown>;
-    status?: 'draft' | 'published';
-    locale?: string;
-}
-export interface WriteContentResult {
-    action: 'create' | 'update';
-    document: any;
-}
-/**
- * Core logic for creating/updating content.
- * Shared between AI SDK tool and MCP tool.
- */
-export declare function writeContent(strapi: Core.Strapi, params: WriteContentParams): Promise<WriteContentResult>;