@nocobase/plugin-ai 2.1.11 → 2.1.14

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

@@ -0,0 +1,265 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var mistral_exports = {};
+__export(mistral_exports, {
+  MistralEmbeddingProvider: () => MistralEmbeddingProvider,
+  MistralProvider: () => MistralProvider,
+  injectMistralReasoningEffort: () => injectMistralReasoningEffort,
+  mistralProviderOptions: () => mistralProviderOptions
+});
+module.exports = __toCommonJS(mistral_exports);
+var import_mistralai = require("@langchain/mistralai");
+var import_utils = require("@nocobase/utils");
+var import_provider = require("./provider");
+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"]);
+function omitDisabledNumberOptions(options) {
+  for (const key of ["maxTokens", "maxRetries"]) {
+    if (options[key] === -1) {
+      delete options[key];
+    }
+  }
+}
+function extractTextChunks(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (!Array.isArray(content)) {
+    return "";
+  }
+  return content.filter((chunk) => chunk && typeof chunk === "object").filter((chunk) => chunk.type === "text" && typeof chunk.text === "string").map((chunk) => chunk.text).join("");
+}
+function extractThinkingChunks(content) {
+  if (!Array.isArray(content)) {
+    return "";
+  }
+  return content.filter((chunk) => chunk && typeof chunk === "object").filter((chunk) => chunk.type === "thinking" && Array.isArray(chunk.thinking)).map((chunk) => extractTextChunks(chunk.thinking)).join("");
+}
+const injectMistralReasoningEffort = async (request) => {
+  if (request.method !== "POST") {
+    return;
+  }
+  const contentType = request.headers.get("content-type") || "";
+  if (!contentType.toLowerCase().includes("application/json")) {
+    return;
+  }
+  const body = await request.clone().json().catch(() => null);
+  if (!body || typeof body !== "object") {
+    return;
+  }
+  if (typeof body.model !== "string" || !MISTRAL_REASONING_MODELS.has(body.model)) {
+    return;
+  }
+  const headers = new Headers(request.headers);
+  headers.delete("content-length");
+  return new Request(request, {
+    headers,
+    body: JSON.stringify({
+      ...body,
+      reasoning_effort: MISTRAL_REASONING_EFFORT
+    })
+  });
+};
+class MistralProvider extends import_provider.LLMProvider {
+  get baseURL() {
+    return MISTRAL_URL;
+  }
+  createModel() {
+    const { apiKey } = this.serviceOptions || {};
+    const { responseFormat, structuredOutput, ...modelOptions } = this.modelOptions || {};
+    omitDisabledNumberOptions(modelOptions);
+    const beforeRequestHooks = Array.isArray(modelOptions.beforeRequestHooks) ? [injectMistralReasoningEffort, ...modelOptions.beforeRequestHooks] : [injectMistralReasoningEffort];
+    return new import_mistralai.ChatMistralAI({
+      apiKey,
+      ...modelOptions,
+      beforeRequestHooks,
+      serverURL: this.getResolvedServerURL(),
+      verbose: false
+    });
+  }
+  async stream(context, options) {
+    return super.stream(context, this.withResponseFormat(options));
+  }
+  async invoke(context, options) {
+    return super.invoke(context, this.withResponseFormat(options));
+  }
+  async listModels() {
+    var _a, _b, _c, _d;
+    const options = this.serviceOptions || {};
+    const apiKey = options.apiKey;
+    let url;
+    try {
+      url = this.buildMistralRequestURL("v1/models");
+    } catch (e) {
+      return { code: 400, errMsg: e instanceof Error ? e.message : String(e) };
+    }
+    if (!apiKey) {
+      return { code: 400, errMsg: "API Key required" };
+    }
+    try {
+      const res = await (0, import_utils.serverRequest)({
+        method: "GET",
+        url,
+        headers: {
+          Authorization: `Bearer ${apiKey}`
+        }
+      });
+      return { models: (_a = res == null ? void 0 : res.data) == null ? void 0 : _a.data };
+    } catch (e) {
+      const error = e;
+      const status = ((_b = error.response) == null ? void 0 : _b.status) || 500;
+      const data = (_c = error.response) == null ? void 0 : _c.data;
+      const errorMsg = (typeof data === "object" ? data.message : void 0) || (typeof data === "object" && typeof data.detail === "object" ? data.detail.message : void 0) || (typeof data === "object" && typeof data.detail === "string" ? data.detail : void 0) || (typeof data === "string" ? data : void 0) || ((_d = error.response) == null ? void 0 : _d.statusText) || error.message;
+      return { code: status, errMsg: errorMsg };
+    }
+  }
+  isApiSupportedAttachment(attachment) {
+    var _a;
+    return ((_a = attachment.mimetype) == null ? void 0 : _a.startsWith("image/")) ?? false;
+  }
+  reshapeAIMessage({ aiMessage, values }) {
+    const reasoning = extractThinkingChunks(aiMessage.content);
+    if (Array.isArray(aiMessage.content)) {
+      aiMessage.content = extractTextChunks(aiMessage.content);
+    }
+    if (!reasoning) {
+      return;
+    }
+    values.metadata.additional_kwargs = {
+      ...values.metadata.additional_kwargs ?? {},
+      reasoning_content: reasoning
+    };
+  }
+  parseResponseMessage(message) {
+    var _a, _b, _c, _d;
+    const result = super.parseResponseMessage(message);
+    if (["user", "tool"].includes(result == null ? void 0 : result.role)) {
+      return result;
+    }
+    const metadataReasoning = (_c = (_b = (_a = result == null ? void 0 : result.content) == null ? void 0 : _a.metadata) == null ? void 0 : _b.additional_kwargs) == null ? void 0 : _c.reasoning_content;
+    if (typeof metadataReasoning === "string" && metadataReasoning) {
+      result.content = {
+        ...result.content ?? {},
+        reasoning: {
+          status: "stop",
+          content: metadataReasoning
+        }
+      };
+      return result;
+    }
+    const rawContent = (_d = result == null ? void 0 : result.content) == null ? void 0 : _d.content;
+    const reasoning = extractThinkingChunks(rawContent);
+    if (reasoning) {
+      result.content = {
+        ...result.content ?? {},
+        content: extractTextChunks(rawContent),
+        reasoning: {
+          status: "stop",
+          content: reasoning
+        }
+      };
+    }
+    return result;
+  }
+  parseResponseChunk(chunk) {
+    if (Array.isArray(chunk)) {
+      return extractTextChunks(chunk);
+    }
+    return super.parseResponseChunk(chunk);
+  }
+  parseReasoningContent(chunk) {
+    const content = extractThinkingChunks(chunk == null ? void 0 : chunk.content);
+    if (!content) {
+      return null;
+    }
+    return {
+      status: "streaming",
+      content
+    };
+  }
+  withResponseFormat(options) {
+    var _a;
+    const responseFormat = (_a = this.modelOptions) == null ? void 0 : _a.responseFormat;
+    if (responseFormat !== "json_object") {
+      return options;
+    }
+    return {
+      ...options || {},
+      response_format: {
+        type: responseFormat
+      }
+    };
+  }
+  getResolvedServerURL() {
+    return this.getResolvedBaseURL().replace(/\/v1$/, "");
+  }
+  buildMistralRequestURL(pathname) {
+    const url = new URL(pathname.replace(/^\/+/, ""), `${this.getResolvedServerURL()}/`).toString();
+    (0, import_utils.checkUrlAgainstWhitelist)(url);
+    return url;
+  }
+}
+class MistralEmbeddingProvider extends import_provider.EmbeddingProvider {
+  getDefaultUrl() {
+    return MISTRAL_URL;
+  }
+  createEmbedding() {
+    return new import_mistralai.MistralAIEmbeddings({
+      apiKey: this.apiKey,
+      serverURL: this.baseURL.replace(/\/v1$/, ""),
+      model: this.model
+    });
+  }
+}
+const mistralProviderOptions = {
+  title: "Mistral AI",
+  supportedModel: [import_ai_manager.SupportedModel.LLM, import_ai_manager.SupportedModel.EMBEDDING],
+  models: {
+    [import_ai_manager.SupportedModel.LLM]: [
+      "mistral-large-latest",
+      "mistral-medium-latest",
+      "mistral-medium-3-5",
+      "mistral-small-latest",
+      "ministral-8b-latest",
+      "ministral-3b-latest",
+      "codestral-latest",
+      "pixtral-large-latest"
+    ],
+    [import_ai_manager.SupportedModel.EMBEDDING]: ["mistral-embed"]
+  },
+  provider: MistralProvider,
+  embedding: MistralEmbeddingProvider
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  MistralEmbeddingProvider,
+  MistralProvider,
+  injectMistralReasoningEffort,
+  mistralProviderOptions
+});

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

@@ -10,9 +10,9 @@ import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { Model } from '@nocobase/database';
 import { AttachmentModel } from '@nocobase/plugin-file-manager';
 import { Application } from '@nocobase/server';
-import { AIChatContext } from '../types/ai-chat-conversation.type';
+import { AIChatContext, AIMessageInput } from '../types/ai-chat-conversation.type';
 import { EmbeddingsInterface } from '@langchain/core/embeddings';
-import { AIMessageChunk } from '@langchain/core/messages';
+import { AIMessage, AIMessageChunk } from '@langchain/core/messages';
 import { Context } from '@nocobase/actions';
 import '@langchain/core/utils/stream';
 import { LLMResult } from '@langchain/core/outputs';
@@ -90,6 +90,10 @@ export declare abstract class LLMProvider {
     } | null;
     parseResponseMetadata(output: LLMResult): any;
     parseResponseError(err: any): any;
+    reshapeAIMessage(_options: {
+        aiMessage: AIMessage;
+        values: AIMessageInput;
+    }): void;
     protected get documentLoader(): CachedDocumentLoader;
     protected getResolvedBaseURL(): string;
     protected buildRequestURL(pathname: string): string;

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

@@ -319,6 +319,8 @@ ${parsed.text}
   parseResponseError(err) {
     return (err == null ? void 0 : err.message) ?? "Unexpected LLM service error";
   }
+  reshapeAIMessage(_options) {
+  }
   get documentLoader() {
     return this.aiPlugin.documentLoaders.cached;
   }

package/dist/server/plugin.js

@@ -74,6 +74,7 @@ var import_xai = require("./llm-providers/xai");
 var import_document_loader = require("./document-loader");
 var import_checkpoints = require("./ai-employees/checkpoints");
 var import_mimo = require("./llm-providers/mimo");
+var import_mistral = require("./llm-providers/mistral");
 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");
@@ -180,6 +181,7 @@ class PluginAIServer extends import_server.Plugin {
     this.aiManager.registerLLMProvider("dashscope", import_dashscope.dashscopeProviderOptions);
     this.aiManager.registerLLMProvider("kimi", import_kimi.kimiProviderOptions);
     this.aiManager.registerLLMProvider("mimo", import_mimo.mimoProviderOptions);
+    this.aiManager.registerLLMProvider("mistral", import_mistral.mistralProviderOptions);
     this.aiManager.registerLLMProvider("ollama", import_ollama.ollamaProviderOptions);
     this.aiManager.registerLLMProvider("openai-completions", import_openai2.openaiCompletionsProviderOptions);
     this.aiManager.registerLLMProvider("kimi", import_kimi.kimiProviderOptions);

package/package.json

@@ -6,7 +6,7 @@
   "description": "Create AI employees with diverse skills to collaborate with humans, build systems, and handle business operations.",
   "description.ru-RU": "Поддержка интеграции с AI-сервисами: предоставляются AI-узлы для рабочих процессов, расширяя возможности бизнес-обработки.",
   "description.zh-CN": "创建各种技能的 AI 员工，与人类协同，搭建系统，处理业务。",
-  "version": "2.1.11",
+  "version": "2.1.14",
   "main": "dist/server/index.js",
   "homepage": "https://docs.nocobase.com/handbook/action-ai",
   "homepage.ru-RU": "https://docs-ru.nocobase.com/handbook/action-ai",
@@ -41,6 +41,7 @@
     "@langchain/google-genai": "2.1.18",
     "@langchain/langgraph": "1.4.4",
     "@langchain/langgraph-checkpoint": "1.1.2",
+    "@langchain/mistralai": "1.0.0",
     "@langchain/ollama": "1.2.7",
     "@langchain/openai": "1.4.7",
     "@langchain/xai": "1.3.19",
@@ -65,5 +66,5 @@
   "keywords": [
     "AI"
   ],
-  "gitHead": "1ccf891d837e21089f65f84b892407b34a0a0cb9"
+  "gitHead": "2a8c3e0e9362aeb49b39c1d02358639b29769f32"
 }