strapi-plugin-ai-sdk 0.7.1 → 0.7.4

package/dist/_chunks/{App-DcU3uMiY.js → App-CEEsJsKL.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-D0dDUeEO.js");
+const index = require("./index-Cw2aiQ8K.js");
 const icons = require("@strapi/icons");
 const Markdown = require("react-markdown");
 const remarkGfm = require("remark-gfm");
@@ -101,11 +101,25 @@ function generateId() {
   return globalThis.crypto?.randomUUID?.() ?? `${Date.now()}-${Math.random()}`;
 }
 function toUIMessages(messages) {
-  return messages.map((message) => ({
-    id: message.id,
-    role: message.role,
-    parts: [{ type: "text", text: message.content }]
-  }));
+  return messages.map((message) => {
+    const parts = [];
+    if (message.content) {
+      parts.push({ type: "text", text: message.content });
+    }
+    if (message.toolCalls) {
+      for (const tc of message.toolCalls) {
+        parts.push({
+          type: `tool-${tc.toolName}`,
+          toolCallId: tc.toolCallId,
+          toolName: tc.toolName,
+          state: tc.output !== void 0 ? "output-available" : "input-available",
+          input: tc.input,
+          ...tc.output !== void 0 ? { output: tc.output } : {}
+        });
+      }
+    }
+    return { id: message.id, role: message.role, parts };
+  });
 }
 async function fetchChatStream(messages, enabledToolSources) {
   const token = getToken();

package/dist/_chunks/{App-Dwx6guC-.mjs → App-DCV7o6Hc.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-skxI4tiW.mjs";
+import { P as PLUGIN_ID } from "./index-BMrDQVQl.mjs";
 import { Plus, Trash, Sparkle, ArrowLeft, Pencil } from "@strapi/icons";
 import Markdown from "react-markdown";
 import remarkGfm from "remark-gfm";
@@ -95,11 +95,25 @@ function generateId() {
   return globalThis.crypto?.randomUUID?.() ?? `${Date.now()}-${Math.random()}`;
 }
 function toUIMessages(messages) {
-  return messages.map((message) => ({
-    id: message.id,
-    role: message.role,
-    parts: [{ type: "text", text: message.content }]
-  }));
+  return messages.map((message) => {
+    const parts = [];
+    if (message.content) {
+      parts.push({ type: "text", text: message.content });
+    }
+    if (message.toolCalls) {
+      for (const tc of message.toolCalls) {
+        parts.push({
+          type: `tool-${tc.toolName}`,
+          toolCallId: tc.toolCallId,
+          toolName: tc.toolName,
+          state: tc.output !== void 0 ? "output-available" : "input-available",
+          input: tc.input,
+          ...tc.output !== void 0 ? { output: tc.output } : {}
+        });
+      }
+    }
+    return { id: message.id, role: message.role, parts };
+  });
 }
 async function fetchChatStream(messages, enabledToolSources) {
   const token = getToken();

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

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

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

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

package/dist/admin/index.js

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

package/dist/admin/index.mjs

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

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.1",
+  "version": "0.7.4",
   "keywords": [
     "strapi",
     "strapi-plugin",