@nocobase/plugin-ai 2.2.0-beta.3 → 2.2.0-beta.6

package/dist/node_modules/openai/src/resources/videos.ts

@@ -6,43 +6,65 @@ import { ConversationCursorPage, type ConversationCursorPageParams, PagePromise
 import { type Uploadable } from '../core/uploads';
 import { buildHeaders } from '../internal/headers';
 import { RequestOptions } from '../internal/request-options';
-import { maybeMultipartFormRequestOptions } from '../internal/uploads';
+import { maybeMultipartFormRequestOptions, multipartFormRequestOptions } from '../internal/uploads';
 import { path } from '../internal/utils/path';
 
 export class Videos extends APIResource {
   /**
-   * Create a video
+   * Create a new video generation job from a prompt and optional reference assets.
    */
   create(body: VideoCreateParams, options?: RequestOptions): APIPromise<Video> {
-    return this._client.post('/videos', maybeMultipartFormRequestOptions({ body, ...options }, this._client));
+    return this._client.post(
+      '/videos',
+      multipartFormRequestOptions({ body, ...options, __security: { bearerAuth: true } }, this._client),
+    );
   }
 
   /**
-   * Retrieve a video
+   * Fetch the latest metadata for a generated video.
    */
   retrieve(videoID: string, options?: RequestOptions): APIPromise<Video> {
-    return this._client.get(path`/videos/${videoID}`, options);
+    return this._client.get(path`/videos/${videoID}`, { ...options, __security: { bearerAuth: true } });
   }
 
   /**
-   * List videos
+   * List recently generated videos for the current project.
    */
   list(
     query: VideoListParams | null | undefined = {},
     options?: RequestOptions,
   ): PagePromise<VideosPage, Video> {
-    return this._client.getAPIList('/videos', ConversationCursorPage<Video>, { query, ...options });
+    return this._client.getAPIList('/videos', ConversationCursorPage<Video>, {
+      query,
+      ...options,
+      __security: { bearerAuth: true },
+    });
   }
 
   /**
-   * Delete a video
+   * Permanently delete a completed or failed video and its stored assets.
    */
   delete(videoID: string, options?: RequestOptions): APIPromise<VideoDeleteResponse> {
-    return this._client.delete(path`/videos/${videoID}`, options);
+    return this._client.delete(path`/videos/${videoID}`, { ...options, __security: { bearerAuth: true } });
+  }
+
+  /**
+   * Create a character from an uploaded video.
+   */
+  createCharacter(
+    body: VideoCreateCharacterParams,
+    options?: RequestOptions,
+  ): APIPromise<VideoCreateCharacterResponse> {
+    return this._client.post(
+      '/videos/characters',
+      multipartFormRequestOptions({ body, ...options, __security: { bearerAuth: true } }, this._client),
+    );
   }
 
   /**
-   * Download video content
+   * Download the generated video bytes or a derived preview asset.
+   *
+   * Streams the rendered video content for the specified video job.
    */
   downloadContent(
     videoID: string,
@@ -53,23 +75,64 @@ export class Videos extends APIResource {
       query,
       ...options,
       headers: buildHeaders([{ Accept: 'application/binary' }, options?.headers]),
+      __security: { bearerAuth: true },
       __binaryResponse: true,
     });
   }
 
   /**
-   * Create a video remix
+   * Create a new video generation job by editing a source video or existing
+   * generated video.
+   */
+  edit(body: VideoEditParams, options?: RequestOptions): APIPromise<Video> {
+    return this._client.post(
+      '/videos/edits',
+      multipartFormRequestOptions({ body, ...options, __security: { bearerAuth: true } }, this._client),
+    );
+  }
+
+  /**
+   * Create an extension of a completed video.
+   */
+  extend(body: VideoExtendParams, options?: RequestOptions): APIPromise<Video> {
+    return this._client.post(
+      '/videos/extensions',
+      multipartFormRequestOptions({ body, ...options, __security: { bearerAuth: true } }, this._client),
+    );
+  }
+
+  /**
+   * Fetch a character.
+   */
+  getCharacter(characterID: string, options?: RequestOptions): APIPromise<VideoGetCharacterResponse> {
+    return this._client.get(path`/videos/characters/${characterID}`, {
+      ...options,
+      __security: { bearerAuth: true },
+    });
+  }
+
+  /**
+   * Create a remix of a completed video using a refreshed prompt.
    */
   remix(videoID: string, body: VideoRemixParams, options?: RequestOptions): APIPromise<Video> {
     return this._client.post(
       path`/videos/${videoID}/remix`,
-      maybeMultipartFormRequestOptions({ body, ...options }, this._client),
+      maybeMultipartFormRequestOptions({ body, ...options, __security: { bearerAuth: true } }, this._client),
     );
   }
 }
 
 export type VideosPage = ConversationCursorPage<Video>;
 
+export interface ImageInputReferenceParam {
+  file_id?: string;
+
+  /**
+   * A fully qualified URL or base64-encoded data URL.
+   */
+  image_url?: string;
+}
+
 /**
  * Structured information describing a generated video job.
  */
@@ -125,9 +188,10 @@ export interface Video {
   remixed_from_video_id: string | null;
 
   /**
-   * Duration of the generated clip in seconds.
+   * Duration of the generated clip in seconds. For extensions, this is the stitched
+   * total duration.
    */
-  seconds: VideoSeconds;
+  seconds: (string & {}) | VideoSeconds;
 
   /**
    * The resolution of the generated video.
@@ -187,6 +251,40 @@ export interface VideoDeleteResponse {
   object: 'video.deleted';
 }
 
+export interface VideoCreateCharacterResponse {
+  /**
+   * Identifier for the character creation cameo.
+   */
+  id: string | null;
+
+  /**
+   * Unix timestamp (in seconds) when the character was created.
+   */
+  created_at: number;
+
+  /**
+   * Display name for the character.
+   */
+  name: string | null;
+}
+
+export interface VideoGetCharacterResponse {
+  /**
+   * Identifier for the character creation cameo.
+   */
+  id: string | null;
+
+  /**
+   * Unix timestamp (in seconds) when the character was created.
+   */
+  created_at: number;
+
+  /**
+   * Display name for the character.
+   */
+  name: string | null;
+}
+
 export interface VideoCreateParams {
   /**
    * Text prompt that describes the video to generate.
@@ -194,9 +292,9 @@ export interface VideoCreateParams {
   prompt: string;
 
   /**
-   * Optional image reference that guides generation.
+   * Optional reference asset upload or reference object that guides generation.
    */
-  input_reference?: Uploadable;
+  input_reference?: Uploadable | ImageInputReferenceParam;
 
   /**
    * The video generation model to use (allowed values: sora-2, sora-2-pro). Defaults
@@ -224,6 +322,18 @@ export interface VideoListParams extends ConversationCursorPageParams {
   order?: 'asc' | 'desc';
 }
 
+export interface VideoCreateCharacterParams {
+  /**
+   * Display name for this API character.
+   */
+  name: string;
+
+  /**
+   * Video file used to create a character.
+   */
+  video: Uploadable;
+}
+
 export interface VideoDownloadContentParams {
   /**
    * Which downloadable asset to return. Defaults to the MP4 video.
@@ -231,6 +341,60 @@ export interface VideoDownloadContentParams {
   variant?: 'video' | 'thumbnail' | 'spritesheet';
 }
 
+export interface VideoEditParams {
+  /**
+   * Text prompt that describes how to edit the source video.
+   */
+  prompt: string;
+
+  /**
+   * Reference to the completed video to edit.
+   */
+  video: Uploadable | VideoEditParams.VideoReferenceInputParam;
+}
+
+export namespace VideoEditParams {
+  /**
+   * Reference to the completed video.
+   */
+  export interface VideoReferenceInputParam {
+    /**
+     * The identifier of the completed video.
+     */
+    id: string;
+  }
+}
+
+export interface VideoExtendParams {
+  /**
+   * Updated text prompt that directs the extension generation.
+   */
+  prompt: string;
+
+  /**
+   * Length of the newly generated extension segment in seconds (allowed values: 4,
+   * 8, 12, 16, 20).
+   */
+  seconds: VideoSeconds;
+
+  /**
+   * Reference to the completed video to extend.
+   */
+  video: Uploadable | VideoExtendParams.VideoReferenceInputParam;
+}
+
+export namespace VideoExtendParams {
+  /**
+   * Reference to the completed video.
+   */
+  export interface VideoReferenceInputParam {
+    /**
+     * The identifier of the completed video.
+     */
+    id: string;
+  }
+}
+
 export interface VideoRemixParams {
   /**
    * Updated text prompt that directs the remix generation.
@@ -240,16 +404,22 @@ export interface VideoRemixParams {
 
 export declare namespace Videos {
   export {
+    type ImageInputReferenceParam as ImageInputReferenceParam,
     type Video as Video,
     type VideoCreateError as VideoCreateError,
     type VideoModel as VideoModel,
     type VideoSeconds as VideoSeconds,
     type VideoSize as VideoSize,
     type VideoDeleteResponse as VideoDeleteResponse,
+    type VideoCreateCharacterResponse as VideoCreateCharacterResponse,
+    type VideoGetCharacterResponse as VideoGetCharacterResponse,
     type VideosPage as VideosPage,
     type VideoCreateParams as VideoCreateParams,
     type VideoListParams as VideoListParams,
+    type VideoCreateCharacterParams as VideoCreateCharacterParams,
     type VideoDownloadContentParams as VideoDownloadContentParams,
+    type VideoEditParams as VideoEditParams,
+    type VideoExtendParams as VideoExtendParams,
     type VideoRemixParams as VideoRemixParams,
   };
 }

package/dist/node_modules/openai/src/version.ts

@@ -1 +1 @@
-export const VERSION = '6.22.0'; // x-release-please-version
+export const VERSION = '6.39.1'; // x-release-please-version

package/dist/node_modules/openai/streaming.d.mts

@@ -1,2 +1,3 @@
+/** @deprecated Import from ./core/streaming instead */
 export * from "./core/streaming.mjs";
 //# sourceMappingURL=streaming.d.mts.map

package/dist/node_modules/openai/streaming.d.ts

@@ -1,2 +1,3 @@
+/** @deprecated Import from ./core/streaming instead */
 export * from "./core/streaming.js";
 //# sourceMappingURL=streaming.d.ts.map

package/dist/node_modules/openai/uploads.d.mts

@@ -1,2 +1,3 @@
+/** @deprecated Import from ./core/uploads instead */
 export * from "./core/uploads.mjs";
 //# sourceMappingURL=uploads.d.mts.map

package/dist/node_modules/openai/uploads.d.ts

@@ -1,2 +1,3 @@
+/** @deprecated Import from ./core/uploads instead */
 export * from "./core/uploads.js";
 //# sourceMappingURL=uploads.d.ts.map

package/dist/node_modules/openai/version.d.mts

@@ -1,2 +1,2 @@
-export declare const VERSION = "6.22.0";
+export declare const VERSION = "6.39.1";
 //# sourceMappingURL=version.d.mts.map

package/dist/node_modules/openai/version.d.ts

@@ -1,2 +1,2 @@
-export declare const VERSION = "6.22.0";
+export declare const VERSION = "6.39.1";
 //# sourceMappingURL=version.d.ts.map

package/dist/node_modules/openai/version.js

@@ -1,5 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VERSION = void 0;
-exports.VERSION = '6.22.0'; // x-release-please-version
+exports.VERSION = '6.39.1'; // x-release-please-version
 //# sourceMappingURL=version.js.map

package/dist/node_modules/zod/package.json

@@ -1 +1 @@
-{"name":"zod","version":"4.3.5","type":"module","license":"MIT","author":"Colin McDonnell <zod@colinhacks.com>","description":"TypeScript-first schema declaration and validation library with static type inference","homepage":"https://zod.dev","llms":"https://zod.dev/llms.txt","llmsFull":"https://zod.dev/llms-full.txt","mcpServer":"https://mcp.inkeep.com/zod/mcp","funding":"https://github.com/sponsors/colinhacks","sideEffects":false,"files":["src","**/*.js","**/*.mjs","**/*.cjs","**/*.d.ts","**/*.d.mts","**/*.d.cts","**/package.json"],"keywords":["typescript","schema","validation","type","inference"],"main":"./index.cjs","types":"./index.d.cts","module":"./index.js","zshy":{"exports":{"./package.json":"./package.json",".":"./src/index.ts","./mini":"./src/mini/index.ts","./locales":"./src/locales/index.ts","./v3":"./src/v3/index.ts","./v4":"./src/v4/index.ts","./v4-mini":"./src/v4-mini/index.ts","./v4/mini":"./src/v4/mini/index.ts","./v4/core":"./src/v4/core/index.ts","./v4/locales":"./src/v4/locales/index.ts","./v4/locales/*":"./src/v4/locales/*"},"conditions":{"@zod/source":"src"}},"exports":{"./package.json":"./package.json",".":{"@zod/source":"./src/index.ts","types":"./index.d.cts","import":"./index.js","require":"./index.cjs"},"./mini":{"@zod/source":"./src/mini/index.ts","types":"./mini/index.d.cts","import":"./mini/index.js","require":"./mini/index.cjs"},"./locales":{"@zod/source":"./src/locales/index.ts","types":"./locales/index.d.cts","import":"./locales/index.js","require":"./locales/index.cjs"},"./v3":{"@zod/source":"./src/v3/index.ts","types":"./v3/index.d.cts","import":"./v3/index.js","require":"./v3/index.cjs"},"./v4":{"@zod/source":"./src/v4/index.ts","types":"./v4/index.d.cts","import":"./v4/index.js","require":"./v4/index.cjs"},"./v4-mini":{"@zod/source":"./src/v4-mini/index.ts","types":"./v4-mini/index.d.cts","import":"./v4-mini/index.js","require":"./v4-mini/index.cjs"},"./v4/mini":{"@zod/source":"./src/v4/mini/index.ts","types":"./v4/mini/index.d.cts","import":"./v4/mini/index.js","require":"./v4/mini/index.cjs"},"./v4/core":{"@zod/source":"./src/v4/core/index.ts","types":"./v4/core/index.d.cts","import":"./v4/core/index.js","require":"./v4/core/index.cjs"},"./v4/locales":{"@zod/source":"./src/v4/locales/index.ts","types":"./v4/locales/index.d.cts","import":"./v4/locales/index.js","require":"./v4/locales/index.cjs"},"./v4/locales/*":{"@zod/source":"./src/v4/locales/*","types":"./v4/locales/*","import":"./v4/locales/*","require":"./v4/locales/*"}},"repository":{"type":"git","url":"git+https://github.com/colinhacks/zod.git"},"bugs":{"url":"https://github.com/colinhacks/zod/issues"},"support":{"backing":{"npm-funding":true}},"scripts":{"clean":"git clean -xdf . -e node_modules","build":"zshy --project tsconfig.build.json","postbuild":"tsx ../../scripts/write-stub-package-jsons.ts && pnpm biome check --write .","test:watch":"pnpm vitest","test":"pnpm vitest run","prepublishOnly":"tsx ../../scripts/check-versions.ts"},"_lastModified":"2026-06-15T15:31:13.421Z"}
+{"name":"zod","version":"4.3.5","type":"module","license":"MIT","author":"Colin McDonnell <zod@colinhacks.com>","description":"TypeScript-first schema declaration and validation library with static type inference","homepage":"https://zod.dev","llms":"https://zod.dev/llms.txt","llmsFull":"https://zod.dev/llms-full.txt","mcpServer":"https://mcp.inkeep.com/zod/mcp","funding":"https://github.com/sponsors/colinhacks","sideEffects":false,"files":["src","**/*.js","**/*.mjs","**/*.cjs","**/*.d.ts","**/*.d.mts","**/*.d.cts","**/package.json"],"keywords":["typescript","schema","validation","type","inference"],"main":"./index.cjs","types":"./index.d.cts","module":"./index.js","zshy":{"exports":{"./package.json":"./package.json",".":"./src/index.ts","./mini":"./src/mini/index.ts","./locales":"./src/locales/index.ts","./v3":"./src/v3/index.ts","./v4":"./src/v4/index.ts","./v4-mini":"./src/v4-mini/index.ts","./v4/mini":"./src/v4/mini/index.ts","./v4/core":"./src/v4/core/index.ts","./v4/locales":"./src/v4/locales/index.ts","./v4/locales/*":"./src/v4/locales/*"},"conditions":{"@zod/source":"src"}},"exports":{"./package.json":"./package.json",".":{"@zod/source":"./src/index.ts","types":"./index.d.cts","import":"./index.js","require":"./index.cjs"},"./mini":{"@zod/source":"./src/mini/index.ts","types":"./mini/index.d.cts","import":"./mini/index.js","require":"./mini/index.cjs"},"./locales":{"@zod/source":"./src/locales/index.ts","types":"./locales/index.d.cts","import":"./locales/index.js","require":"./locales/index.cjs"},"./v3":{"@zod/source":"./src/v3/index.ts","types":"./v3/index.d.cts","import":"./v3/index.js","require":"./v3/index.cjs"},"./v4":{"@zod/source":"./src/v4/index.ts","types":"./v4/index.d.cts","import":"./v4/index.js","require":"./v4/index.cjs"},"./v4-mini":{"@zod/source":"./src/v4-mini/index.ts","types":"./v4-mini/index.d.cts","import":"./v4-mini/index.js","require":"./v4-mini/index.cjs"},"./v4/mini":{"@zod/source":"./src/v4/mini/index.ts","types":"./v4/mini/index.d.cts","import":"./v4/mini/index.js","require":"./v4/mini/index.cjs"},"./v4/core":{"@zod/source":"./src/v4/core/index.ts","types":"./v4/core/index.d.cts","import":"./v4/core/index.js","require":"./v4/core/index.cjs"},"./v4/locales":{"@zod/source":"./src/v4/locales/index.ts","types":"./v4/locales/index.d.cts","import":"./v4/locales/index.js","require":"./v4/locales/index.cjs"},"./v4/locales/*":{"@zod/source":"./src/v4/locales/*","types":"./v4/locales/*","import":"./v4/locales/*","require":"./v4/locales/*"}},"repository":{"type":"git","url":"git+https://github.com/colinhacks/zod.git"},"bugs":{"url":"https://github.com/colinhacks/zod/issues"},"support":{"backing":{"npm-funding":true}},"scripts":{"clean":"git clean -xdf . -e node_modules","build":"zshy --project tsconfig.build.json","postbuild":"tsx ../../scripts/write-stub-package-jsons.ts && pnpm biome check --write .","test:watch":"pnpm vitest","test":"pnpm vitest run","prepublishOnly":"tsx ../../scripts/check-versions.ts"},"_lastModified":"2026-06-22T05:35:39.331Z"}

package/dist/server/ai-employees/ai-employee.d.ts

@@ -9,8 +9,6 @@
 import { Model, Transaction } from '@nocobase/database';
 import { LLMProvider } from '../llm-providers/provider';
 import { AIChatContext, AIChatConversation, AIMessageInput, AIToolCall, UserDecision } from '../types';
-import { DocumentSegmentedWithScore } from '../features';
-import { KnowledgeBaseGroup } from '../types';
 import { ToolsEntry } from '@nocobase/ai';
 import { AIToolMessage } from '../types/ai-message.type';
 import { Context } from '@nocobase/actions';
@@ -55,7 +53,7 @@ export declare class AIEmployee {
     private areSkillsEnabled;
     private areToolsEnabled;
     getFormatMessages(userMessages: AIMessageInput[]): Promise<{
-        role: "user" | "assistant" | "tool" | "system";
+        role: "system" | "user" | "assistant" | "tool";
         content: any;
         tool_call_id?: string;
         tool_calls?: AIToolCall[];
@@ -116,13 +114,6 @@ export declare class AIEmployee {
     private handleInterruptedToolCalls;
     getEmployeeDataSourceContext(): string;
     getSystemPrompt(userMessages: AIMessageInput[]): Promise<any>;
-    retrieveKnowledgeBase(userMessage: AIMessageInput): Promise<DocumentSegmentedWithScore[]>;
-    isEnabledKnowledgeBase(): boolean;
-    getAIEmployeeKnowledgeBaseConfig(): {
-        topK: number;
-        score: string;
-    };
-    getKnowledgeBaseGroup(): Promise<KnowledgeBaseGroup[]>;
     initToolCall(transaction: Transaction, messageId: string, toolCalls: {
         id: string;
         name: string;

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

@@ -45,7 +45,6 @@ var import_utils = require("../utils");
 var import_prompts = require("./prompts");
 var import_lodash = __toESM(require("lodash"));
 var import_ai_chat_conversation = require("../manager/ai-chat-conversation");
-var import_ai_feature_manager = require("../manager/ai-feature-manager");
 var import_middleware = require("./middleware");
 var import_ai = require("@nocobase/ai");
 var import_checkpoints = require("./checkpoints");
@@ -756,89 +755,6 @@ If information is missing, clearly state it in the summary.</Important>`;
       return systemPrompt;
     }
   }
-  async retrieveKnowledgeBase(userMessage) {
-    const vectorStoreProvider = this.plugin.features.vectorStoreProvider;
-    let queryResult = [];
-    const queryString = userMessage.content.content;
-    if (!queryString || import_lodash.default.isEmpty(queryString)) {
-      return queryResult;
-    }
-    const { topK, score } = this.getAIEmployeeKnowledgeBaseConfig();
-    const knowledgeBaseGroup = await this.getKnowledgeBaseGroup();
-    for (const entry of knowledgeBaseGroup) {
-      const { vectorStoreConfig, knowledgeBaseType, knowledgeBaseList } = entry;
-      if (!knowledgeBaseList || import_lodash.default.isEmpty(knowledgeBaseList)) {
-        continue;
-      }
-      if (knowledgeBaseType === "LOCAL") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const knowledgeBaseOuterIds = knowledgeBaseList.map((x) => x.knowledgeBaseOuterId);
-        const result = await vectorStoreService.search(queryString, {
-          topK,
-          score,
-          filter: {
-            knowledgeBaseOuterId: { in: knowledgeBaseOuterIds }
-          }
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "READONLY") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const result = await vectorStoreService.search(queryString, {
-          topK,
-          score
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "EXTERNAL") {
-        for (const knowledgeBase of knowledgeBaseList) {
-          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-            vectorStoreConfig.vectorStoreProvider,
-            knowledgeBase.vectorStoreProps
-          );
-          const result = await vectorStoreService.search(queryString, {
-            topK,
-            score
-          });
-          queryResult = [...queryResult, ...result];
-        }
-      }
-    }
-    return queryResult;
-  }
-  isEnabledKnowledgeBase() {
-    const featureEnabled = this.plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
-    const knowledgeBaseEnabled = this.employee.enableKnowledgeBase;
-    return featureEnabled && knowledgeBaseEnabled;
-  }
-  getAIEmployeeKnowledgeBaseConfig() {
-    const { topK, score } = this.employee.knowledgeBase ?? {};
-    return {
-      topK,
-      score
-    };
-  }
-  async getKnowledgeBaseGroup() {
-    const { knowledgeBaseKeys } = this.employee.knowledgeBase ?? {};
-    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
-      return [];
-    }
-    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseKeys);
-  }
   // === Tool calls ===
   async initToolCall(transaction, messageId, toolCalls) {
     const nowTime = /* @__PURE__ */ new Date();

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

@@ -7,7 +7,6 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { AIEmployee } from '../../collections/ai-employees';
-import { DocumentSegmentedWithScore } from '../features';
 import type PluginAIServer from '../plugin';
 export type KnowledgeBaseRetrieveOptions = {
     username?: string;
@@ -18,11 +17,8 @@ export declare class KnowledgeBaseManager {
     private readonly plugin;
     constructor(plugin: PluginAIServer);
     retrievePrompt({ username, employee, query }: KnowledgeBaseRetrieveOptions): Promise<string>;
-    retrieve({ username, employee, query }: KnowledgeBaseRetrieveOptions): Promise<DocumentSegmentedWithScore[]>;
     isEnabledKnowledgeBase(username: string): Promise<boolean>;
     isEnabledKnowledgeBase(employee: AIEmployee): Promise<boolean>;
-    private getAIEmployeeKnowledgeBaseConfig;
-    private getKnowledgeBaseGroup;
     private getEmployee;
     private get aiEmployeeRepo();
 }

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

@@ -42,115 +42,51 @@ module.exports = __toCommonJS(ai_knowledge_base_exports);
 var import_prompts = require("@langchain/core/prompts");
 var import_ai_feature_manager = require("../manager/ai-feature-manager");
 var import_lodash = __toESM(require("lodash"));
+const normalizeMatchedQuestions = (value) => {
+  if (!Array.isArray(value)) {
+    return [];
+  }
+  return value.filter((item) => typeof item === "string" && item.trim() !== "");
+};
+const buildKnowledgeBaseContent = (content, metadata) => {
+  const matchedQuestions = normalizeMatchedQuestions(metadata == null ? void 0 : metadata.matchedQuestions);
+  if (!matchedQuestions.length) {
+    return content;
+  }
+  return `Related questions:
+${matchedQuestions.join("\n")}
+
+${content}`;
+};
 class KnowledgeBaseManager {
   constructor(plugin) {
     this.plugin = plugin;
   }
   async retrievePrompt({ username, employee, query }) {
-    employee = employee ?? await this.getEmployee(username);
+    employee = employee ?? (username ? await this.getEmployee(username) : null);
     if (!employee) {
       return "Specified knowledge base not existed";
     }
-    const promptTemplate = import_prompts.ChatPromptTemplate.fromTemplate(employee.knowledgeBasePrompt);
-    const docs = await this.retrieve({
-      employee,
-      query
-    });
+    const { knowledgeBaseKeys, topK, score } = employee.knowledgeBase ?? {};
+    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
+      return "Specified knowledge base not existed";
+    }
+    const promptTemplate = import_prompts.ChatPromptTemplate.fromTemplate(employee.knowledgeBasePrompt ?? "{knowledgeBaseData}");
+    const docs = await this.plugin.features.knowledgeBase.search({ knowledgeBaseKeys, query, topK, score });
     if (!(docs == null ? void 0 : docs.length)) {
       return "No document match in knowledge base";
     }
-    const knowledgeBaseData = docs.map((x) => x.content).join("\n");
-    return import_lodash.default.isEmpty(knowledgeBaseData) ? void 0 : await promptTemplate.format({
+    const knowledgeBaseData = docs.map((doc) => buildKnowledgeBaseContent(doc.content, doc.metadata)).join("\n");
+    return import_lodash.default.isEmpty(knowledgeBaseData) ? "No document match in knowledge base" : await promptTemplate.format({
       knowledgeBaseData
     });
   }
-  async retrieve({ username, employee, query }) {
-    const vectorStoreProvider = this.plugin.features.vectorStoreProvider;
-    let queryResult = [];
-    if (!query || import_lodash.default.isEmpty(query)) {
-      return queryResult;
-    }
-    employee = employee ?? await this.getEmployee(username);
-    if (!employee) {
-      return queryResult;
-    }
-    const { topK, score } = this.getAIEmployeeKnowledgeBaseConfig(employee);
-    const knowledgeBaseGroup = await this.getKnowledgeBaseGroup(employee);
-    for (const entry of knowledgeBaseGroup) {
-      const { vectorStoreConfig, knowledgeBaseType, knowledgeBaseList } = entry;
-      if (!knowledgeBaseList || import_lodash.default.isEmpty(knowledgeBaseList)) {
-        continue;
-      }
-      if (knowledgeBaseType === "LOCAL") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const knowledgeBaseOuterIds = knowledgeBaseList.map((x) => x.knowledgeBaseOuterId);
-        const result = await vectorStoreService.search(query, {
-          topK,
-          score,
-          filter: {
-            knowledgeBaseOuterId: { in: knowledgeBaseOuterIds }
-          }
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "READONLY") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const result = await vectorStoreService.search(query, {
-          topK,
-          score
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "EXTERNAL") {
-        for (const knowledgeBase of knowledgeBaseList) {
-          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-            vectorStoreConfig.vectorStoreProvider,
-            this.plugin.app.environment.renderJsonTemplate(knowledgeBase.vectorStoreProps ?? [])
-          );
-          const result = await vectorStoreService.search(query, {
-            topK,
-            score
-          });
-          queryResult = [...queryResult, ...result];
-        }
-      }
-    }
-    return queryResult;
-  }
   async isEnabledKnowledgeBase(usernameOrEmployee) {
     const featureEnabled = this.plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
     const employee = typeof usernameOrEmployee === "string" ? await this.getEmployee(usernameOrEmployee) : usernameOrEmployee;
     const knowledgeBaseEnabled = employee == null ? void 0 : employee.enableKnowledgeBase;
     return featureEnabled && knowledgeBaseEnabled;
   }
-  getAIEmployeeKnowledgeBaseConfig(employee) {
-    const { topK, score } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
-    return {
-      topK,
-      score
-    };
-  }
-  async getKnowledgeBaseGroup(employee) {
-    const { knowledgeBaseKeys } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
-    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
-      return [];
-    }
-    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseKeys);
-  }
   async getEmployee(username) {
     return await this.aiEmployeeRepo.findOne({ filter: { username } });
   }