@nocobase/plugin-ai 2.2.0-beta.13 → 2.2.0-beta.15

package/dist/server/llm-providers/kimi/provider.js

@@ -46,6 +46,7 @@ var import_ai_manager = require("../../manager/ai-manager");
 var import_document_loader = require("../../document-loader");
 var import_document_loader2 = require("./document-loader");
 var import_reasoning = require("../common/reasoning");
+const KIMI_THINKING_SWITCH_MODELS = /* @__PURE__ */ new Set(["kimi-k2.5", "kimi-k2.6"]);
 class KimiProvider extends import_provider.LLMProvider {
   _documentLoader;
   get baseURL() {
@@ -55,6 +56,7 @@ class KimiProvider extends import_provider.LLMProvider {
     const { apiKey } = this.serviceOptions || {};
     const { responseFormat, structuredOutput } = this.modelOptions || {};
     const { name, schema } = structuredOutput || {};
+    const reasoningOptions = this.resolveReasoningOptions(this.modelReasoningOptions);
     const responseFormatOptions = {
       type: responseFormat ?? "text"
     };
@@ -65,8 +67,10 @@ class KimiProvider extends import_provider.LLMProvider {
       apiKey,
       ...this.modelOptions,
       modelKwargs: {
-        response_format: responseFormatOptions
+        response_format: responseFormatOptions,
+        ...reasoningOptions.modelKwargs || {}
       },
+      ...reasoningOptions.modelRequestParams || {},
       configuration: {
         baseURL: this.getResolvedBaseURL()
       }
@@ -100,6 +104,22 @@ class KimiProvider extends import_provider.LLMProvider {
     }
     return null;
   }
+  resolveReasoningOptions(reasoning) {
+    var _a;
+    if (!reasoning || reasoning.mode === "default") {
+      return {};
+    }
+    if (!KIMI_THINKING_SWITCH_MODELS.has((_a = this.modelOptions) == null ? void 0 : _a.model)) {
+      return {};
+    }
+    return {
+      modelKwargs: {
+        thinking: {
+          type: reasoning.mode === "off" ? "disabled" : "enabled"
+        }
+      }
+    };
+  }
   isApiSupportedAttachment(attachment) {
     var _a;
     return ((_a = attachment.mimetype) == null ? void 0 : _a.startsWith("image/")) ?? false;
@@ -126,7 +146,7 @@ const kimiProviderOptions = {
   title: "Kimi",
   supportedModel: [import_ai_manager.SupportedModel.LLM],
   models: {
-    [import_ai_manager.SupportedModel.LLM]: ["kimi-k2.5", "kimi-k2-0905-Preview", "kimi-k2-turbo-preview"]
+    [import_ai_manager.SupportedModel.LLM]: ["kimi-k2.7-code", "kimi-k2.7-code-highspeed", "kimi-k2.6", "kimi-k2.5"]
   },
   provider: KimiProvider
 };

package/dist/server/llm-providers/mistral.d.ts

@@ -10,21 +10,55 @@ import { EmbeddingsInterface } from '@langchain/core/embeddings';
 import { AIMessage, AIMessageChunk } from '@langchain/core/messages';
 import { ChatMistralAI } from '@langchain/mistralai';
 import { AttachmentModel } from '@nocobase/plugin-file-manager';
-import { EmbeddingProvider, LLMProvider, LLMProviderInvokeOptions } from './provider';
+import { EmbeddingProvider, LLMProvider, LLMProviderInvokeOptions, ReasoningMode, ReasoningOptions, ResolvedReasoningOptions } from './provider';
 import { LLMProviderMeta } from '../manager/ai-manager';
 import { AIChatContext, AIMessageInput } from '../types/ai-chat-conversation.type';
 import { Model } from '@nocobase/database';
+declare const MISTRAL_REASONING_MODE_KEY = "__nb_reasoning_mode";
 type MistralCallOptions = LLMProviderInvokeOptions & {
     response_format?: {
         type: 'text' | 'json_object';
     };
 };
 type MistralBeforeRequestHook = (request: Request) => Request | void | Promise<Request | void>;
+type ReasoningChatMistralAIFields = ConstructorParameters<typeof ChatMistralAI>[0] & {
+    [MISTRAL_REASONING_MODE_KEY]?: ReasoningMode;
+};
 export declare const injectMistralReasoningEffort: MistralBeforeRequestHook;
+declare class ReasoningChatMistralAI extends ChatMistralAI {
+    private readonly reasoningMode?;
+    constructor(fields: ReasoningChatMistralAIFields);
+    invocationParams(options?: MistralCallOptions): Omit<import("@mistralai/mistralai/models/components/chatcompletionrequest").ChatCompletionRequest | import("@mistralai/mistralai/models/components/chatcompletionstreamrequest").ChatCompletionStreamRequest, "messages"> | {
+        __nb_reasoning_mode: "off" | "minimal" | "low" | "medium" | "high" | "xhigh";
+        model: string;
+        tools?: import("@mistralai/mistralai/models/components/tool").Tool[];
+        metadata?: {
+            [k: string]: any;
+        } | {
+            [k: string]: any;
+        };
+        stream?: boolean;
+        responseFormat?: import("@mistralai/mistralai/models/components/responseformat").ResponseFormat;
+        temperature?: number;
+        maxTokens?: number;
+        topP?: number;
+        frequencyPenalty?: number;
+        presencePenalty?: number;
+        n?: number;
+        stop?: string | string[];
+        toolChoice?: import("@mistralai/mistralai/models/components/toolchoice").ToolChoice | import("@mistralai/mistralai/models/components/toolchoiceenum").ToolChoiceEnum;
+        randomSeed?: number;
+        prediction?: import("@mistralai/mistralai/models/components/prediction").Prediction;
+        parallelToolCalls?: boolean;
+        promptMode?: import("@mistralai/mistralai/models/components/mistralpromptmode").MistralPromptMode;
+        safePrompt?: boolean;
+    };
+}
 export declare class MistralProvider extends LLMProvider {
     chatModel: ChatMistralAI;
     get baseURL(): string;
-    createModel(): ChatMistralAI<import("@langchain/mistralai/dist/chat_models").ChatMistralAICallOptions>;
+    createModel(): ReasoningChatMistralAI;
+    protected resolveReasoningOptions(reasoning?: ReasoningOptions): ResolvedReasoningOptions;
     stream(context: AIChatContext, options?: MistralCallOptions): Promise<any>;
     invoke(context: AIChatContext, options?: MistralCallOptions): Promise<any>;
     listModels(): Promise<{

package/dist/server/llm-providers/mistral.js

@@ -39,6 +39,8 @@ var import_ai_manager = require("../manager/ai-manager");
 const MISTRAL_URL = "https://api.mistral.ai";
 const MISTRAL_REASONING_EFFORT = "high";
 const MISTRAL_REASONING_MODELS = /* @__PURE__ */ new Set(["mistral-small-latest", "mistral-medium-3-5"]);
+const MISTRAL_REASONING_EFFORTS = /* @__PURE__ */ new Set(["off", "low", "medium", "high"]);
+const MISTRAL_REASONING_MODE_KEY = "__nb_reasoning_mode";
 function omitDisabledNumberOptions(options) {
   for (const key of ["maxTokens", "maxRetries"]) {
     if (options[key] === -1) {
@@ -73,19 +75,62 @@ const injectMistralReasoningEffort = async (request) => {
   if (!body || typeof body !== "object") {
     return;
   }
+  const reasoningMode = body[MISTRAL_REASONING_MODE_KEY];
+  delete body[MISTRAL_REASONING_MODE_KEY];
+  const shouldRewriteBody = reasoningMode != null;
   if (typeof body.model !== "string" || !MISTRAL_REASONING_MODELS.has(body.model)) {
+    if (shouldRewriteBody) {
+      const headers2 = new Headers(request.headers);
+      headers2.delete("content-length");
+      return new Request(request, {
+        headers: headers2,
+        body: JSON.stringify(body)
+      });
+    }
     return;
   }
+  if (reasoningMode === "off") {
+    delete body.reasoning_effort;
+    const headers2 = new Headers(request.headers);
+    headers2.delete("content-length");
+    return new Request(request, {
+      headers: headers2,
+      body: JSON.stringify(body)
+    });
+  }
+  if (body.reasoning_effort && !shouldRewriteBody) {
+    return;
+  }
+  const reasoningEffort = reasoningMode && reasoningMode !== "default" ? reasoningMode : MISTRAL_REASONING_EFFORT;
   const headers = new Headers(request.headers);
   headers.delete("content-length");
   return new Request(request, {
     headers,
     body: JSON.stringify({
       ...body,
-      reasoning_effort: MISTRAL_REASONING_EFFORT
+      reasoning_effort: reasoningEffort
     })
   });
 };
+class ReasoningChatMistralAI extends import_mistralai.ChatMistralAI {
+  reasoningMode;
+  constructor(fields) {
+    const { [MISTRAL_REASONING_MODE_KEY]: reasoningMode, ...modelFields } = fields;
+    super(modelFields);
+    this.reasoningMode = reasoningMode;
+  }
+  invocationParams(options) {
+    const params = super.invocationParams(options);
+    const mode = this.reasoningMode;
+    if (!mode || mode === "default") {
+      return params;
+    }
+    return {
+      ...params,
+      [MISTRAL_REASONING_MODE_KEY]: mode
+    };
+  }
+}
 class MistralProvider extends import_provider.LLMProvider {
   get baseURL() {
     return MISTRAL_URL;
@@ -95,14 +140,29 @@ class MistralProvider extends import_provider.LLMProvider {
     const { responseFormat, structuredOutput, ...modelOptions } = this.modelOptions || {};
     omitDisabledNumberOptions(modelOptions);
     const beforeRequestHooks = Array.isArray(modelOptions.beforeRequestHooks) ? [injectMistralReasoningEffort, ...modelOptions.beforeRequestHooks] : [injectMistralReasoningEffort];
-    return new import_mistralai.ChatMistralAI({
+    const reasoningOptions = this.resolveReasoningOptions(this.modelReasoningOptions);
+    return new ReasoningChatMistralAI({
       apiKey,
       ...modelOptions,
+      ...reasoningOptions.modelRequestParams || {},
       beforeRequestHooks,
       serverURL: this.getResolvedServerURL(),
       verbose: false
     });
   }
+  resolveReasoningOptions(reasoning) {
+    if (!reasoning || reasoning.mode === "default") {
+      return {};
+    }
+    if (!MISTRAL_REASONING_EFFORTS.has(reasoning.mode)) {
+      return {};
+    }
+    return {
+      modelRequestParams: {
+        [MISTRAL_REASONING_MODE_KEY]: reasoning.mode
+      }
+    };
+  }
   async stream(context, options) {
     return super.stream(context, this.withResponseFormat(options));
   }

package/dist/server/llm-providers/openai/completions.d.ts

@@ -7,9 +7,10 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { ChatOpenAI } from '@langchain/openai';
-import { LLMProvider } from '../provider';
+import { LLMProvider, ReasoningOptions, ResolvedReasoningOptions } from '../provider';
 export declare class OpenAICompletionsProvider extends LLMProvider {
     chatModel: ChatOpenAI;
     get baseURL(): string;
     createModel(): ChatOpenAI<import("@langchain/openai/dist/chat_models/index.cjs").ChatOpenAICallOptions>;
+    protected resolveReasoningOptions(reasoning?: ReasoningOptions): ResolvedReasoningOptions;
 }

package/dist/server/llm-providers/openai/completions.js

@@ -39,6 +39,7 @@ class OpenAICompletionsProvider extends import_provider.LLMProvider {
     const { apiKey } = this.serviceOptions || {};
     const { responseFormat, structuredOutput } = this.modelOptions || {};
     const { name, schema } = structuredOutput || {};
+    const reasoningOptions = this.resolveReasoningOptions(this.modelReasoningOptions);
     const responseFormatOptions = {
       type: responseFormat ?? "text"
     };
@@ -48,14 +49,29 @@ class OpenAICompletionsProvider extends import_provider.LLMProvider {
     return new import_openai.ChatOpenAI({
       apiKey,
       ...this.modelOptions,
+      ...reasoningOptions.modelRequestParams || {},
       modelKwargs: {
-        response_format: responseFormatOptions
+        response_format: responseFormatOptions,
+        ...reasoningOptions.modelKwargs || {}
       },
       configuration: {
         baseURL: this.getResolvedBaseURL()
       }
     });
   }
+  resolveReasoningOptions(reasoning) {
+    if (!reasoning || reasoning.mode === "default") {
+      return {};
+    }
+    const effort = reasoning.mode === "off" ? "none" : reasoning.mode;
+    return {
+      modelRequestParams: {
+        reasoning: {
+          effort
+        }
+      }
+    };
+  }
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/server/llm-providers/openai/responses.d.ts

@@ -7,7 +7,7 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { ChatOpenAI } from '@langchain/openai';
-import { LLMProvider } from '../provider';
+import { LLMProvider, ReasoningOptions, ResolvedReasoningOptions } from '../provider';
 import { Model } from '@nocobase/database';
 import { AIMessageChunk } from '@langchain/core/messages';
 export declare class OpenAIResponsesProvider extends LLMProvider {
@@ -22,6 +22,7 @@ export declare class OpenAIResponsesProvider extends LLMProvider {
         role: any;
     };
     protected builtInTools(): any[];
+    protected resolveReasoningOptions(reasoning?: ReasoningOptions): ResolvedReasoningOptions;
     isToolConflict(): boolean;
     parseWebSearchAction(chunk: AIMessageChunk): {
         type: string;

package/dist/server/llm-providers/openai/responses.js

@@ -40,6 +40,7 @@ class OpenAIResponsesProvider extends import_provider.LLMProvider {
     const { apiKey } = this.serviceOptions || {};
     const { responseFormat, structuredOutput } = this.modelOptions || {};
     const { name, schema, strict } = structuredOutput || {};
+    const reasoningOptions = this.resolveReasoningOptions(this.modelReasoningOptions);
     let responseFormatOptions = {
       type: responseFormat ?? "text"
     };
@@ -54,10 +55,12 @@ class OpenAIResponsesProvider extends import_provider.LLMProvider {
     return new import_openai.ChatOpenAI({
       apiKey,
       ...this.modelOptions,
+      ...reasoningOptions.modelRequestParams || {},
       modelKwargs: {
         text: {
           format: responseFormatOptions
-        }
+        },
+        ...reasoningOptions.modelKwargs || {}
       },
       configuration: {
         baseURL: this.getResolvedBaseURL()
@@ -130,6 +133,19 @@ class OpenAIResponsesProvider extends import_provider.LLMProvider {
     }
     return [];
   }
+  resolveReasoningOptions(reasoning) {
+    if (!reasoning || reasoning.mode === "default") {
+      return {};
+    }
+    const effort = reasoning.mode === "off" ? "none" : reasoning.mode;
+    return {
+      modelRequestParams: {
+        reasoning: {
+          effort
+        }
+      }
+    };
+  }
   isToolConflict() {
     return false;
   }

package/dist/server/llm-providers/provider.d.ts

@@ -8,7 +8,7 @@
  */
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { Model } from '@nocobase/database';
-import { AttachmentModel } from '@nocobase/plugin-file-manager';
+import { type AttachmentModel } from '@nocobase/plugin-file-manager';
 import { Application } from '@nocobase/server';
 import { AIChatContext, AIMessageInput } from '../types/ai-chat-conversation.type';
 import { EmbeddingsInterface } from '@langchain/core/embeddings';
@@ -27,6 +27,11 @@ export type LLMProviderInvokeOptions = {
     modelRequestParams?: Record<string, any>;
     [key: string]: any;
 };
+export type ReasoningMode = 'default' | 'off' | 'minimal' | 'low' | 'medium' | 'high' | 'xhigh';
+export type ReasoningOptions = {
+    mode: ReasoningMode;
+};
+export type ResolvedReasoningOptions = Pick<LLMProviderInvokeOptions, 'modelKwargs' | 'modelRequestParams'>;
 export type LLMModelRequestBuilderResult = {
     context: AIChatContext;
     options?: LLMProviderInvokeOptions;
@@ -45,10 +50,12 @@ export declare abstract class LLMProvider {
     serviceOptions: Record<string, any>;
     modelOptions: Record<string, any> | undefined;
     chatModel: any;
+    protected modelReasoningOptions: ReasoningOptions | undefined;
     abstract createModel(): BaseChatModel | any;
     get baseURL(): string | null;
     constructor(opts: LLMProviderOptions);
     protected getModelRequestBuilder(_model?: string): LLMModelRequestBuilder | null;
+    protected resolveReasoningOptions(_reasoning?: ReasoningOptions): ResolvedReasoningOptions;
     prepareChain(context: AIChatContext): any;
     invoke(context: AIChatContext, options?: LLMProviderInvokeOptions): Promise<any>;
     stream(context: AIChatContext, options?: any): Promise<any>;
@@ -69,7 +76,8 @@ export declare abstract class LLMProvider {
     parseAttachment(ctx: Context, attachment: AttachmentModel): Promise<ParsedAttachmentResult>;
     protected isApiSupportedAttachment(attachment: AttachmentModel): boolean;
     protected isDocumentLoaderSupportedAttachment(attachment: AttachmentModel): boolean;
-    protected convertToContent(ctx: Context, attachment: any): Promise<ParsedAttachmentResult>;
+    protected encodeAttachment(ctx: Context, attachment: AttachmentModel): Promise<string>;
+    protected convertToContent(ctx: Context, attachment: AttachmentModel): Promise<ParsedAttachmentResult>;
     protected loadDocument(ctx: Context, attachment: any): Promise<any>;
     getStructuredOutputOptions(structuredOutput: AIChatContext['structuredOutput']): any;
     testFlight(): Promise<{

package/dist/server/llm-providers/provider.js

@@ -84,6 +84,7 @@ class LLMProvider {
   serviceOptions;
   modelOptions;
   chatModel;
+  modelReasoningOptions;
   get baseURL() {
     return null;
   }
@@ -92,13 +93,18 @@ class LLMProvider {
     this.app = app;
     this.serviceOptions = resolveServiceOptions(serviceOptions, app);
     if (modelOptions) {
-      this.modelOptions = modelOptions;
+      const { _reasoning, ...restModelOptions } = modelOptions;
+      this.modelReasoningOptions = _reasoning;
+      this.modelOptions = restModelOptions;
       this.chatModel = this.createModel();
     }
   }
   getModelRequestBuilder(_model) {
     return null;
   }
+  resolveReasoningOptions(_reasoning) {
+    return {};
+  }
   prepareChain(context) {
     var _a, _b, _c, _d;
     let chain = this.chatModel;
@@ -183,6 +189,15 @@ class LLMProvider {
     return (0, import_utils2.stripToolCallTags)(chunk);
   }
   async parseAttachment(ctx, attachment) {
+    var _a;
+    const dataSourceKey = (_a = attachment == null ? void 0 : attachment.source) == null ? void 0 : _a.dataSourceKey;
+    const isExternalAttachment = Boolean(dataSourceKey && dataSourceKey !== "main");
+    if (!(attachment == null ? void 0 : attachment.storageId) && !isExternalAttachment || !(attachment == null ? void 0 : attachment.filename)) {
+      return {
+        placement: "system",
+        content: "The user provided an attachment, but it is unavailable or invalid and cannot be parsed. Do not use this attachment as evidence; tell the user the attachment is unavailable."
+      };
+    }
     if (this.isApiSupportedAttachment(attachment)) {
       return await this.convertToContent(ctx, attachment);
     } else if (this.isDocumentLoaderSupportedAttachment(attachment)) {
@@ -212,10 +227,24 @@ class LLMProvider {
     const ext = import_node_path.default.extname((attachment == null ? void 0 : attachment.filename) ?? "").toLocaleLowerCase();
     return import_document_loader.SUPPORTED_DOCUMENT_EXTNAMES.includes(ext);
   }
-  async convertToContent(ctx, attachment) {
+  async encodeAttachment(ctx, attachment) {
     const fileManager = this.app.pm.get("file-manager");
-    const url = await fileManager.getFileURL(attachment);
-    const data = await (0, import_utils2.encodeFile)(ctx, decodeURIComponent(url));
+    if (typeof ctx.get !== "function") {
+      const { stream: stream2 } = await fileManager.getFileStream(attachment);
+      return await (0, import_utils2.encodeReadableStream)(stream2);
+    }
+    const { stream } = await fileManager.getFileStream(attachment, {
+      requestOptions: {
+        headers: {
+          Referer: ctx.get("referer") || "",
+          "User-Agent": ctx.get("user-agent") || ""
+        }
+      }
+    });
+    return await (0, import_utils2.encodeReadableStream)(stream);
+  }
+  async convertToContent(ctx, attachment) {
+    const data = await this.encodeAttachment(ctx, attachment);
     if (attachment.mimetype.startsWith("image/")) {
       return {
         placement: "contentBlocks",

package/dist/server/manager/ai-context-datasource-manager.js

@@ -60,6 +60,34 @@ function serializeQueryFieldValue(value) {
   }
   return value;
 }
+function getRecordFieldValue(record, field) {
+  return field.split(".").reduce((current, key) => {
+    if (current === null || current === void 0) {
+      return current;
+    }
+    if (typeof current.get === "function") {
+      return current.get(key);
+    }
+    return current[key];
+  }, record);
+}
+function getQueryFieldOptions(collection, field) {
+  const parts = field.split(".");
+  let currentCollection = collection;
+  for (const [index, part] of parts.entries()) {
+    const currentField = currentCollection == null ? void 0 : currentCollection.getField(part);
+    if (!currentField) {
+      return;
+    }
+    if (index === parts.length - 1) {
+      return currentField.options;
+    }
+    if (!currentField.isRelationField()) {
+      return;
+    }
+    currentCollection = currentField.targetCollection();
+  }
+}
 class AIContextDatasourceManager {
   constructor(plugin) {
     this.plugin = plugin;
@@ -149,17 +177,16 @@ class AIContextDatasourceManager {
         }
       });
     }
-    const { fields, filter, sort, offset, limit } = options;
-    const result = await collection.repository.find({ fields, filter, sort, offset: offset ?? 0, limit });
+    const { fields, appends, filter, sort, offset, limit } = options;
+    const result = await collection.repository.find({ fields, appends, filter, sort, offset: offset ?? 0, limit });
     const total = await collection.repository.count({ fields, filter });
     const records = result.map(
       (x) => fields.map((field) => {
-        var _a2;
-        const { name, type } = ((_a2 = collection.getField(field)) == null ? void 0 : _a2.options) || {};
-        const value = serializeQueryFieldValue(x[field]);
+        const { name, type } = getQueryFieldOptions(collection, field) || {};
+        const value = serializeQueryFieldValue(getRecordFieldValue(x, field));
         return {
-          name,
-          type,
+          name: field.includes(".") ? field : name || field,
+          type: type || "unknown",
           value
         };
       })

package/dist/server/manager/ai-manager.d.ts

@@ -6,7 +6,7 @@
  * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
-import { LLMProvider, LLMProviderOptions, EmbeddingProvider, EmbeddingProviderOptions } from '../llm-providers/provider';
+import { LLMProvider, LLMProviderOptions, EmbeddingProvider, EmbeddingProviderOptions, ReasoningOptions } from '../llm-providers/provider';
 import PluginAIServer from '../plugin';
 import { ToolManager } from './tool-manager';
 export type LLMProviderMeta = {
@@ -25,6 +25,7 @@ export type LLMModelOptions = {
     llmService: string;
     model: string;
     webSearch?: boolean;
+    reasoning?: ReasoningOptions;
 };
 export type EnabledLLMModel = {
     label: string;

package/dist/server/manager/ai-manager.js

@@ -123,7 +123,7 @@ class AIManager {
     return (0, import_recommended_models.getRecommendedModels)(provider);
   }
   async getLLMService(options) {
-    const { llmService, model, webSearch } = options ?? {};
+    const { llmService, model, webSearch, reasoning } = options ?? {};
     if (!llmService || !model) {
       throw new Error("LLM service not configured");
     }
@@ -134,6 +134,9 @@ class AIManager {
     if (webSearch === true) {
       modelOptions.builtIn = { webSearch: true };
     }
+    if (reasoning) {
+      modelOptions._reasoning = reasoning;
+    }
     const service = await this.plugin.db.getRepository("llmServices").findOne({
       filter: {
         name: llmService

package/dist/server/plugin.js

@@ -80,6 +80,7 @@ var import_sub_agents = require("./ai-employees/sub-agents");
 var import_employee = require("./workflow/nodes/employee");
 var import_ai_knowledge_base = require("./ai-employees/ai-knowledge-base");
 var import_llm_stream_manager = require("./manager/llm-stream-manager");
+var import_attachments = require("./attachments");
 class PluginAIServer extends import_server.Plugin {
   features = new import_ai_feature_manager.AIPluginFeatureManagerImpl();
   aiManager = new import_ai_manager.AIManager(this);
@@ -218,6 +219,19 @@ class PluginAIServer extends import_server.Plugin {
       },
       { before: "createMiddleware" }
     );
+    this.app.resourceManager.use(
+      async (ctx, next) => {
+        const { resourceName, actionName } = ctx.action;
+        if (resourceName === "aiFiles" && actionName === "create") {
+          (0, import_attachments.appendAIFileAttachmentSource)(ctx.action.params.values);
+        }
+        await next();
+        if (resourceName === "aiFiles" && actionName === "create") {
+          (0, import_attachments.appendAIFileAttachmentSource)(ctx.body);
+        }
+      },
+      { after: "createMiddleware" }
+    );
     Object.entries(aiEmployeeActions).forEach(([name, action]) => {
       this.app.resourceManager.registerActionHandler(`aiEmployees:${name}`, action);
     });

package/dist/server/resource/aiConversations.js

@@ -103,6 +103,33 @@ const saveUserMessages = async (ctx, sessionId, messages, messageId) => {
     await conversation.addMessages(userMessages);
   });
 };
+function isRecord(value) {
+  return Boolean(value && typeof value === "object" && !Array.isArray(value));
+}
+function normalizeIncomingMessageAttachments(ctx, messages) {
+  for (const message of messages) {
+    if (message.attachments == null) {
+      continue;
+    }
+    if (!Array.isArray(message.attachments)) {
+      throw new import_utils.ResourceActionError(400, ctx.t("Invalid attachment"));
+    }
+    message.attachments = message.attachments.map((attachment) => {
+      if (!isRecord(attachment) || !isRecord(attachment.source)) {
+        throw new import_utils.ResourceActionError(400, ctx.t("Invalid attachment"));
+      }
+      const source = { ...attachment.source };
+      delete source.trustworthy;
+      if (typeof source.collectionName !== "string" || !source.collectionName) {
+        throw new import_utils.ResourceActionError(400, ctx.t("Invalid attachment"));
+      }
+      return {
+        ...attachment,
+        source
+      };
+    });
+  }
+}
 var aiConversations_default = {
   name: "aiConversations",
   middlewares: [
@@ -329,6 +356,7 @@ var aiConversations_default = {
         if (!Array.isArray(messages)) {
           throw new import_utils.ResourceActionError(400, ctx.t("messages must be an array"));
         }
+        normalizeIncomingMessageAttachments(ctx, messages);
         const userMessage = messages.find((message) => message.role === "user");
         if (!userMessage) {
           throw new import_utils.ResourceActionError(400, ctx.t("user message is required"));
@@ -633,6 +661,13 @@ var aiConversations_default = {
       if (!message) {
         ctx.throw(400);
       }
+      const messageConversation = await plugin.aiConversationsManager.getConversation({
+        sessionId: message.sessionId,
+        userId
+      });
+      if (!messageConversation) {
+        ctx.throw(400);
+      }
       const toolCalls = message.toolCalls;
       if (!(toolCalls == null ? void 0 : toolCalls.length)) {
         ctx.throw(400);
@@ -730,6 +765,14 @@ var aiConversations_default = {
           sendErrorResponse(ctx, "message not found");
           return next();
         }
+        const messageConversation = await plugin.aiConversationsManager.getConversation({
+          sessionId: message.sessionId,
+          userId
+        });
+        if (!messageConversation) {
+          sendErrorResponse(ctx, "conversation not found");
+          return next();
+        }
         const tools = message.toolCalls;
         if (!(tools == null ? void 0 : tools.length)) {
           sendErrorResponse(ctx, "No tool calls found");
@@ -746,7 +789,7 @@ var aiConversations_default = {
           webSearch,
           model: resolvedModel
         });
-        const userDecisions = await plugin.aiConversationsManager.getUserDecisions(messageId);
+        const userDecisions = await plugin.aiConversationsManager.getUserDecisions(message.messageId);
         await aiEmployee.stream({
           userDecisions
         });