strapi-plugin-ai-sdk 0.7.3 → 0.7.4

package/dist/server/index.js

@@ -2644,15 +2644,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 +2806,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

@@ -2624,15 +2624,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 +2786,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/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.4",
   "keywords": [
     "strapi",
     "strapi-plugin",