strapi-plugin-ai-sdk 0.2.0 → 0.5.0

package/dist/server/index.js

@@ -3,6 +3,9 @@ const anthropic = require("@ai-sdk/anthropic");
 const mcp_js = require("@modelcontextprotocol/sdk/server/mcp.js");
 const ai = require("ai");
 const zod = require("zod");
+const fs = require("fs");
+const os = require("os");
+const path = require("path");
 const node_stream = require("node:stream");
 const node_crypto = require("node:crypto");
 const streamableHttp_js = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
@@ -144,6 +147,8 @@ function createTTSRegistry() {
 }
 const DEFAULT_MODEL = "claude-sonnet-4-20250514";
 const DEFAULT_TEMPERATURE = 0.7;
+const DEFAULT_MAX_OUTPUT_TOKENS = 8192;
+const DEFAULT_MAX_CONVERSATION_MESSAGES = 40;
 function isPromptInput(input) {
   return "prompt" in input;
 }
@@ -337,20 +342,46 @@ async function listContentTypes(strapi) {
   return { contentTypes: contentTypes2, components };
 }
 const MAX_PAGE_SIZE = 50;
+const LARGE_CONTENT_FIELDS = ["content", "blocks", "body", "richText", "markdown", "html"];
 const searchContentSchema = zod.z.object({
   contentType: zod.z.string().describe(
     'The content type UID to search, e.g. "api::article.article" or "plugin::users-permissions.user"'
   ),
   query: zod.z.string().optional().describe("Full-text search query string (searches across all searchable text fields)"),
   filters: zod.z.record(zod.z.string(), zod.z.unknown()).optional().describe('Strapi filter object, e.g. { username: { $containsi: "john" } }'),
-  fields: zod.z.array(zod.z.string()).optional().describe("Specific fields to return. If omitted, returns all fields."),
+  fields: zod.z.array(zod.z.string()).optional().describe("Specific fields to return. If omitted, returns all fields (large content fields stripped unless includeContent is true)."),
   sort: zod.z.string().optional().describe('Sort order, e.g. "createdAt:desc"'),
   page: zod.z.number().optional().default(1).describe("Page number (starts at 1)"),
-  pageSize: zod.z.number().optional().default(10).describe(`Results per page (max ${MAX_PAGE_SIZE})`)
+  pageSize: zod.z.number().optional().default(10).describe(`Results per page (max ${MAX_PAGE_SIZE})`),
+  status: zod.z.enum(["draft", "published"]).optional().describe("Filter by document status. Published documents have a publishedAt date. If omitted, Strapi returns drafts by default."),
+  locale: zod.z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"'),
+  populate: zod.z.union([zod.z.string(), zod.z.array(zod.z.string()), zod.z.record(zod.z.string(), zod.z.unknown())]).optional().describe('Relations to populate. Defaults to "*" (all). Can be a string, array, or object.'),
+  includeContent: zod.z.boolean().optional().default(false).describe("When true, includes large content fields (content, blocks, body, etc.) in results. Default false to reduce context size.")
 });
-const searchContentDescription = "Search and query any Strapi content type. Use listContentTypes first to discover available content types and their fields, then use this tool to query specific collections.";
+const searchContentDescription = "Search and query any Strapi content type. Use listContentTypes first to discover available content types and their fields, then use this tool to query specific collections. By default, large content fields are stripped from results — set includeContent to true or use fields to get full content.";
+function stripLargeFields(obj) {
+  const stripped = {};
+  for (const [key, value] of Object.entries(obj)) {
+    if (!LARGE_CONTENT_FIELDS.includes(key)) {
+      stripped[key] = value;
+    }
+  }
+  return stripped;
+}
 async function searchContent(strapi, params) {
-  const { contentType, query, filters, fields, sort, page = 1, pageSize = 10 } = params;
+  const {
+    contentType,
+    query,
+    filters,
+    fields,
+    sort,
+    page = 1,
+    pageSize = 10,
+    status,
+    locale,
+    populate = "*",
+    includeContent = false
+  } = params;
   if (!strapi.contentTypes[contentType]) {
     throw new Error(`Content type "${contentType}" does not exist.`);
   }
@@ -360,16 +391,22 @@ async function searchContent(strapi, params) {
     ...filters ? { filters } : {},
     ...fields ? { fields } : {},
     ...sort ? { sort } : {},
+    ...status ? { status } : {},
+    ...locale ? { locale } : {},
     page,
     pageSize: clampedPageSize,
-    populate: "*"
+    populate
   });
   const total = await strapi.documents(contentType).count({
     ...query ? { _q: query } : {},
-    ...filters ? { filters } : {}
+    ...filters ? { filters } : {},
+    ...status ? { status } : {},
+    ...locale ? { locale } : {}
   });
+  const shouldStrip = !includeContent && !fields;
+  const processedResults = shouldStrip ? results.map((doc) => stripLargeFields(doc)) : results;
   return {
-    results,
+    results: processedResults,
     pagination: {
       page,
       pageSize: clampedPageSize,
@@ -382,11 +419,12 @@ const writeContentSchema = zod.z.object({
   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.")
+  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 } = params;
+  const { contentType, action, documentId, data, status, locale } = params;
   if (!strapi.contentTypes[contentType]) {
     throw new Error(`Content type "${contentType}" does not exist.`);
   }
@@ -398,14 +436,25 @@ async function writeContent(strapi, params) {
     const document2 = await docs.create({
       data,
       ...status ? { status } : {},
+      ...locale ? { locale } : {},
       populate: "*"
     });
     return { action: "create", document: document2 };
   }
+  const existing = await docs.findOne({
+    documentId,
+    ...locale ? { locale } : {}
+  });
+  if (!existing) {
+    throw new Error(
+      `Document with ID "${documentId}" not found in "${contentType}".`
+    );
+  }
   const document = await docs.update({
     documentId,
     data,
     ...status ? { status } : {},
+    ...locale ? { locale } : {},
     populate: "*"
   });
   return { action: "update", document };
@@ -514,6 +563,88 @@ async function recallMemories(strapi, params, context) {
     return { success: false, memories: [], count: 0 };
   }
 }
+const findOneContentSchema = zod.z.object({
+  contentType: zod.z.string().describe(
+    'The content type UID to fetch from, e.g. "api::article.article"'
+  ),
+  documentId: zod.z.string().describe("The document ID to retrieve"),
+  populate: zod.z.union([zod.z.string(), zod.z.array(zod.z.string()), zod.z.record(zod.z.string(), zod.z.unknown())]).optional().default("*").describe(
+    'Relations to populate. Defaults to "*" (all). Can be a string, array, or object.'
+  ),
+  fields: zod.z.array(zod.z.string()).optional().describe("Specific fields to return. If omitted, returns all fields."),
+  status: zod.z.enum(["draft", "published"]).optional().describe("Document status filter."),
+  locale: zod.z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"')
+});
+const findOneContentDescription = "Fetch a single document by its documentId from any Strapi content type. Returns the full document with all fields and populated relations. Use searchContent first to discover document IDs.";
+async function findOneContent(strapi, params) {
+  const { contentType, documentId, populate = "*", fields, status, locale } = params;
+  if (!strapi.contentTypes[contentType]) {
+    throw new Error(`Content type "${contentType}" does not exist.`);
+  }
+  const document = await strapi.documents(contentType).findOne({
+    documentId,
+    ...fields ? { fields } : {},
+    ...status ? { status } : {},
+    ...locale ? { locale } : {},
+    populate
+  });
+  if (!document) {
+    throw new Error(
+      `Document with ID "${documentId}" not found in "${contentType}".`
+    );
+  }
+  return { document };
+}
+const uploadMediaSchema = zod.z.object({
+  url: zod.z.string().describe("URL of the file to upload"),
+  name: zod.z.string().optional().describe("Custom filename (optional, defaults to filename from URL)"),
+  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.";
+async function uploadMedia(strapi, params) {
+  const { url, name, caption, alternativeText } = params;
+  const response = await fetch(url, { redirect: "follow" });
+  if (!response.ok) {
+    throw new Error(`Failed to fetch file from URL: ${response.status} ${response.statusText}`);
+  }
+  const contentType = response.headers.get("content-type") || "application/octet-stream";
+  const arrayBuffer = await response.arrayBuffer();
+  const buffer = Buffer.from(arrayBuffer);
+  const finalUrl = response.url || url;
+  const urlPath = new URL(finalUrl).pathname;
+  const urlFilename = urlPath.split("/").pop() || "upload";
+  const filename = name || urlFilename;
+  const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "strapi-upload-"));
+  const tmpPath = path.join(tmpDir, filename);
+  fs.writeFileSync(tmpPath, buffer);
+  const fileData = {
+    filepath: tmpPath,
+    originalFilename: filename,
+    mimetype: contentType,
+    size: buffer.length
+  };
+  const fileInfo = {};
+  if (caption) fileInfo.caption = caption;
+  if (alternativeText) fileInfo.alternativeText = alternativeText;
+  let uploadedFile;
+  try {
+    [uploadedFile] = await strapi.plugins.upload.services.upload.upload({
+      data: { fileInfo },
+      files: fileData
+    });
+  } finally {
+    try {
+      fs.unlinkSync(tmpPath);
+    } catch {
+    }
+  }
+  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} }`
+  };
+}
 const listContentTypesTool = {
   name: "listContentTypes",
   description: listContentTypesDescription,
@@ -613,10 +744,28 @@ const recallMemoriesTool = {
   execute: async (args, strapi, context) => recallMemories(strapi, args, context),
   internal: true
 };
+const findOneContentTool = {
+  name: "findOneContent",
+  description: findOneContentDescription,
+  schema: findOneContentSchema,
+  execute: async (args, strapi) => {
+    const result = await findOneContent(strapi, args);
+    const sanitizedDoc = await sanitizeOutput(strapi, args.contentType, result.document);
+    return { ...result, document: sanitizedDoc };
+  }
+};
+const uploadMediaTool = {
+  name: "uploadMedia",
+  description: uploadMediaDescription,
+  schema: uploadMediaSchema,
+  execute: async (args, strapi) => uploadMedia(strapi, args)
+};
 const builtInTools = [
   listContentTypesTool,
   searchContentTool,
   writeContentTool,
+  findOneContentTool,
+  uploadMediaTool,
   triggerAnimationTool,
   sendEmailTool,
   saveMemoryTool,
@@ -702,6 +851,8 @@ const config = {
     chatModel: "claude-sonnet-4-20250514",
     baseURL: void 0,
     systemPrompt: "",
+    maxOutputTokens: 8192,
+    maxConversationMessages: 40,
     mcp: {
       sessionTimeoutMs: 4 * 60 * 60 * 1e3,
       maxSessions: 100,
@@ -1359,19 +1510,19 @@ function loadPatterns(config2) {
   return sources.map((p) => new RegExp(p, "i"));
 }
 function extractUserInput(ctx) {
-  const path = ctx.path;
+  const path2 = ctx.path;
   const method = ctx.method;
   const body = ctx.request.body;
-  if (path.endsWith("/mcp") && (method === "GET" || method === "DELETE")) {
+  if (path2.endsWith("/mcp") && (method === "GET" || method === "DELETE")) {
     return null;
   }
-  if (path.endsWith("/mcp") && method === "POST") {
+  if (path2.endsWith("/mcp") && method === "POST") {
     if (body && typeof body === "object" && "params" in body) {
       return { text: JSON.stringify(body.params), route: "mcp" };
     }
     return null;
   }
-  if (path.endsWith("/chat") && method === "POST") {
+  if (path2.endsWith("/chat") && method === "POST") {
     if (body && Array.isArray(body.messages)) {
       const messages = body.messages;
       for (let i = messages.length - 1; i >= 0; i--) {
@@ -1390,9 +1541,9 @@ function extractUserInput(ctx) {
     }
     return null;
   }
-  if ((path.endsWith("/ask") || path.endsWith("/ask-stream")) && method === "POST") {
+  if ((path2.endsWith("/ask") || path2.endsWith("/ask-stream")) && method === "POST") {
     if (body && typeof body.prompt === "string") {
-      return { text: body.prompt, route: path.endsWith("/ask-stream") ? "ask-stream" : "ask" };
+      return { text: body.prompt, route: path2.endsWith("/ask-stream") ? "ask-stream" : "ask" };
     }
     return null;
   }
@@ -1660,7 +1811,10 @@ const service = ({ strapi }) => {
      */
     async chat(messages, options2) {
       const config2 = strapi.config.get("plugin::ai-sdk");
-      const modelMessages = await ai.convertToModelMessages(messages);
+      const maxMessages = config2?.maxConversationMessages ?? DEFAULT_MAX_CONVERSATION_MESSAGES;
+      const maxOutputTokens = config2?.maxOutputTokens ?? DEFAULT_MAX_OUTPUT_TOKENS;
+      const trimmedMessages = messages.length > maxMessages ? messages.slice(-maxMessages) : messages;
+      const modelMessages = await ai.convertToModelMessages(trimmedMessages);
       const tools = createTools(strapi, { adminUserId: options2?.adminUserId });
       const toolsDescription = describeTools(tools);
       let system = composeSystemPrompt(config2, toolsDescription, options2?.system);
@@ -1685,6 +1839,7 @@ ${lines.join("\n")}`;
         messages: modelMessages,
         system,
         tools,
+        maxOutputTokens,
         stopWhen: ai.stepCountIs(6)
       });
     },

package/dist/server/index.mjs

@@ -2,6 +2,9 @@ import { createAnthropic } from "@ai-sdk/anthropic";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { generateText, streamText, tool, zodSchema, convertToModelMessages, stepCountIs } from "ai";
 import { z } from "zod";
+import { mkdtempSync, writeFileSync, unlinkSync } from "fs";
+import { tmpdir } from "os";
+import { join } from "path";
 import { PassThrough, Readable } from "node:stream";
 import { randomUUID } from "node:crypto";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
@@ -143,6 +146,8 @@ function createTTSRegistry() {
 }
 const DEFAULT_MODEL = "claude-sonnet-4-20250514";
 const DEFAULT_TEMPERATURE = 0.7;
+const DEFAULT_MAX_OUTPUT_TOKENS = 8192;
+const DEFAULT_MAX_CONVERSATION_MESSAGES = 40;
 function isPromptInput(input) {
   return "prompt" in input;
 }
@@ -336,20 +341,46 @@ async function listContentTypes(strapi) {
   return { contentTypes: contentTypes2, components };
 }
 const MAX_PAGE_SIZE = 50;
+const LARGE_CONTENT_FIELDS = ["content", "blocks", "body", "richText", "markdown", "html"];
 const searchContentSchema = z.object({
   contentType: z.string().describe(
     'The content type UID to search, e.g. "api::article.article" or "plugin::users-permissions.user"'
   ),
   query: z.string().optional().describe("Full-text search query string (searches across all searchable text fields)"),
   filters: z.record(z.string(), z.unknown()).optional().describe('Strapi filter object, e.g. { username: { $containsi: "john" } }'),
-  fields: z.array(z.string()).optional().describe("Specific fields to return. If omitted, returns all fields."),
+  fields: z.array(z.string()).optional().describe("Specific fields to return. If omitted, returns all fields (large content fields stripped unless includeContent is true)."),
   sort: z.string().optional().describe('Sort order, e.g. "createdAt:desc"'),
   page: z.number().optional().default(1).describe("Page number (starts at 1)"),
-  pageSize: z.number().optional().default(10).describe(`Results per page (max ${MAX_PAGE_SIZE})`)
+  pageSize: z.number().optional().default(10).describe(`Results per page (max ${MAX_PAGE_SIZE})`),
+  status: z.enum(["draft", "published"]).optional().describe("Filter by document status. Published documents have a publishedAt date. If omitted, Strapi returns drafts by default."),
+  locale: z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"'),
+  populate: z.union([z.string(), z.array(z.string()), z.record(z.string(), z.unknown())]).optional().describe('Relations to populate. Defaults to "*" (all). Can be a string, array, or object.'),
+  includeContent: z.boolean().optional().default(false).describe("When true, includes large content fields (content, blocks, body, etc.) in results. Default false to reduce context size.")
 });
-const searchContentDescription = "Search and query any Strapi content type. Use listContentTypes first to discover available content types and their fields, then use this tool to query specific collections.";
+const searchContentDescription = "Search and query any Strapi content type. Use listContentTypes first to discover available content types and their fields, then use this tool to query specific collections. By default, large content fields are stripped from results — set includeContent to true or use fields to get full content.";
+function stripLargeFields(obj) {
+  const stripped = {};
+  for (const [key, value] of Object.entries(obj)) {
+    if (!LARGE_CONTENT_FIELDS.includes(key)) {
+      stripped[key] = value;
+    }
+  }
+  return stripped;
+}
 async function searchContent(strapi, params) {
-  const { contentType, query, filters, fields, sort, page = 1, pageSize = 10 } = params;
+  const {
+    contentType,
+    query,
+    filters,
+    fields,
+    sort,
+    page = 1,
+    pageSize = 10,
+    status,
+    locale,
+    populate = "*",
+    includeContent = false
+  } = params;
   if (!strapi.contentTypes[contentType]) {
     throw new Error(`Content type "${contentType}" does not exist.`);
   }
@@ -359,16 +390,22 @@ async function searchContent(strapi, params) {
     ...filters ? { filters } : {},
     ...fields ? { fields } : {},
     ...sort ? { sort } : {},
+    ...status ? { status } : {},
+    ...locale ? { locale } : {},
     page,
     pageSize: clampedPageSize,
-    populate: "*"
+    populate
   });
   const total = await strapi.documents(contentType).count({
     ...query ? { _q: query } : {},
-    ...filters ? { filters } : {}
+    ...filters ? { filters } : {},
+    ...status ? { status } : {},
+    ...locale ? { locale } : {}
   });
+  const shouldStrip = !includeContent && !fields;
+  const processedResults = shouldStrip ? results.map((doc) => stripLargeFields(doc)) : results;
   return {
-    results,
+    results: processedResults,
     pagination: {
       page,
       pageSize: clampedPageSize,
@@ -381,11 +418,12 @@ const writeContentSchema = z.object({
   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.")
+  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 } = params;
+  const { contentType, action, documentId, data, status, locale } = params;
   if (!strapi.contentTypes[contentType]) {
     throw new Error(`Content type "${contentType}" does not exist.`);
   }
@@ -397,14 +435,25 @@ async function writeContent(strapi, params) {
     const document2 = await docs.create({
       data,
       ...status ? { status } : {},
+      ...locale ? { locale } : {},
       populate: "*"
     });
     return { action: "create", document: document2 };
   }
+  const existing = await docs.findOne({
+    documentId,
+    ...locale ? { locale } : {}
+  });
+  if (!existing) {
+    throw new Error(
+      `Document with ID "${documentId}" not found in "${contentType}".`
+    );
+  }
   const document = await docs.update({
     documentId,
     data,
     ...status ? { status } : {},
+    ...locale ? { locale } : {},
     populate: "*"
   });
   return { action: "update", document };
@@ -513,6 +562,88 @@ async function recallMemories(strapi, params, context) {
     return { success: false, memories: [], count: 0 };
   }
 }
+const findOneContentSchema = z.object({
+  contentType: z.string().describe(
+    'The content type UID to fetch from, e.g. "api::article.article"'
+  ),
+  documentId: z.string().describe("The document ID to retrieve"),
+  populate: z.union([z.string(), z.array(z.string()), z.record(z.string(), z.unknown())]).optional().default("*").describe(
+    'Relations to populate. Defaults to "*" (all). Can be a string, array, or object.'
+  ),
+  fields: z.array(z.string()).optional().describe("Specific fields to return. If omitted, returns all fields."),
+  status: z.enum(["draft", "published"]).optional().describe("Document status filter."),
+  locale: z.string().optional().describe('Locale code for i18n content, e.g. "en" or "fr"')
+});
+const findOneContentDescription = "Fetch a single document by its documentId from any Strapi content type. Returns the full document with all fields and populated relations. Use searchContent first to discover document IDs.";
+async function findOneContent(strapi, params) {
+  const { contentType, documentId, populate = "*", fields, status, locale } = params;
+  if (!strapi.contentTypes[contentType]) {
+    throw new Error(`Content type "${contentType}" does not exist.`);
+  }
+  const document = await strapi.documents(contentType).findOne({
+    documentId,
+    ...fields ? { fields } : {},
+    ...status ? { status } : {},
+    ...locale ? { locale } : {},
+    populate
+  });
+  if (!document) {
+    throw new Error(
+      `Document with ID "${documentId}" not found in "${contentType}".`
+    );
+  }
+  return { document };
+}
+const uploadMediaSchema = z.object({
+  url: z.string().describe("URL of the file to upload"),
+  name: z.string().optional().describe("Custom filename (optional, defaults to filename from URL)"),
+  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.";
+async function uploadMedia(strapi, params) {
+  const { url, name, caption, alternativeText } = params;
+  const response = await fetch(url, { redirect: "follow" });
+  if (!response.ok) {
+    throw new Error(`Failed to fetch file from URL: ${response.status} ${response.statusText}`);
+  }
+  const contentType = response.headers.get("content-type") || "application/octet-stream";
+  const arrayBuffer = await response.arrayBuffer();
+  const buffer = Buffer.from(arrayBuffer);
+  const finalUrl = response.url || url;
+  const urlPath = new URL(finalUrl).pathname;
+  const urlFilename = urlPath.split("/").pop() || "upload";
+  const filename = name || urlFilename;
+  const tmpDir = mkdtempSync(join(tmpdir(), "strapi-upload-"));
+  const tmpPath = join(tmpDir, filename);
+  writeFileSync(tmpPath, buffer);
+  const fileData = {
+    filepath: tmpPath,
+    originalFilename: filename,
+    mimetype: contentType,
+    size: buffer.length
+  };
+  const fileInfo = {};
+  if (caption) fileInfo.caption = caption;
+  if (alternativeText) fileInfo.alternativeText = alternativeText;
+  let uploadedFile;
+  try {
+    [uploadedFile] = await strapi.plugins.upload.services.upload.upload({
+      data: { fileInfo },
+      files: fileData
+    });
+  } finally {
+    try {
+      unlinkSync(tmpPath);
+    } catch {
+    }
+  }
+  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} }`
+  };
+}
 const listContentTypesTool = {
   name: "listContentTypes",
   description: listContentTypesDescription,
@@ -612,10 +743,28 @@ const recallMemoriesTool = {
   execute: async (args, strapi, context) => recallMemories(strapi, args, context),
   internal: true
 };
+const findOneContentTool = {
+  name: "findOneContent",
+  description: findOneContentDescription,
+  schema: findOneContentSchema,
+  execute: async (args, strapi) => {
+    const result = await findOneContent(strapi, args);
+    const sanitizedDoc = await sanitizeOutput(strapi, args.contentType, result.document);
+    return { ...result, document: sanitizedDoc };
+  }
+};
+const uploadMediaTool = {
+  name: "uploadMedia",
+  description: uploadMediaDescription,
+  schema: uploadMediaSchema,
+  execute: async (args, strapi) => uploadMedia(strapi, args)
+};
 const builtInTools = [
   listContentTypesTool,
   searchContentTool,
   writeContentTool,
+  findOneContentTool,
+  uploadMediaTool,
   triggerAnimationTool,
   sendEmailTool,
   saveMemoryTool,
@@ -701,6 +850,8 @@ const config = {
     chatModel: "claude-sonnet-4-20250514",
     baseURL: void 0,
     systemPrompt: "",
+    maxOutputTokens: 8192,
+    maxConversationMessages: 40,
     mcp: {
       sessionTimeoutMs: 4 * 60 * 60 * 1e3,
       maxSessions: 100,
@@ -1659,7 +1810,10 @@ const service = ({ strapi }) => {
      */
     async chat(messages, options2) {
       const config2 = strapi.config.get("plugin::ai-sdk");
-      const modelMessages = await convertToModelMessages(messages);
+      const maxMessages = config2?.maxConversationMessages ?? DEFAULT_MAX_CONVERSATION_MESSAGES;
+      const maxOutputTokens = config2?.maxOutputTokens ?? DEFAULT_MAX_OUTPUT_TOKENS;
+      const trimmedMessages = messages.length > maxMessages ? messages.slice(-maxMessages) : messages;
+      const modelMessages = await convertToModelMessages(trimmedMessages);
       const tools = createTools(strapi, { adminUserId: options2?.adminUserId });
       const toolsDescription = describeTools(tools);
       let system = composeSystemPrompt(config2, toolsDescription, options2?.system);
@@ -1684,6 +1838,7 @@ ${lines.join("\n")}`;
         messages: modelMessages,
         system,
         tools,
+        maxOutputTokens,
         stopWhen: stepCountIs(6)
       });
     },

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

@@ -5,6 +5,8 @@ declare const _default: {
         chatModel: string;
         baseURL: any;
         systemPrompt: string;
+        maxOutputTokens: number;
+        maxConversationMessages: number;
         mcp: {
             sessionTimeoutMs: number;
             maxSessions: number;

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

@@ -16,6 +16,8 @@ declare const _default: {
             chatModel: string;
             baseURL: any;
             systemPrompt: string;
+            maxOutputTokens: number;
+            maxConversationMessages: number;
             mcp: {
                 sessionTimeoutMs: number;
                 maxSessions: number;

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

@@ -16,12 +16,16 @@ export interface MCPConfig {
     maxSessions?: number;
     cleanupInterval?: number;
 }
+export declare const DEFAULT_MAX_OUTPUT_TOKENS = 8192;
+export declare const DEFAULT_MAX_CONVERSATION_MESSAGES = 40;
 export interface PluginConfig {
     anthropicApiKey: string;
     provider?: string;
     chatModel?: string;
     baseURL?: string;
     systemPrompt?: string;
+    maxOutputTokens?: number;
+    maxConversationMessages?: number;
     typecastApiKey?: string;
     typecastActorId?: string;
     mcp?: MCPConfig;

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

@@ -0,0 +1,30 @@
+import type { Core } from '@strapi/strapi';
+import { z } from 'zod';
+export declare const findOneContentSchema: z.ZodObject<{
+    contentType: z.ZodString;
+    documentId: z.ZodString;
+    populate: z.ZodDefault<z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>>;
+    fields: z.ZodOptional<z.ZodArray<z.ZodString>>;
+    status: z.ZodOptional<z.ZodEnum<{
+        draft: "draft";
+        published: "published";
+    }>>;
+    locale: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export declare const findOneContentDescription = "Fetch a single document by its documentId from any Strapi content type. Returns the full document with all fields and populated relations. Use searchContent first to discover document IDs.";
+export interface FindOneContentParams {
+    contentType: string;
+    documentId: string;
+    populate?: string | string[] | Record<string, unknown>;
+    fields?: string[];
+    status?: 'draft' | 'published';
+    locale?: string;
+}
+export interface FindOneContentResult {
+    document: any;
+}
+/**
+ * Core logic for fetching a single document by ID.
+ * Shared between AI SDK tool and MCP tool.
+ */
+export declare function findOneContent(strapi: Core.Strapi, params: FindOneContentParams): Promise<FindOneContentResult>;

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

@@ -10,3 +10,7 @@ export { saveMemory, saveMemorySchema, saveMemoryDescription } from './save-memo
 export type { SaveMemoryParams, SaveMemoryResult } from './save-memory';
 export { recallMemories, recallMemoriesSchema, recallMemoriesDescription } from './recall-memories';
 export type { RecallMemoriesParams, RecallMemoriesResult } from './recall-memories';
+export { findOneContent, findOneContentSchema, findOneContentDescription } from './find-one-content';
+export type { FindOneContentParams, FindOneContentResult } from './find-one-content';
+export { uploadMedia, uploadMediaSchema, uploadMediaDescription } from './upload-media';
+export type { UploadMediaParams, UploadMediaResult } from './upload-media';

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

@@ -8,8 +8,15 @@ export declare const searchContentSchema: z.ZodObject<{
     sort: z.ZodOptional<z.ZodString>;
     page: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
     pageSize: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+    status: z.ZodOptional<z.ZodEnum<{
+        draft: "draft";
+        published: "published";
+    }>>;
+    locale: z.ZodOptional<z.ZodString>;
+    populate: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
+    includeContent: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
 }, z.core.$strip>;
-export declare const searchContentDescription = "Search and query any Strapi content type. Use listContentTypes first to discover available content types and their fields, then use this tool to query specific collections.";
+export declare const searchContentDescription = "Search and query any Strapi content type. Use listContentTypes first to discover available content types and their fields, then use this tool to query specific collections. By default, large content fields are stripped from results \u2014 set includeContent to true or use fields to get full content.";
 export interface SearchContentParams {
     contentType: string;
     query?: string;
@@ -18,6 +25,10 @@ export interface SearchContentParams {
     sort?: string;
     page?: number;
     pageSize?: number;
+    status?: 'draft' | 'published';
+    locale?: string;
+    populate?: string | string[] | Record<string, unknown>;
+    includeContent?: boolean;
 }
 export interface SearchContentResult {
     results: any[];

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

@@ -0,0 +1,25 @@
+import type { Core } from '@strapi/strapi';
+import { z } from 'zod';
+export declare const uploadMediaSchema: z.ZodObject<{
+    url: z.ZodString;
+    name: z.ZodOptional<z.ZodString>;
+    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 interface UploadMediaParams {
+    url: string;
+    name?: string;
+    caption?: string;
+    alternativeText?: string;
+}
+export interface UploadMediaResult {
+    file: any;
+    message: string;
+    usage: string;
+}
+/**
+ * Core logic for uploading media from a URL.
+ * Shared between AI SDK tool and MCP tool.
+ */
+export declare function uploadMedia(strapi: Core.Strapi, params: UploadMediaParams): Promise<UploadMediaResult>;

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

@@ -12,6 +12,7 @@ export declare const writeContentSchema: z.ZodObject<{
         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 {
@@ -20,6 +21,7 @@ export interface WriteContentParams {
     documentId?: string;
     data: Record<string, unknown>;
     status?: 'draft' | 'published';
+    locale?: string;
 }
 export interface WriteContentResult {
     action: 'create' | 'update';

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

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

package/dist/server/src/tools/definitions/upload-media.d.ts

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

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.2.0",
+  "version": "0.5.0",
   "keywords": [
     "strapi",
     "strapi-plugin",
@@ -41,7 +41,8 @@
     "test:api": "npx tsx tests/ai-sdk.test.ts",
     "test:stream": "node tests/test-stream.mjs",
     "test:chat": "node tests/test-chat.mjs",
-    "test:guardrails": "npx tsx tests/test-guardrails.ts"
+    "test:guardrails": "npx tsx tests/test-guardrails.ts",
+    "test:mcp": "npx tsx tests/mcp.test.ts"
   },
   "dependencies": {
     "@ai-sdk/anthropic": "^3.0.15",