strapi-plugin-ai-sdk 0.6.10 → 0.7.1

package/README.md

@@ -286,9 +286,12 @@ The AI assistant has access to these tools. Tools marked as **public** are also
 |------|----------|-------------|
 | `listContentTypes` | `list_content_types` | List all Strapi content types and components with their fields and relations |
 | `searchContent` | `search_content` | Search and query any content type with filters, sorting, and pagination |
+| `aggregateContent` | `aggregate_content` | Count, group, and analyze content (faster than searchContent for analytics) |
 | `writeContent` | `write_content` | Create or update documents in any content type |
 | `sendEmail` | `send_email` | Send emails via the configured email provider (e.g. Resend) |
 
+Additionally, the AI SDK automatically discovers tools from other installed plugins (see [Extending the Plugin](#adding-tools-from-other-plugins-convention-based-discovery)). For example, with the mentions and embeddings plugins installed, the AI also has access to `searchMentions`, `semanticSearch`, `ragQuery`, and more.
+
 ### Tool Details
 
 **searchContent** parameters: `contentType` (required), `query`, `filters`, `fields`, `sort`, `page`, `pageSize` (max 50)
@@ -465,7 +468,151 @@ curl -N -X POST http://localhost:1337/api/ai-sdk/ask-stream \
 
 ## Extending the Plugin
 
-### Adding a Custom Tool
+### Adding Tools from Other Plugins (Convention-Based Discovery)
+
+Any Strapi plugin can contribute tools to the AI SDK by exposing an `ai-tools` service with a `getTools()` method. The AI SDK discovers these automatically at boot time -- no configuration required.
+
+```mermaid
+flowchart LR
+  subgraph "AI SDK Bootstrap"
+    B[Scan all plugins]
+  end
+
+  subgraph "Plugin A"
+    A1[ai-tools service] --> A2["getTools()"]
+  end
+
+  subgraph "Plugin B"
+    B1[ai-tools service] --> B2["getTools()"]
+  end
+
+  B --> A1
+  B --> B1
+  A2 --> R[ToolRegistry]
+  B2 --> R
+  R --> Chat[Admin Chat]
+  R --> MCP[MCP Server]
+  R --> Public[Public Chat]
+```
+
+#### How It Works
+
+1. On startup, the AI SDK scans every loaded plugin for an `ai-tools` service
+2. If found, it calls `getTools()` which returns an array of `ToolDefinition` objects
+3. Each tool is namespaced as `pluginName__toolName` (e.g., `octalens_mentions__searchMentions`) to prevent collisions
+4. Discovered tools are registered in the shared `ToolRegistry` alongside built-in tools
+5. All registered tools are available in admin chat, public chat (if `publicSafe: true`), and MCP
+
+#### Creating an `ai-tools` Service in Your Plugin
+
+**1. Define canonical tools** in `server/src/tools/`:
+
+```typescript
+// server/src/tools/my-tool.ts
+import { z } from 'zod';
+import type { Core } from '@strapi/strapi';
+
+const schema = z.object({
+  query: z.string().describe('Search query'),
+  limit: z.number().min(1).max(50).optional().default(10).describe('Max results'),
+});
+
+export const mySearchTool = {
+  name: 'mySearch',
+  description: 'Search my plugin data with relevance ranking.',
+  schema,
+  execute: async (args: z.infer<typeof schema>, strapi: Core.Strapi) => {
+    const validated = schema.parse(args);
+    const results = await strapi.documents('plugin::my-plugin.item' as any).findMany({
+      filters: { title: { $containsi: validated.query } },
+      limit: validated.limit,
+    });
+    return { results, total: results.length };
+  },
+  publicSafe: true, // available in public chat (read-only operations)
+};
+```
+
+**2. Create the `ai-tools` service:**
+
+```typescript
+// server/src/services/ai-tools.ts
+import type { Core } from '@strapi/strapi';
+import { tools } from '../tools';
+
+export default ({ strapi }: { strapi: Core.Strapi }) => ({
+  getTools() {
+    return tools;
+  },
+});
+```
+
+**3. Register the service:**
+
+```typescript
+// server/src/services/index.ts
+import myService from './my-service';
+import aiTools from './ai-tools';
+
+export default {
+  'my-service': myService,
+  'ai-tools': aiTools,
+};
+```
+
+That's it. The AI SDK will discover and register your tools on the next Strapi restart.
+
+#### ToolDefinition Interface
+
+```typescript
+interface ToolDefinition {
+  name: string;                    // camelCase, unique within your plugin
+  description: string;             // Clear description for the AI model
+  schema: z.ZodObject<any>;        // Zod schema for parameter validation
+  execute: (args: any, strapi: Core.Strapi, context?: ToolContext) => Promise<unknown>;
+  internal?: boolean;              // If true, hidden from MCP (AI chat only)
+  publicSafe?: boolean;            // If true, available in public/widget chat
+}
+```
+
+#### Canonical Architecture Pattern
+
+The recommended pattern is to define tools once in `server/src/tools/` and consume them from both the AI SDK service and MCP handlers:
+
+```mermaid
+flowchart TB
+  subgraph "Your Plugin"
+    T["server/src/tools/<br/>Canonical tool definitions<br/>(Zod schema + business logic)"]
+
+    subgraph "AI SDK Path"
+      S["services/ai-tools.ts<br/>getTools() → tools array"]
+    end
+
+    subgraph "MCP Path"
+      M["mcp/tools/*.ts<br/>Thin wrappers → MCP envelope"]
+      MS["mcp/server.ts"]
+    end
+
+    T --> S
+    T --> M
+    M --> MS
+  end
+
+  S -->|"Discovery"| SDK["AI SDK ToolRegistry"]
+  MS -->|"JSON-RPC"| Clients["Claude Desktop / Cursor"]
+```
+
+This eliminates duplication -- business logic lives in one place, and each consumer (AI SDK, MCP) uses a thin adapter.
+
+#### Real-World Examples
+
+Two plugins already use this pattern:
+
+**[strapi-octolens-mentions-plugin](../strapi-octolens-mentions-plugin/)** -- Contributes 4 tools: `searchMentions` (BM25 relevance search), `listMentions`, `getMention`, `updateMention`
+
+**[strapi-content-embeddings](../strapi-content-embeddings/)** -- Contributes 5 tools: `semanticSearch` (vector similarity), `ragQuery` (RAG), `listEmbeddings`, `getEmbedding`, `createEmbedding`
+
+### Adding a Custom Tool (Without a Plugin)
 
 **Option A: Inside the plugin** -- create files in `tools/definitions/` and `tool-logic/`, add to the `builtInTools` array.
 
@@ -578,7 +725,7 @@ Error response format:
 server/src/
   index.ts                    # Server entry point
   register.ts                 # Plugin register lifecycle
-  bootstrap.ts                # Initialize providers, tools, MCP
+  bootstrap.ts                # Initialize providers, tools, MCP, plugin tool discovery
   destroy.ts                  # Graceful shutdown
   config/index.ts             # Plugin config defaults
   guardrails/                 # Input safety middleware
@@ -650,6 +797,8 @@ STRAPI_TOKEN=your-api-token npm run test:guardrails
 ## Documentation
 
 - [Architecture](./docs/architecture.md) -- full system architecture, data flows, extension guides
+- [Plugin Tool Discovery](./docs/plugin-tool-discovery.md) -- cross-plugin tool discovery architecture and implementation
+- [Tool Standardization Spec](./docs/tool-standardization-spec.md) -- canonical tool format, Zod-first vs MCP-native comparison, portability
 - [Guardrails](./docs/guardrails.md) -- guardrail system, pattern lists, `beforeProcess` hook API
 - [Sending Emails with Resend](./docs/sending-emails-with-resend.md) -- Resend setup, email tool, domain verification
 

package/dist/_chunks/{App-BGIUzHMh.js → App-DcU3uMiY.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-BNk29VRc.js");
+const index = require("./index-D0dDUeEO.js");
 const icons = require("@strapi/icons");
 const Markdown = require("react-markdown");
 const remarkGfm = require("remark-gfm");
@@ -107,15 +107,17 @@ function toUIMessages(messages) {
     parts: [{ type: "text", text: message.content }]
   }));
 }
-async function fetchChatStream(messages) {
+async function fetchChatStream(messages, enabledToolSources) {
   const token = getToken();
+  const body = { messages: toUIMessages(messages) };
+  if (enabledToolSources) body.enabledToolSources = enabledToolSources;
   const response = await fetch(`${getBackendURL()}/${index.PLUGIN_ID}/chat`, {
     method: "POST",
     headers: {
       "Content-Type": "application/json",
       ...token ? { Authorization: `Bearer ${token}` } : {}
     },
-    body: JSON.stringify({ messages: toUIMessages(messages) })
+    body: JSON.stringify(body)
   });
   if (!response.ok) throw new Error(`Request failed: ${response.status}`);
   const reader = response.body?.getReader();
@@ -160,7 +162,7 @@ function useChat(options) {
       setIsLoading(true);
       setError(null);
       try {
-        const reader = await fetchChatStream([...messages, userMessage]);
+        const reader = await fetchChatStream([...messages, userMessage], options?.enabledToolSources);
         let streamStarted = false;
         const result = await readSSEStream(reader, {
           onTextDelta: (content) => {
@@ -392,6 +394,51 @@ function useMemories() {
   }, []);
   return { memories, loading, addMemory, editMemory, removeMemory, refresh: load };
 }
+const STORAGE_KEY = `${index.PLUGIN_ID}:enabledToolSources`;
+function useToolSources() {
+  const [sources, setSources] = react.useState([]);
+  const [enabledSources, setEnabledSources] = react.useState(() => {
+    try {
+      const stored = localStorage.getItem(STORAGE_KEY);
+      if (stored) return new Set(JSON.parse(stored));
+    } catch {
+    }
+    return /* @__PURE__ */ new Set();
+  });
+  const [loaded, setLoaded] = react.useState(false);
+  react.useEffect(() => {
+    const token = getToken();
+    fetch(`${getBackendURL()}/${index.PLUGIN_ID}/tool-sources`, {
+      headers: token ? { Authorization: `Bearer ${token}` } : {}
+    }).then((res) => res.json()).then((json) => {
+      const data = json.data ?? [];
+      setSources(data);
+      setEnabledSources((prev) => {
+        if (prev.size === 0) {
+          const all = new Set(data.filter((s) => s.id !== "built-in").map((s) => s.id));
+          localStorage.setItem(STORAGE_KEY, JSON.stringify([...all]));
+          return all;
+        }
+        return prev;
+      });
+      setLoaded(true);
+    }).catch(() => setLoaded(true));
+  }, []);
+  const toggleSource = react.useCallback((id) => {
+    setEnabledSources((prev) => {
+      const next = new Set(prev);
+      if (next.has(id)) next.delete(id);
+      else next.add(id);
+      localStorage.setItem(STORAGE_KEY, JSON.stringify([...next]));
+      return next;
+    });
+  }, []);
+  const enabledToolSources = react.useMemo(
+    () => loaded ? [...enabledSources] : void 0,
+    [enabledSources, loaded]
+  );
+  return { sources, enabledSources, enabledToolSources, toggleSource, loaded };
+}
 const SidebarRoot = styled__default.default.div`
   width: ${({ $open }) => $open ? "260px" : "0px"};
   min-width: ${({ $open }) => $open ? "260px" : "0px"};
@@ -618,6 +665,142 @@ function MemoryPanel({ memories, open, onDelete }) {
     ] })
   ] });
 }
+const Wrapper = styled__default.default.div`
+  position: relative;
+`;
+const IconBtn = styled__default.default.button`
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  width: 32px;
+  height: 32px;
+  border: 1px solid #dcdce4;
+  border-radius: 4px;
+  background: #ffffff;
+  color: #666687;
+  cursor: pointer;
+  flex-shrink: 0;
+
+  &:hover {
+    background: #f0f0f5;
+    color: #4945ff;
+    border-color: #4945ff;
+  }
+
+  svg {
+    width: 16px;
+    height: 16px;
+  }
+`;
+const Popover = styled__default.default.div`
+  position: absolute;
+  top: 40px;
+  left: 0;
+  z-index: 10;
+  width: 240px;
+  max-height: 360px;
+  overflow-y: auto;
+  background: #ffffff;
+  border: 1px solid #dcdce4;
+  border-radius: 4px;
+  box-shadow: 0 2px 12px rgba(33, 33, 52, 0.12);
+  padding: 8px 0;
+`;
+const SourceRow = styled__default.default.label`
+  display: flex;
+  flex-direction: column;
+  padding: 6px 12px;
+  cursor: pointer;
+  font-size: 13px;
+  color: #32324d;
+
+  &:hover {
+    background: #f6f6f9;
+  }
+`;
+const SourceMain = styled__default.default.div`
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+  gap: 8px;
+`;
+const Badge = styled__default.default.div`
+  font-size: 11px;
+  color: #a5a5ba;
+  margin-left: 24px;
+  margin-top: 1px;
+`;
+const Toggle = styled__default.default.input`
+  accent-color: #4945ff;
+  flex-shrink: 0;
+  width: 16px;
+  height: 16px;
+  margin: 0;
+`;
+const Header = styled__default.default.div`
+  padding: 6px 12px 4px;
+  font-size: 11px;
+  font-weight: 600;
+  color: #a5a5ba;
+  text-transform: uppercase;
+  letter-spacing: 0.5px;
+`;
+function ToolSourcePicker({ sources, enabledSources, onToggle }) {
+  const [open, setOpen] = react.useState(false);
+  const ref = react.useRef(null);
+  react.useEffect(() => {
+    if (!open) return;
+    function handleClick(e) {
+      if (ref.current && !ref.current.contains(e.target)) setOpen(false);
+    }
+    document.addEventListener("mousedown", handleClick);
+    return () => document.removeEventListener("mousedown", handleClick);
+  }, [open]);
+  const builtIn = sources.find((s) => s.id === "built-in");
+  const plugins = sources.filter((s) => s.id !== "built-in");
+  if (sources.length === 0) return null;
+  return /* @__PURE__ */ jsxRuntime.jsxs(Wrapper, { ref, children: [
+    /* @__PURE__ */ jsxRuntime.jsx(IconBtn, { onClick: () => setOpen((p) => !p), "aria-label": "Tool sources", children: /* @__PURE__ */ jsxRuntime.jsxs("svg", { viewBox: "0 0 16 16", fill: "none", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", children: [
+      /* @__PURE__ */ jsxRuntime.jsx("path", { d: "M9.5 2.5L13 6l-7 7H2.5v-3.5l7-7z" }),
+      /* @__PURE__ */ jsxRuntime.jsx("path", { d: "M8 4l4 4" })
+    ] }) }),
+    open && /* @__PURE__ */ jsxRuntime.jsxs(Popover, { children: [
+      builtIn && /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+        /* @__PURE__ */ jsxRuntime.jsx(Header, { children: "Built-in" }),
+        /* @__PURE__ */ jsxRuntime.jsxs(SourceRow, { children: [
+          /* @__PURE__ */ jsxRuntime.jsxs(SourceMain, { children: [
+            /* @__PURE__ */ jsxRuntime.jsx(Toggle, { type: "checkbox", checked: true, disabled: true }),
+            builtIn.label
+          ] }),
+          /* @__PURE__ */ jsxRuntime.jsxs(Badge, { children: [
+            builtIn.toolCount,
+            " tools"
+          ] })
+        ] })
+      ] }),
+      plugins.length > 0 && /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+        /* @__PURE__ */ jsxRuntime.jsx(Header, { children: "Plugins" }),
+        plugins.map((s) => /* @__PURE__ */ jsxRuntime.jsxs(SourceRow, { children: [
+          /* @__PURE__ */ jsxRuntime.jsxs(SourceMain, { children: [
+            /* @__PURE__ */ jsxRuntime.jsx(
+              Toggle,
+              {
+                type: "checkbox",
+                checked: enabledSources.has(s.id),
+                onChange: () => onToggle(s.id)
+              }
+            ),
+            s.label
+          ] }),
+          /* @__PURE__ */ jsxRuntime.jsxs(Badge, { children: [
+            s.toolCount,
+            " tools"
+          ] })
+        ] }, s.id))
+      ] })
+    ] })
+  ] });
+}
 const SCORE_LABELS = {
   1: "Negligible",
   2: "Minor",
@@ -1355,9 +1538,11 @@ function Chat() {
     removeConversation
   } = useConversations();
   const { memories, removeMemory, refresh: refreshMemories } = useMemories();
+  const { sources, enabledSources, enabledToolSources, toggleSource } = useToolSources();
   const { messages, sendMessage, isLoading, error } = useChat({
     initialMessages,
-    conversationId: activeId
+    conversationId: activeId,
+    enabledToolSources
   });
   react.useEffect(() => {
     if (prevIsLoadingRef.current && !isLoading && messages.length > 0) {
@@ -1431,6 +1616,14 @@ function Chat() {
             ] })
           }
         ),
+        /* @__PURE__ */ jsxRuntime.jsx(
+          ToolSourcePicker,
+          {
+            sources,
+            enabledSources,
+            onToggle: toggleSource
+          }
+        ),
         /* @__PURE__ */ jsxRuntime.jsx("div", { style: { flex: 1 } }),
         /* @__PURE__ */ jsxRuntime.jsx(
           ToggleSidebarBtn,

package/dist/_chunks/{App-v0CobEGM.mjs → App-Dwx6guC-.mjs}

@@ -1,10 +1,10 @@
-import { jsxs, jsx } from "react/jsx-runtime";
+import { jsxs, jsx, Fragment } from "react/jsx-runtime";
 import { Layouts, Page } from "@strapi/strapi/admin";
 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, forwardRef, useRef, useMemo } from "react";
+import { useState, useEffect, useCallback, useMemo, useRef, forwardRef } from "react";
 import styled from "styled-components";
-import { P as PLUGIN_ID } from "./index-CFO5UshL.mjs";
+import { P as PLUGIN_ID } from "./index-skxI4tiW.mjs";
 import { Plus, Trash, Sparkle, ArrowLeft, Pencil } from "@strapi/icons";
 import Markdown from "react-markdown";
 import remarkGfm from "remark-gfm";
@@ -101,15 +101,17 @@ function toUIMessages(messages) {
     parts: [{ type: "text", text: message.content }]
   }));
 }
-async function fetchChatStream(messages) {
+async function fetchChatStream(messages, enabledToolSources) {
   const token = getToken();
+  const body = { messages: toUIMessages(messages) };
+  if (enabledToolSources) body.enabledToolSources = enabledToolSources;
   const response = await fetch(`${getBackendURL()}/${PLUGIN_ID}/chat`, {
     method: "POST",
     headers: {
       "Content-Type": "application/json",
       ...token ? { Authorization: `Bearer ${token}` } : {}
     },
-    body: JSON.stringify({ messages: toUIMessages(messages) })
+    body: JSON.stringify(body)
   });
   if (!response.ok) throw new Error(`Request failed: ${response.status}`);
   const reader = response.body?.getReader();
@@ -154,7 +156,7 @@ function useChat(options) {
       setIsLoading(true);
       setError(null);
       try {
-        const reader = await fetchChatStream([...messages, userMessage]);
+        const reader = await fetchChatStream([...messages, userMessage], options?.enabledToolSources);
         let streamStarted = false;
         const result = await readSSEStream(reader, {
           onTextDelta: (content) => {
@@ -386,6 +388,51 @@ function useMemories() {
   }, []);
   return { memories, loading, addMemory, editMemory, removeMemory, refresh: load };
 }
+const STORAGE_KEY = `${PLUGIN_ID}:enabledToolSources`;
+function useToolSources() {
+  const [sources, setSources] = useState([]);
+  const [enabledSources, setEnabledSources] = useState(() => {
+    try {
+      const stored = localStorage.getItem(STORAGE_KEY);
+      if (stored) return new Set(JSON.parse(stored));
+    } catch {
+    }
+    return /* @__PURE__ */ new Set();
+  });
+  const [loaded, setLoaded] = useState(false);
+  useEffect(() => {
+    const token = getToken();
+    fetch(`${getBackendURL()}/${PLUGIN_ID}/tool-sources`, {
+      headers: token ? { Authorization: `Bearer ${token}` } : {}
+    }).then((res) => res.json()).then((json) => {
+      const data = json.data ?? [];
+      setSources(data);
+      setEnabledSources((prev) => {
+        if (prev.size === 0) {
+          const all = new Set(data.filter((s) => s.id !== "built-in").map((s) => s.id));
+          localStorage.setItem(STORAGE_KEY, JSON.stringify([...all]));
+          return all;
+        }
+        return prev;
+      });
+      setLoaded(true);
+    }).catch(() => setLoaded(true));
+  }, []);
+  const toggleSource = useCallback((id) => {
+    setEnabledSources((prev) => {
+      const next = new Set(prev);
+      if (next.has(id)) next.delete(id);
+      else next.add(id);
+      localStorage.setItem(STORAGE_KEY, JSON.stringify([...next]));
+      return next;
+    });
+  }, []);
+  const enabledToolSources = useMemo(
+    () => loaded ? [...enabledSources] : void 0,
+    [enabledSources, loaded]
+  );
+  return { sources, enabledSources, enabledToolSources, toggleSource, loaded };
+}
 const SidebarRoot = styled.div`
   width: ${({ $open }) => $open ? "260px" : "0px"};
   min-width: ${({ $open }) => $open ? "260px" : "0px"};
@@ -612,6 +659,142 @@ function MemoryPanel({ memories, open, onDelete }) {
     ] })
   ] });
 }
+const Wrapper = styled.div`
+  position: relative;
+`;
+const IconBtn = styled.button`
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  width: 32px;
+  height: 32px;
+  border: 1px solid #dcdce4;
+  border-radius: 4px;
+  background: #ffffff;
+  color: #666687;
+  cursor: pointer;
+  flex-shrink: 0;
+
+  &:hover {
+    background: #f0f0f5;
+    color: #4945ff;
+    border-color: #4945ff;
+  }
+
+  svg {
+    width: 16px;
+    height: 16px;
+  }
+`;
+const Popover = styled.div`
+  position: absolute;
+  top: 40px;
+  left: 0;
+  z-index: 10;
+  width: 240px;
+  max-height: 360px;
+  overflow-y: auto;
+  background: #ffffff;
+  border: 1px solid #dcdce4;
+  border-radius: 4px;
+  box-shadow: 0 2px 12px rgba(33, 33, 52, 0.12);
+  padding: 8px 0;
+`;
+const SourceRow = styled.label`
+  display: flex;
+  flex-direction: column;
+  padding: 6px 12px;
+  cursor: pointer;
+  font-size: 13px;
+  color: #32324d;
+
+  &:hover {
+    background: #f6f6f9;
+  }
+`;
+const SourceMain = styled.div`
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+  gap: 8px;
+`;
+const Badge = styled.div`
+  font-size: 11px;
+  color: #a5a5ba;
+  margin-left: 24px;
+  margin-top: 1px;
+`;
+const Toggle = styled.input`
+  accent-color: #4945ff;
+  flex-shrink: 0;
+  width: 16px;
+  height: 16px;
+  margin: 0;
+`;
+const Header = styled.div`
+  padding: 6px 12px 4px;
+  font-size: 11px;
+  font-weight: 600;
+  color: #a5a5ba;
+  text-transform: uppercase;
+  letter-spacing: 0.5px;
+`;
+function ToolSourcePicker({ sources, enabledSources, onToggle }) {
+  const [open, setOpen] = useState(false);
+  const ref = useRef(null);
+  useEffect(() => {
+    if (!open) return;
+    function handleClick(e) {
+      if (ref.current && !ref.current.contains(e.target)) setOpen(false);
+    }
+    document.addEventListener("mousedown", handleClick);
+    return () => document.removeEventListener("mousedown", handleClick);
+  }, [open]);
+  const builtIn = sources.find((s) => s.id === "built-in");
+  const plugins = sources.filter((s) => s.id !== "built-in");
+  if (sources.length === 0) return null;
+  return /* @__PURE__ */ jsxs(Wrapper, { ref, children: [
+    /* @__PURE__ */ jsx(IconBtn, { onClick: () => setOpen((p) => !p), "aria-label": "Tool sources", children: /* @__PURE__ */ jsxs("svg", { viewBox: "0 0 16 16", fill: "none", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", children: [
+      /* @__PURE__ */ jsx("path", { d: "M9.5 2.5L13 6l-7 7H2.5v-3.5l7-7z" }),
+      /* @__PURE__ */ jsx("path", { d: "M8 4l4 4" })
+    ] }) }),
+    open && /* @__PURE__ */ jsxs(Popover, { children: [
+      builtIn && /* @__PURE__ */ jsxs(Fragment, { children: [
+        /* @__PURE__ */ jsx(Header, { children: "Built-in" }),
+        /* @__PURE__ */ jsxs(SourceRow, { children: [
+          /* @__PURE__ */ jsxs(SourceMain, { children: [
+            /* @__PURE__ */ jsx(Toggle, { type: "checkbox", checked: true, disabled: true }),
+            builtIn.label
+          ] }),
+          /* @__PURE__ */ jsxs(Badge, { children: [
+            builtIn.toolCount,
+            " tools"
+          ] })
+        ] })
+      ] }),
+      plugins.length > 0 && /* @__PURE__ */ jsxs(Fragment, { children: [
+        /* @__PURE__ */ jsx(Header, { children: "Plugins" }),
+        plugins.map((s) => /* @__PURE__ */ jsxs(SourceRow, { children: [
+          /* @__PURE__ */ jsxs(SourceMain, { children: [
+            /* @__PURE__ */ jsx(
+              Toggle,
+              {
+                type: "checkbox",
+                checked: enabledSources.has(s.id),
+                onChange: () => onToggle(s.id)
+              }
+            ),
+            s.label
+          ] }),
+          /* @__PURE__ */ jsxs(Badge, { children: [
+            s.toolCount,
+            " tools"
+          ] })
+        ] }, s.id))
+      ] })
+    ] })
+  ] });
+}
 const SCORE_LABELS = {
   1: "Negligible",
   2: "Minor",
@@ -1349,9 +1532,11 @@ function Chat() {
     removeConversation
   } = useConversations();
   const { memories, removeMemory, refresh: refreshMemories } = useMemories();
+  const { sources, enabledSources, enabledToolSources, toggleSource } = useToolSources();
   const { messages, sendMessage, isLoading, error } = useChat({
     initialMessages,
-    conversationId: activeId
+    conversationId: activeId,
+    enabledToolSources
   });
   useEffect(() => {
     if (prevIsLoadingRef.current && !isLoading && messages.length > 0) {
@@ -1425,6 +1610,14 @@ function Chat() {
             ] })
           }
         ),
+        /* @__PURE__ */ jsx(
+          ToolSourcePicker,
+          {
+            sources,
+            enabledSources,
+            onToggle: toggleSource
+          }
+        ),
         /* @__PURE__ */ jsx("div", { style: { flex: 1 } }),
         /* @__PURE__ */ jsx(
           ToggleSidebarBtn,

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

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

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

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

package/dist/admin/index.js

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

package/dist/admin/index.mjs

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

package/dist/admin/src/components/ToolSourcePicker.d.ts

@@ -0,0 +1,8 @@
+import type { ToolSource } from '../hooks/useToolSources';
+interface Props {
+    sources: ToolSource[];
+    enabledSources: Set<string>;
+    onToggle: (id: string) => void;
+}
+export declare function ToolSourcePicker({ sources, enabledSources, onToggle }: Props): import("react/jsx-runtime").JSX.Element | null;
+export {};

package/dist/admin/src/hooks/useChat.d.ts

@@ -15,6 +15,7 @@ export interface UseChatOptions {
     initialMessages?: Message[];
     conversationId?: string | null;
     onStreamStart?: () => void;
+    enabledToolSources?: string[];
 }
 export declare function useChat(options?: UseChatOptions): {
     messages: Message[];

package/dist/admin/src/hooks/useToolSources.d.ts

@@ -0,0 +1,13 @@
+export interface ToolSource {
+    id: string;
+    label: string;
+    toolCount: number;
+    tools: string[];
+}
+export declare function useToolSources(): {
+    sources: ToolSource[];
+    enabledSources: Set<string>;
+    enabledToolSources: string[] | undefined;
+    toggleSource: (id: string) => void;
+    loaded: boolean;
+};

package/dist/server/index.js

@@ -31,7 +31,7 @@ function _interopNamespace(e) {
 const fs__namespace = /* @__PURE__ */ _interopNamespace(fs$1);
 const path__namespace = /* @__PURE__ */ _interopNamespace(path$1);
 function toSnakeCase(str) {
-  return str.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
+  return str.replace(/:/g, "__").replace(/-/g, "_").replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
 }
 function createMcpServer(strapi) {
   const plugin = strapi.plugin("ai-sdk");
@@ -204,6 +204,29 @@ class ToolRegistry {
     }
     return result;
   }
+  /** Returns metadata about tool sources grouped by plugin prefix */
+  getToolSources() {
+    const groups = /* @__PURE__ */ new Map();
+    for (const name of this.tools.keys()) {
+      const sepIndex = name.indexOf("__");
+      if (sepIndex === -1) {
+        const list = groups.get("built-in") ?? [];
+        list.push(name);
+        groups.set("built-in", list);
+      } else {
+        const prefix = name.substring(0, sepIndex);
+        const list = groups.get(prefix) ?? [];
+        list.push(name);
+        groups.set(prefix, list);
+      }
+    }
+    return Array.from(groups.entries()).map(([id, tools]) => ({
+      id,
+      label: id === "built-in" ? "Built-in Tools" : id,
+      toolCount: tools.length,
+      tools
+    }));
+  }
   /** Only tools marked safe for unauthenticated public chat */
   getPublicSafe() {
     const result = /* @__PURE__ */ new Map();
@@ -1225,6 +1248,51 @@ const bootstrap = ({ strapi }) => {
     toolRegistry.register(tool);
   }
   plugin.toolRegistry = toolRegistry;
+  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 {
+        }
+      }
+      if (!aiToolsService?.getTools) {
+        strapi.log.debug(`[${PLUGIN_ID$2}] No ai-tools service on plugin: ${pluginName}`);
+        continue;
+      }
+      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++;
+      }
+      if (count > 0) {
+        strapi.log.info(`[${PLUGIN_ID$2}] Registered ${count} tools from plugin: ${pluginName}`);
+      }
+    } catch (err) {
+      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`);
@@ -1537,7 +1605,7 @@ function validateBody(ctx) {
   return { prompt, system };
 }
 function validateChatBody(ctx) {
-  const { messages, system } = ctx.request.body;
+  const { messages, system, enabledToolSources } = ctx.request.body;
   if (!messages || !Array.isArray(messages) || messages.length === 0) {
     ctx.badRequest("messages is required and must be a non-empty array");
     return null;
@@ -1546,7 +1614,11 @@ function validateChatBody(ctx) {
     ctx.badRequest("system must be a string if provided");
     return null;
   }
-  return { messages, system };
+  if (enabledToolSources !== void 0 && (!Array.isArray(enabledToolSources) || !enabledToolSources.every((s) => typeof s === "string"))) {
+    ctx.badRequest("enabledToolSources must be an array of strings if provided");
+    return null;
+  }
+  return { messages, system, enabledToolSources };
 }
 function createSSEStream(ctx) {
   ctx.set({
@@ -1613,7 +1685,11 @@ const controller = ({ strapi }) => ({
     const service2 = getService(strapi, ctx);
     if (!service2) return;
     const adminUserId = ctx.state?.user?.id;
-    const result = await service2.chat(body.messages, { system: body.system, adminUserId });
+    const result = await service2.chat(body.messages, {
+      system: body.system,
+      adminUserId,
+      enabledToolSources: body.enabledToolSources
+    });
     const response = result.toUIMessageStreamResponse();
     ctx.status = 200;
     ctx.set("Content-Type", "text/event-stream; charset=utf-8");
@@ -1641,6 +1717,15 @@ const controller = ({ strapi }) => ({
     ctx.set("x-vercel-ai-ui-message-stream", "v1");
     ctx.body = node_stream.Readable.fromWeb(response.body);
   },
+  async getToolSources(ctx) {
+    const plugin = strapi.plugin("ai-sdk");
+    const registry = plugin.toolRegistry;
+    if (!registry) {
+      ctx.badRequest("Tool registry not initialized");
+      return;
+    }
+    ctx.body = { data: registry.getToolSources() };
+  },
   async serveWidget(ctx) {
     const pluginRoot = path__namespace.resolve(__dirname, "..", "..");
     const widgetPath = path__namespace.join(pluginRoot, "dist", "widget", "widget.js");
@@ -2411,6 +2496,12 @@ const contentAPIRoutes = {
 const adminAPIRoutes = {
   type: "admin",
   routes: [
+    {
+      method: "GET",
+      path: "/tool-sources",
+      handler: "controller.getToolSources",
+      config: { policies: [] }
+    },
     {
       method: "POST",
       path: "/chat",
@@ -2534,8 +2625,16 @@ function createTools(strapi, context) {
   if (!registry) {
     throw new Error("Tool registry not initialized");
   }
+  const enabledSources = context?.enabledToolSources;
   const tools = {};
   for (const [name, def] of registry.getAll()) {
+    if (enabledSources) {
+      const sepIndex = name.indexOf("__");
+      if (sepIndex !== -1) {
+        const prefix = name.substring(0, sepIndex);
+        if (!enabledSources.includes(prefix)) continue;
+      }
+    }
     tools[name] = ai.tool({
       description: def.description,
       inputSchema: ai.zodSchema(def.schema),
@@ -2598,6 +2697,8 @@ const DEFAULT_PREAMBLE = `You are a Strapi CMS assistant. Use your tools to fulf
 
 For analytics and counting questions (e.g. "how many", "count", "distribution", "trends", "breakdown"), use the aggregateContent tool instead of searchContent — it is faster and purpose-built for these queries. Present analytics results as markdown tables. After showing analytics results, suggest 2-3 follow-up questions the user might find useful under a "**You might also want to know:**" heading.
 
+Plugin tools: When specialized tools from plugins are available (listed below), ALWAYS prefer them over the generic searchContent tool. For example, use searchMentions for social mentions (it has BM25 relevance scoring), semanticSearch for vector similarity search across embeddings, and ragQuery for question-answering grounded in embedded content. Only fall back to searchContent when no specialized tool covers the query.
+
 Strapi filter syntax for searchContent and aggregateContent:
 - Scalar fields: { title: { $containsi: "hello" } }
 - Relation (manyToOne): { author: { name: { $eq: "John" } } }
@@ -2657,7 +2758,7 @@ const service = ({ strapi }) => {
       const maxSteps = config2?.maxSteps ?? DEFAULT_MAX_STEPS;
       const trimmedMessages = messages.length > maxMessages ? messages.slice(-maxMessages) : messages;
       const modelMessages = await ai.convertToModelMessages(trimmedMessages);
-      const tools = createTools(strapi, { adminUserId: options2?.adminUserId });
+      const tools = createTools(strapi, { adminUserId: options2?.adminUserId, enabledToolSources: options2?.enabledToolSources });
       const toolsDescription = describeTools(tools);
       let system = composeSystemPrompt(config2, toolsDescription, options2?.system);
       if (options2?.adminUserId) {

package/dist/server/index.mjs

@@ -11,7 +11,7 @@ import * as path from "node:path";
 import { randomUUID } from "node:crypto";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
 function toSnakeCase(str) {
-  return str.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
+  return str.replace(/:/g, "__").replace(/-/g, "_").replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
 }
 function createMcpServer(strapi) {
   const plugin = strapi.plugin("ai-sdk");
@@ -184,6 +184,29 @@ class ToolRegistry {
     }
     return result;
   }
+  /** Returns metadata about tool sources grouped by plugin prefix */
+  getToolSources() {
+    const groups = /* @__PURE__ */ new Map();
+    for (const name of this.tools.keys()) {
+      const sepIndex = name.indexOf("__");
+      if (sepIndex === -1) {
+        const list = groups.get("built-in") ?? [];
+        list.push(name);
+        groups.set("built-in", list);
+      } else {
+        const prefix = name.substring(0, sepIndex);
+        const list = groups.get(prefix) ?? [];
+        list.push(name);
+        groups.set(prefix, list);
+      }
+    }
+    return Array.from(groups.entries()).map(([id, tools]) => ({
+      id,
+      label: id === "built-in" ? "Built-in Tools" : id,
+      toolCount: tools.length,
+      tools
+    }));
+  }
   /** Only tools marked safe for unauthenticated public chat */
   getPublicSafe() {
     const result = /* @__PURE__ */ new Map();
@@ -1205,6 +1228,51 @@ const bootstrap = ({ strapi }) => {
     toolRegistry.register(tool2);
   }
   plugin.toolRegistry = toolRegistry;
+  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 {
+        }
+      }
+      if (!aiToolsService?.getTools) {
+        strapi.log.debug(`[${PLUGIN_ID$2}] No ai-tools service on plugin: ${pluginName}`);
+        continue;
+      }
+      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++;
+      }
+      if (count > 0) {
+        strapi.log.info(`[${PLUGIN_ID$2}] Registered ${count} tools from plugin: ${pluginName}`);
+      }
+    } catch (err) {
+      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`);
@@ -1517,7 +1585,7 @@ function validateBody(ctx) {
   return { prompt, system };
 }
 function validateChatBody(ctx) {
-  const { messages, system } = ctx.request.body;
+  const { messages, system, enabledToolSources } = ctx.request.body;
   if (!messages || !Array.isArray(messages) || messages.length === 0) {
     ctx.badRequest("messages is required and must be a non-empty array");
     return null;
@@ -1526,7 +1594,11 @@ function validateChatBody(ctx) {
     ctx.badRequest("system must be a string if provided");
     return null;
   }
-  return { messages, system };
+  if (enabledToolSources !== void 0 && (!Array.isArray(enabledToolSources) || !enabledToolSources.every((s) => typeof s === "string"))) {
+    ctx.badRequest("enabledToolSources must be an array of strings if provided");
+    return null;
+  }
+  return { messages, system, enabledToolSources };
 }
 function createSSEStream(ctx) {
   ctx.set({
@@ -1593,7 +1665,11 @@ const controller = ({ strapi }) => ({
     const service2 = getService(strapi, ctx);
     if (!service2) return;
     const adminUserId = ctx.state?.user?.id;
-    const result = await service2.chat(body.messages, { system: body.system, adminUserId });
+    const result = await service2.chat(body.messages, {
+      system: body.system,
+      adminUserId,
+      enabledToolSources: body.enabledToolSources
+    });
     const response = result.toUIMessageStreamResponse();
     ctx.status = 200;
     ctx.set("Content-Type", "text/event-stream; charset=utf-8");
@@ -1621,6 +1697,15 @@ const controller = ({ strapi }) => ({
     ctx.set("x-vercel-ai-ui-message-stream", "v1");
     ctx.body = Readable.fromWeb(response.body);
   },
+  async getToolSources(ctx) {
+    const plugin = strapi.plugin("ai-sdk");
+    const registry = plugin.toolRegistry;
+    if (!registry) {
+      ctx.badRequest("Tool registry not initialized");
+      return;
+    }
+    ctx.body = { data: registry.getToolSources() };
+  },
   async serveWidget(ctx) {
     const pluginRoot = path.resolve(__dirname, "..", "..");
     const widgetPath = path.join(pluginRoot, "dist", "widget", "widget.js");
@@ -2391,6 +2476,12 @@ const contentAPIRoutes = {
 const adminAPIRoutes = {
   type: "admin",
   routes: [
+    {
+      method: "GET",
+      path: "/tool-sources",
+      handler: "controller.getToolSources",
+      config: { policies: [] }
+    },
     {
       method: "POST",
       path: "/chat",
@@ -2514,8 +2605,16 @@ function createTools(strapi, context) {
   if (!registry) {
     throw new Error("Tool registry not initialized");
   }
+  const enabledSources = context?.enabledToolSources;
   const tools = {};
   for (const [name, def] of registry.getAll()) {
+    if (enabledSources) {
+      const sepIndex = name.indexOf("__");
+      if (sepIndex !== -1) {
+        const prefix = name.substring(0, sepIndex);
+        if (!enabledSources.includes(prefix)) continue;
+      }
+    }
     tools[name] = tool({
       description: def.description,
       inputSchema: zodSchema(def.schema),
@@ -2578,6 +2677,8 @@ const DEFAULT_PREAMBLE = `You are a Strapi CMS assistant. Use your tools to fulf
 
 For analytics and counting questions (e.g. "how many", "count", "distribution", "trends", "breakdown"), use the aggregateContent tool instead of searchContent — it is faster and purpose-built for these queries. Present analytics results as markdown tables. After showing analytics results, suggest 2-3 follow-up questions the user might find useful under a "**You might also want to know:**" heading.
 
+Plugin tools: When specialized tools from plugins are available (listed below), ALWAYS prefer them over the generic searchContent tool. For example, use searchMentions for social mentions (it has BM25 relevance scoring), semanticSearch for vector similarity search across embeddings, and ragQuery for question-answering grounded in embedded content. Only fall back to searchContent when no specialized tool covers the query.
+
 Strapi filter syntax for searchContent and aggregateContent:
 - Scalar fields: { title: { $containsi: "hello" } }
 - Relation (manyToOne): { author: { name: { $eq: "John" } } }
@@ -2637,7 +2738,7 @@ const service = ({ strapi }) => {
       const maxSteps = config2?.maxSteps ?? DEFAULT_MAX_STEPS;
       const trimmedMessages = messages.length > maxMessages ? messages.slice(-maxMessages) : messages;
       const modelMessages = await convertToModelMessages(trimmedMessages);
-      const tools = createTools(strapi, { adminUserId: options2?.adminUserId });
+      const tools = createTools(strapi, { adminUserId: options2?.adminUserId, enabledToolSources: options2?.enabledToolSources });
       const toolsDescription = describeTools(tools);
       let system = composeSystemPrompt(config2, toolsDescription, options2?.system);
       if (options2?.adminUserId) {

package/dist/server/src/controllers/controller.d.ts

@@ -14,6 +14,7 @@ declare const controller: ({ strapi }: {
      * Public chat endpoint - restricted tools, public memories, no admin auth
      */
     publicChat(ctx: Context): Promise<void>;
+    getToolSources(ctx: Context): Promise<void>;
     serveWidget(ctx: Context): Promise<void>;
 };
 export default controller;

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

@@ -7,6 +7,7 @@ declare const _default: {
         askStream(ctx: import("koa").Context): Promise<void>;
         chat(ctx: import("koa").Context): Promise<void>;
         publicChat(ctx: import("koa").Context): Promise<void>;
+        getToolSources(ctx: import("koa").Context): Promise<void>;
         serveWidget(ctx: import("koa").Context): Promise<void>;
     };
     mcp: ({ strapi }: {

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

@@ -45,6 +45,7 @@ declare const _default: {
             askStream(ctx: import("koa").Context): Promise<void>;
             chat(ctx: import("koa").Context): Promise<void>;
             publicChat(ctx: import("koa").Context): Promise<void>;
+            getToolSources(ctx: import("koa").Context): Promise<void>;
             serveWidget(ctx: import("koa").Context): Promise<void>;
         };
         mcp: ({ strapi }: {
@@ -118,7 +119,7 @@ declare const _default: {
                 handler: string;
                 config: {
                     policies: any[];
-                    middlewares: string[];
+                    middlewares?: undefined;
                 };
             } | {
                 method: string;
@@ -126,7 +127,7 @@ declare const _default: {
                 handler: string;
                 config: {
                     policies: any[];
-                    middlewares?: undefined;
+                    middlewares: string[];
                 };
             })[];
         };
@@ -144,6 +145,7 @@ declare const _default: {
             chat(messages: import("ai").UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[], options?: {
                 system?: string;
                 adminUserId?: number;
+                enabledToolSources?: string[];
             }): Promise<import("./lib/ai-provider").StreamTextRawResult>;
             publicChat(messages: import("ai").UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[], options?: {
                 system?: string;

package/dist/server/src/lib/tool-registry.d.ts

@@ -2,6 +2,7 @@ import type { Core } from '@strapi/strapi';
 import type { z } from 'zod';
 export interface ToolContext {
     adminUserId?: number;
+    enabledToolSources?: string[];
 }
 export interface ToolDefinition {
     name: string;
@@ -13,6 +14,8 @@ export interface ToolDefinition {
     /** If true, tool is safe for unauthenticated public chat (read-only) */
     publicSafe?: boolean;
 }
+/** Type alias for external plugin authors to import when contributing tools */
+export type AiToolContribution = ToolDefinition;
 export declare class ToolRegistry {
     private readonly tools;
     register(def: ToolDefinition): void;
@@ -23,6 +26,13 @@ export declare class ToolRegistry {
     getAll(): Map<string, ToolDefinition>;
     /** Only tools that should be exposed via MCP (non-internal) */
     getPublic(): Map<string, ToolDefinition>;
+    /** Returns metadata about tool sources grouped by plugin prefix */
+    getToolSources(): Array<{
+        id: string;
+        label: string;
+        toolCount: number;
+        tools: string[];
+    }>;
     /** Only tools marked safe for unauthenticated public chat */
     getPublicSafe(): Map<string, ToolDefinition>;
 }

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

@@ -20,6 +20,7 @@ export declare function validateBody(ctx: Context): {
 export declare function validateChatBody(ctx: Context): {
     messages: UIMessage[];
     system?: string;
+    enabledToolSources?: string[];
 } | null;
 /**
  * Setup SSE stream with proper headers

package/dist/server/src/routes/admin/index.d.ts

@@ -6,7 +6,7 @@ declare const _default: {
         handler: string;
         config: {
             policies: any[];
-            middlewares: string[];
+            middlewares?: undefined;
         };
     } | {
         method: string;
@@ -14,7 +14,7 @@ declare const _default: {
         handler: string;
         config: {
             policies: any[];
-            middlewares?: undefined;
+            middlewares: string[];
         };
     })[];
 };

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

@@ -27,7 +27,7 @@ declare const routes: {
             handler: string;
             config: {
                 policies: any[];
-                middlewares: string[];
+                middlewares?: undefined;
             };
         } | {
             method: string;
@@ -35,7 +35,7 @@ declare const routes: {
             handler: string;
             config: {
                 policies: any[];
-                middlewares?: undefined;
+                middlewares: string[];
             };
         })[];
     };

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

@@ -11,6 +11,7 @@ declare const _default: {
         chat(messages: import("ai").UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[], options?: {
             system?: string;
             adminUserId?: number;
+            enabledToolSources?: string[];
         }): Promise<import("../lib/ai-provider").StreamTextRawResult>;
         publicChat(messages: import("ai").UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[], options?: {
             system?: string;

package/dist/server/src/services/service.d.ts

@@ -17,6 +17,7 @@ declare const service: ({ strapi }: {
     chat(messages: UIMessage[], options?: {
         system?: string;
         adminUserId?: number;
+        enabledToolSources?: string[];
     }): Promise<StreamTextRawResult>;
     /**
      * Public chat - restricted tools, public memories, no admin auth

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.6.10",
+  "version": "0.7.1",
   "keywords": [
     "strapi",
     "strapi-plugin",