@nocobase/plugin-ai 2.2.18 → 2.2.20

package/dist/server/ai-employees/ai-employee.js

@@ -133,9 +133,12 @@ class AIEmployee {
     return this.employee.toJSON();
   }
   async getFormatMessages(userMessages) {
-    const { provider } = await this.plugin.aiManager.getLLMService({
-      ...this.model
-    });
+    const { provider } = await this.plugin.aiManager.getLLMService(
+      {
+        ...this.model
+      },
+      { sessionId: this.sessionId }
+    );
     const { messages } = await this.aiChatConversation.getChatContext({
       userMessages,
       formatMessages: (messages2) => this.formatMessages({ messages: messages2, provider })
@@ -211,9 +214,12 @@ class AIEmployee {
     userDecisions
   }) {
     var _a, _b;
-    const { provider, model, service } = await this.plugin.aiManager.getLLMService({
-      ...this.model
-    });
+    const { provider, model, service } = await this.plugin.aiManager.getLLMService(
+      {
+        ...this.model
+      },
+      { sessionId: this.sessionId }
+    );
     this.userMessageCount = (userMessages ?? []).filter((message) => message.role === "user").length;
     const { historyMessages, tools, resolvedTools, middleware, config, state } = await this.initSession({
       messageId,
@@ -222,12 +228,14 @@ class AIEmployee {
       providerName: service.provider,
       llmService: ((_a = service.get) == null ? void 0 : _a.call(service, "name")) || service.name
     });
+    const contextMessages = [...historyMessages, ...userMessages ?? []];
+    const knowledgeBaseContext = await this.retrieveKnowledgeBase(contextMessages);
     const chatContext = await this.aiChatConversation.getChatContext({
-      userMessages: [...historyMessages, ...userMessages ?? []],
+      userMessages: contextMessages,
       userDecisions,
       tools: resolvedTools,
       middleware,
-      getSystemPrompt: (userMessages2) => this.getSystemPrompt(userMessages2),
+      getSystemPrompt: (userMessages2) => this.getSystemPrompt(userMessages2, knowledgeBaseContext),
       formatMessages: (messages) => this.formatMessages({ messages, provider })
     });
     return {
@@ -237,7 +245,8 @@ class AIEmployee {
       provider,
       chatContext,
       config,
-      state
+      state,
+      knowledgeBaseDocuments: knowledgeBaseContext.documents
     };
   }
   async stream({
@@ -253,7 +262,7 @@ class AIEmployee {
       }
     });
     try {
-      const { providerName, llmService, model, provider, chatContext, config, state } = await this.buildChatContext({
+      const { providerName, llmService, model, provider, chatContext, config, state, knowledgeBaseDocuments } = await this.buildChatContext({
         messageId,
         userMessages,
         userDecisions
@@ -272,7 +281,8 @@ class AIEmployee {
         llmService,
         model,
         provider,
-        responseMetadata
+        responseMetadata,
+        knowledgeBaseDocuments
       });
       return true;
     } catch (err) {
@@ -422,7 +432,23 @@ class AIEmployee {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z;
     const aiMessageIdMap = /* @__PURE__ */ new Map();
     const persistedAIMessageIdMap = /* @__PURE__ */ new Map();
-    const { signal, providerName, llmService, model, provider, responseMetadata, allowEmpty = false } = options;
+    const {
+      signal,
+      providerName,
+      llmService,
+      model,
+      provider,
+      responseMetadata,
+      allowEmpty = false,
+      knowledgeBaseDocuments = []
+    } = options;
+    const knowledgeBaseReferences = /* @__PURE__ */ new Map();
+    const collectKnowledgeBaseReferences = (documents) => {
+      for (const document of documents) {
+        knowledgeBaseReferences.set(String(document.id), document);
+      }
+    };
+    let lastAIMessageId;
     const reasoningState = new import_reasoning_stream_state.ReasoningStreamState();
     const stopReasoning = async (conversation) => {
       if (reasoningState.stop(conversation)) {
@@ -443,7 +469,7 @@ class AIEmployee {
           try {
             await stopAllReasoning();
             if ((gathered == null ? void 0 : gathered.type) === "ai") {
-              await this.finalizeAbortedAIMessage({
+              const result = await this.finalizeAbortedAIMessage({
                 aiMessage: gathered,
                 providerName,
                 provider,
@@ -451,6 +477,9 @@ class AIEmployee {
                 model,
                 knownMessageId: persistedAIMessageIdMap.get(gathered.id)
               });
+              if (result) {
+                lastAIMessageId = result.message.messageId;
+              }
             }
           } catch (e) {
             this.logger.error("Fail to save message after conversation abort", gathered);
@@ -474,6 +503,10 @@ class AIEmployee {
         username: this.employee.username
       };
       await this.protocol.with(aiEmployeeConversation).startStream();
+      if (knowledgeBaseDocuments.length) {
+        collectKnowledgeBaseReferences(knowledgeBaseDocuments);
+        await this.protocol.with(aiEmployeeConversation).knowledgeBaseReferences({ documents: knowledgeBaseDocuments });
+      }
       for await (const [mode, chunks] of stream) {
         if (mode === "messages") {
           const [chunk, metadata] = chunks;
@@ -530,6 +563,9 @@ class AIEmployee {
             await this.streamCached.skipped();
             aiMessageIdMap.set(currentConversation.sessionId, chunks.body.messageId);
             persistedAIMessageIdMap.set(chunks.body.id, chunks.body.messageId);
+            if (currentConversation.sessionId === this.sessionId) {
+              lastAIMessageId = chunks.body.messageId;
+            }
             const data = responseMetadata.get(chunks.body.id);
             if (data) {
               const savedMessage = await this.aiMessagesModel.findOne({
@@ -616,6 +652,13 @@ class AIEmployee {
             await this.protocol.with(currentConversation).newMessage();
           } else if (chunks.action === "afterSubAgentInvoke") {
             await this.protocol.with(currentConversation).subAgentCompleted();
+          } else if (chunks.action === "knowledgeBaseRetrieved") {
+            const { toolCallId, documents = [] } = chunks.body ?? {};
+            collectKnowledgeBaseReferences(documents);
+            await this.protocol.with({
+              ...currentConversation,
+              from: currentConversation.sessionId === this.sessionId ? this.from : "sub-agent"
+            }).knowledgeBaseReferences({ toolCallId, documents });
           }
         }
       }
@@ -635,11 +678,32 @@ class AIEmployee {
       }
     } finally {
       await abortFinalization;
+      await this.saveKnowledgeBaseReferences(lastAIMessageId, Array.from(knowledgeBaseReferences.values()));
       if (this.from === "main-agent") {
         this.ctx.res.end();
       }
     }
   }
+  async saveKnowledgeBaseReferences(messageId, documents) {
+    if (!messageId || !documents.length) {
+      return;
+    }
+    try {
+      const savedMessage = await this.aiMessagesModel.findOne({
+        where: { sessionId: this.sessionId, messageId }
+      });
+      if (!savedMessage) {
+        return;
+      }
+      const metadata = savedMessage.get("metadata") ?? {};
+      await this.aiMessagesModel.update(
+        { metadata: { ...metadata, knowledgeBaseReferences: documents } },
+        { where: { sessionId: this.sessionId, messageId } }
+      );
+    } catch (error) {
+      this.logger.error("Failed to save knowledge base references", { sessionId: this.sessionId, messageId, error });
+    }
+  }
   async handleInterruptedToolCalls(interrupt, getMessageId, onInterrupted) {
     const interruptId = interrupt == null ? void 0 : interrupt.id;
     const interruptActions = this.toInterruptActions(interrupt == null ? void 0 : interrupt.value);
@@ -715,14 +779,10 @@ Field: ${field.name}, Title: ${(_a = field.options.uiSchema) == null ? void 0 :
     }
     return message;
   }
-  async getSystemPrompt(userMessages) {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
-    if (this.systemPromptMode === "none") {
-      return "";
-    }
-    const about = await (0, import_utils.parseVariables)(this.ctx, this.employee.about ?? this.employee.defaultPrompt ?? "");
-    if (this.systemPromptMode === "raw") {
-      return about;
+  async retrieveKnowledgeBase(userMessages) {
+    var _a;
+    if (this.systemPromptMode !== "default") {
+      return { documents: [] };
     }
     const employee = this.getAIEmployeeRecord();
     const knowledgeBaseManager = this.plugin.knowledgeBaseManager;
@@ -731,9 +791,37 @@ Field: ${field.name}, Title: ${(_a = field.options.uiSchema) == null ? void 0 :
     const hasAccessibleKnowledgeBase = knowledgeBaseEnabled ? await knowledgeBaseManager.hasAccessibleKnowledgeBase({ employee, roleNames }) : false;
     const knowledgeBaseAccessDenied = knowledgeBaseEnabled && !hasAccessibleKnowledgeBase;
     const knowledgeBaseOnDemand = knowledgeBaseEnabled && hasAccessibleKnowledgeBase && (0, import_ai_knowledge_base.normalizeKnowledgeBaseRetrievalStrategy)((_a = employee.knowledgeBase) == null ? void 0 : _a.retrievalStrategy) === "onDemand";
+    let prompt;
+    let documents = [];
+    if (knowledgeBaseEnabled && hasAccessibleKnowledgeBase && !knowledgeBaseOnDemand && (userMessages == null ? void 0 : userMessages.length)) {
+      const lastUserMessage = userMessages.filter((x) => x.role === "user").at(-1);
+      if (lastUserMessage) {
+        ({ prompt, documents } = await knowledgeBaseManager.retrievePrompt({
+          employee,
+          query: lastUserMessage.content.content,
+          roleNames
+        }));
+      }
+    }
+    const backgroundPrompt = (0, import_ai_knowledge_base.getKnowledgeBaseBackgroundPrompt)({
+      accessDenied: knowledgeBaseAccessDenied,
+      onDemand: knowledgeBaseOnDemand,
+      preRetrieved: Boolean(prompt)
+    });
+    return { prompt, backgroundPrompt, documents };
+  }
+  async getSystemPrompt(userMessages, knowledgeBaseContext) {
+    var _a, _b, _c, _d, _e, _f, _g;
+    if (this.systemPromptMode === "none") {
+      return "";
+    }
+    const about = await (0, import_utils.parseVariables)(this.ctx, this.employee.about ?? this.employee.defaultPrompt ?? "");
+    if (this.systemPromptMode === "raw") {
+      return about;
+    }
     const userConfig = await this.db.getRepository("usersAiEmployees").findOne({
       filter: {
-        userId: ((_b = this.ctx.auth) == null ? void 0 : _b.user.id) ?? 0,
+        userId: ((_a = this.ctx.auth) == null ? void 0 : _a.user.id) ?? 0,
         aiEmployee: this.employee.username
       }
     });
@@ -757,25 +845,9 @@ ${workContextBackground.join("\n")}`;
       background = `${background}
 ${addSystemPrompt.map((it) => it.content).join("\n")}`;
     }
-    let knowledgeBase;
-    if (knowledgeBaseEnabled && hasAccessibleKnowledgeBase && !knowledgeBaseOnDemand && (userMessages == null ? void 0 : userMessages.length)) {
-      const lastUserMessage = userMessages.filter((x) => x.role === "user").at(-1);
-      if (lastUserMessage) {
-        knowledgeBase = await knowledgeBaseManager.retrievePrompt({
-          employee,
-          query: lastUserMessage.content.content,
-          roleNames
-        });
-      }
-    }
-    const knowledgeBaseBackgroundPrompt = (0, import_ai_knowledge_base.getKnowledgeBaseBackgroundPrompt)({
-      accessDenied: knowledgeBaseAccessDenied,
-      onDemand: knowledgeBaseOnDemand,
-      preRetrieved: Boolean(knowledgeBase)
-    });
-    if (knowledgeBaseBackgroundPrompt) {
+    if (knowledgeBaseContext == null ? void 0 : knowledgeBaseContext.backgroundPrompt) {
       background = `${background}
-${knowledgeBaseBackgroundPrompt}`;
+${knowledgeBaseContext.backgroundPrompt}`;
     }
     const availableSkills = await this.getAvailableSkills();
     const availableAIEmployees = await this.getAvailableAIEmployees();
@@ -790,20 +862,20 @@ ${knowledgeBaseBackgroundPrompt}`;
       personal: userConfig == null ? void 0 : userConfig.prompt,
       environment: {
         database: this.db.sequelize.getDialect(),
-        locale: ((_d = (_c = this.ctx).getCurrentLocale) == null ? void 0 : _d.call(_c)) || "en-US",
+        locale: ((_c = (_b = this.ctx).getCurrentLocale) == null ? void 0 : _c.call(_b)) || "en-US",
         currentDateTime: getCurrentDateTimeForPrompt(
           await this.aiChatConversation.getCreatedAt(),
-          (_f = (_e = this.ctx).getCurrentLocale) == null ? void 0 : _f.call(_e),
+          (_e = (_d = this.ctx).getCurrentLocale) == null ? void 0 : _e.call(_d),
           getCurrentTimezone(this.ctx)
         ),
         timezone: getCurrentTimezone(this.ctx)
       },
-      knowledgeBase,
+      knowledgeBase: knowledgeBaseContext == null ? void 0 : knowledgeBaseContext.prompt,
       availableSkills,
       availableAIEmployees,
       webSearch: this.webSearch
     });
-    const { important } = ((_h = (_g = this.ctx.action) == null ? void 0 : _g.params) == null ? void 0 : _h.values) || {};
+    const { important } = ((_g = (_f = this.ctx.action) == null ? void 0 : _f.params) == null ? void 0 : _g.values) || {};
     if (important === "GraphRecursionError") {
       const importantPrompt = `<Important>You have already called tools multiple times and gathered sufficient information.
 First, provide a summary based on the existing information. Do not call additional tools.
@@ -1186,9 +1258,12 @@ If information is missing, clearly state it in the summary.</Important>`;
     if (!toolMessages || import_lodash.default.isEmpty(toolMessages)) {
       return;
     }
-    const { model, service } = await this.plugin.aiManager.getLLMService({
-      ...this.model
-    });
+    const { model, service } = await this.plugin.aiManager.getLLMService(
+      {
+        ...this.model
+      },
+      { sessionId: this.sessionId }
+    );
     const toolCallMap = await this.getToolCallMap(messageId);
     const now = /* @__PURE__ */ new Date();
     const toolMessageContent = "The user ignored the application for tools usage and will continued to ask questions";
@@ -1802,6 +1877,9 @@ class ChatStreamProtocol {
       subAgentCompleted: async () => {
         await write({ type: "sub_agent_completed" });
       },
+      knowledgeBaseReferences: async (body) => {
+        await write({ type: "knowledge_base_references", body });
+      },
       newMessage: async (content) => {
         await write({ type: "new_message", body: content });
       },

package/dist/server/ai-employees/ai-knowledge-base.d.ts

@@ -8,6 +8,7 @@
  */
 import { AIEmployee } from '../../collections/ai-employees';
 import type PluginAIServer from '../plugin';
+import type { KnowledgeBaseReference } from '../types';
 export declare const KNOWLEDGE_BASE_RETRIEVAL_STRATEGIES: readonly ["always", "onDemand"];
 export type KnowledgeBaseRetrievalStrategy = (typeof KNOWLEDGE_BASE_RETRIEVAL_STRATEGIES)[number];
 export declare const isKnowledgeBaseRetrievalStrategy: (value: unknown) => value is "always" | "onDemand";
@@ -28,6 +29,10 @@ export type KnowledgeBaseRetrieveOptions = {
     query: string;
     roleNames?: string[];
 };
+export type KnowledgeBaseRetrieveResult = {
+    prompt: string;
+    documents: KnowledgeBaseReference[];
+};
 export type KnowledgeBaseAccessOptions = {
     employee: AIEmployee;
     roleNames: string[];
@@ -35,7 +40,8 @@ export type KnowledgeBaseAccessOptions = {
 export declare class KnowledgeBaseManager {
     private readonly plugin;
     constructor(plugin: PluginAIServer);
-    retrievePrompt({ username, employee, query, roleNames }: KnowledgeBaseRetrieveOptions): Promise<string>;
+    retrievePrompt({ username, employee, query, roleNames, }: KnowledgeBaseRetrieveOptions): Promise<KnowledgeBaseRetrieveResult>;
+    private getDocumentReferences;
     hasAccessibleKnowledgeBase({ employee, roleNames }: KnowledgeBaseAccessOptions): Promise<boolean>;
     isEnabledKnowledgeBase(username: string): Promise<boolean>;
     isEnabledKnowledgeBase(employee: AIEmployee): Promise<boolean>;

package/dist/server/ai-employees/ai-knowledge-base.js

@@ -115,27 +115,50 @@ class KnowledgeBaseManager {
   constructor(plugin) {
     this.plugin = plugin;
   }
-  async retrievePrompt({ username, employee, query, roleNames }) {
+  async retrievePrompt({
+    username,
+    employee,
+    query,
+    roleNames
+  }) {
     employee = employee ?? (username ? await this.getEmployee(username) : null);
     if (!employee) {
-      return "Specified knowledge base not existed";
+      return { prompt: "Specified knowledge base not existed", documents: [] };
     }
     const { knowledgeBaseKeys = [], topK, score } = employee.knowledgeBase ?? {};
     const knowledgeBasePrompt = employee.knowledgeBasePrompt ?? "{knowledgeBaseData}";
     const docs = await this.plugin.features.knowledgeBase.search({ knowledgeBaseKeys, query, topK, score, roleNames });
     if (!(docs == null ? void 0 : docs.length)) {
-      return "No document match in knowledge base";
+      return { prompt: "No document match in knowledge base", documents: [] };
     }
     const knowledgeBaseData = docs.map((doc) => buildKnowledgeBaseContent(doc.content, doc.metadata)).join("\n");
     if (import_lodash.default.isEmpty(knowledgeBaseData)) {
-      return "No document match in knowledge base";
+      return { prompt: "No document match in knowledge base", documents: [] };
     }
+    const documents = await this.getDocumentReferences(docs);
     if (!(0, import_ai_employee_validation.hasKnowledgeBaseDataPlaceholder)(knowledgeBasePrompt)) {
-      return `${knowledgeBasePrompt}
+      return { prompt: `${knowledgeBasePrompt}
 
-${knowledgeBaseData}`;
+${knowledgeBaseData}`, documents };
     }
-    return import_prompts.ChatPromptTemplate.fromTemplate(knowledgeBasePrompt).format({ knowledgeBaseData });
+    const prompt = await import_prompts.ChatPromptTemplate.fromTemplate(knowledgeBasePrompt).format({ knowledgeBaseData });
+    return { prompt, documents };
+  }
+  async getDocumentReferences(docs) {
+    const knowledgeBaseFeature = this.plugin.features.knowledgeBase;
+    if (!knowledgeBaseFeature.getDocumentReferences) {
+      return [];
+    }
+    const knowledgeBaseDocsIds = import_lodash.default.uniq(
+      docs.map((doc) => {
+        var _a;
+        return (_a = doc.metadata) == null ? void 0 : _a.knowledgeBaseDocsId;
+      }).filter((id) => typeof id === "number" || typeof id === "string" && id !== "")
+    );
+    if (!knowledgeBaseDocsIds.length) {
+      return [];
+    }
+    return await knowledgeBaseFeature.getDocumentReferences(knowledgeBaseDocsIds);
   }
   async hasAccessibleKnowledgeBase({ employee, roleNames }) {
     var _a;

package/dist/server/features/knowledge-base.d.ts

@@ -6,11 +6,12 @@
  * 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 { DocumentSegmentedWithScore, KnowledgeBase, KnowledgeBaseGroup, SearchOptions } from '../types';
+import { DocumentSegmentedWithScore, KnowledgeBase, KnowledgeBaseGroup, KnowledgeBaseReference, SearchOptions } from '../types';
 export type KnowledgeBaseAccessOptions = Pick<SearchOptions, 'knowledgeBaseKeys' | 'roleNames'>;
 export interface KnowledgeBaseFeature {
     getKnowledgeBase(knowledgeBaseKeys: string[]): Promise<KnowledgeBase[]>;
     getKnowledgeBaseGroup(knowledgeBaseKeys: string[]): Promise<KnowledgeBaseGroup[]>;
     getAccessibleKnowledgeBaseKeys(options: KnowledgeBaseAccessOptions): Promise<string[]>;
     search(options: SearchOptions): Promise<DocumentSegmentedWithScore[]>;
+    getDocumentReferences?(knowledgeBaseDocsIds: Array<number | string>): Promise<KnowledgeBaseReference[]>;
 }

package/dist/server/index.d.ts

@@ -7,7 +7,7 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 export { default } from './plugin';
-export { LLMProvider, LLMProviderOptions } from './llm-providers/provider';
+export { LLMProvider, LLMProviderOptions, LLMProviderRequestContext } from './llm-providers/provider';
 export { LLMProviderMeta } from './manager/ai-manager';
 export { ToolOptions } from './manager/tool-manager';
 export { DocumentLoader } from './document-loader';

package/dist/server/index.js

@@ -41,6 +41,7 @@ __export(server_exports, {
   LLMProvider: () => import_provider.LLMProvider,
   LLMProviderMeta: () => import_ai_manager.LLMProviderMeta,
   LLMProviderOptions: () => import_provider.LLMProviderOptions,
+  LLMProviderRequestContext: () => import_provider.LLMProviderRequestContext,
   ToolOptions: () => import_tool_manager.ToolOptions,
   default: () => import_plugin.default
 });
@@ -58,5 +59,6 @@ var import_ai_employee = require("./ai-employees/ai-employee");
   LLMProvider,
   LLMProviderMeta,
   LLMProviderOptions,
+  LLMProviderRequestContext,
   ToolOptions
 });

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

@@ -72,6 +72,9 @@ class AnthropicProvider extends import_provider.LLMProvider {
       ...sanitizedModelOptions,
       model,
       anthropicApiUrl: this.getResolvedBaseURL(),
+      clientOptions: {
+        defaultHeaders: this.getDefaultHeaders()
+      },
       verbose: false
     });
   }
@@ -96,6 +99,7 @@ class AnthropicProvider extends import_provider.LLMProvider {
         method: "GET",
         url,
         headers: {
+          ...this.getDefaultHeaders(),
           "x-api-key": apiKey,
           "anthropic-version": "2023-06-01"
         }

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

@@ -83,7 +83,8 @@ class DashscopeProvider extends import_provider.LLMProvider {
         ...reasoningOptions.modelKwargs || {}
       },
       configuration: {
-        baseURL: this.getResolvedBaseURL()
+        baseURL: this.getResolvedBaseURL(),
+        defaultHeaders: this.getDefaultHeaders()
       },
       verbose: false
     });

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

@@ -12,6 +12,7 @@ import { ChatOpenAI } from '@langchain/openai';
 import type { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
 import type OpenAI from 'openai';
 import { Model } from '@nocobase/database';
+import type { AttachmentModel } from '@nocobase/plugin-file-manager';
 import { LLMProvider, ReasoningOptions, ResolvedReasoningOptions } from '../provider';
 import { LLMProviderMeta } from '../../manager/ai-manager';
 import { DeepSeekReasoningConfig } from './reasoning';
@@ -51,7 +52,7 @@ export declare class DeepSeekProvider extends LLMProvider {
     }[];
     protected resolveReasoningOptions(reasoning?: ReasoningOptions): ResolvedReasoningOptions;
     isToolConflict(): boolean;
-    protected isApiSupportedAttachment(): boolean;
+    protected isApiSupportedAttachment(attachment: AttachmentModel): boolean;
 }
 export declare const deepseekProviderOptions: LLMProviderMeta;
 export {};

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

@@ -166,7 +166,8 @@ class DeepSeekProvider extends import_provider.LLMProvider {
           ...reasoningParams
         },
         configuration: {
-          baseURL: this.getResolvedBaseURL()
+          baseURL: this.getResolvedBaseURL(),
+          defaultHeaders: this.getDefaultHeaders()
         },
         verbose: false
       };
@@ -190,7 +191,8 @@ class DeepSeekProvider extends import_provider.LLMProvider {
         ...modelOptions,
         modelKwargs,
         configuration: {
-          baseURL: this.getResolvedBaseURL()
+          baseURL: this.getResolvedBaseURL(),
+          defaultHeaders: this.getDefaultHeaders()
         },
         verbose: false
       },
@@ -299,18 +301,26 @@ class DeepSeekProvider extends import_provider.LLMProvider {
   isToolConflict() {
     return false;
   }
-  isApiSupportedAttachment() {
-    return false;
+  isApiSupportedAttachment(attachment) {
+    var _a, _b;
+    const capabilities = (0, import_reasoning2.getDeepSeekModelCapabilities)(String((_a = this.modelOptions) == null ? void 0 : _a.model));
+    return (capabilities == null ? void 0 : capabilities.supportsImageInput) === true && (((_b = attachment == null ? void 0 : attachment.mimetype) == null ? void 0 : _b.startsWith("image/")) ?? false);
   }
 }
 const deepseekProviderOptions = {
   title: "DeepSeek",
   supportedModel: [import_ai_manager.SupportedModel.LLM],
   models: {
-    [import_ai_manager.SupportedModel.LLM]: ["deepseek-v4-pro", "deepseek-v4-flash", "deepseek-chat", "deepseek-reasoner"]
+    [import_ai_manager.SupportedModel.LLM]: [
+      "deepseek-v4-pro",
+      "deepseek-flash",
+      "deepseek-v4-flash",
+      "deepseek-chat",
+      "deepseek-reasoner"
+    ]
   },
   supportWebSearch: true,
-  webSearchModels: ["deepseek-v4-flash", "deepseek-v4-pro"],
+  webSearchModels: ["deepseek-v4-pro"],
   provider: DeepSeekProvider
 };
 // Annotate the CommonJS export names for ESM import in node:

package/dist/server/llm-providers/deepseek/reasoning.d.ts

@@ -20,6 +20,7 @@ type DeepSeekModelCapabilities = {
     supportsFunctionTools: boolean;
     supportsStructuredOutput: boolean;
     supportsWebSearch: boolean;
+    supportsImageInput: boolean;
 };
 export type DeepSeekReasoningConfig = {
     protocol: DeepSeekApiProtocol;
@@ -34,6 +35,16 @@ type ReasoningSource = {
     response_metadata?: Record<string, unknown>;
 };
 export declare const DEEPSEEK_MODEL_CAPABILITIES: {
+    readonly 'deepseek-flash': {
+        readonly protocol: "responses";
+        readonly reasoning: "switchable";
+        readonly defaultThinking: "enabled";
+        readonly efforts: Set<DeepSeekReasoningEffort>;
+        readonly supportsFunctionTools: true;
+        readonly supportsStructuredOutput: true;
+        readonly supportsWebSearch: false;
+        readonly supportsImageInput: true;
+    };
     readonly 'deepseek-v4-flash': {
         readonly protocol: "responses";
         readonly reasoning: "switchable";
@@ -41,7 +52,28 @@ export declare const DEEPSEEK_MODEL_CAPABILITIES: {
         readonly efforts: Set<DeepSeekReasoningEffort>;
         readonly supportsFunctionTools: true;
         readonly supportsStructuredOutput: true;
-        readonly supportsWebSearch: true;
+        readonly supportsWebSearch: false;
+        readonly supportsImageInput: true;
+    };
+    readonly 'deepseek-v4-flash-vision-exp': {
+        readonly protocol: "responses";
+        readonly reasoning: "switchable";
+        readonly defaultThinking: "enabled";
+        readonly efforts: Set<DeepSeekReasoningEffort>;
+        readonly supportsFunctionTools: true;
+        readonly supportsStructuredOutput: true;
+        readonly supportsWebSearch: false;
+        readonly supportsImageInput: true;
+    };
+    readonly 'deepseek-v4.1-flash': {
+        readonly protocol: "responses";
+        readonly reasoning: "switchable";
+        readonly defaultThinking: "enabled";
+        readonly efforts: Set<DeepSeekReasoningEffort>;
+        readonly supportsFunctionTools: true;
+        readonly supportsStructuredOutput: true;
+        readonly supportsWebSearch: false;
+        readonly supportsImageInput: true;
     };
     readonly 'deepseek-v4-pro': {
         readonly protocol: "responses";
@@ -51,6 +83,7 @@ export declare const DEEPSEEK_MODEL_CAPABILITIES: {
         readonly supportsFunctionTools: true;
         readonly supportsStructuredOutput: true;
         readonly supportsWebSearch: true;
+        readonly supportsImageInput: false;
     };
     readonly 'deepseek-chat': {
         readonly protocol: "chat-completions";
@@ -60,6 +93,7 @@ export declare const DEEPSEEK_MODEL_CAPABILITIES: {
         readonly supportsFunctionTools: true;
         readonly supportsStructuredOutput: true;
         readonly supportsWebSearch: false;
+        readonly supportsImageInput: false;
     };
     readonly 'deepseek-reasoner': {
         readonly protocol: "chat-completions";
@@ -69,6 +103,7 @@ export declare const DEEPSEEK_MODEL_CAPABILITIES: {
         readonly supportsFunctionTools: true;
         readonly supportsStructuredOutput: true;
         readonly supportsWebSearch: false;
+        readonly supportsImageInput: false;
     };
 };
 export type OfficialDeepSeekModel = keyof typeof DEEPSEEK_MODEL_CAPABILITIES;