@liujitcn/kratos-admin-system 0.0.65 → 0.0.66

package/dist/package/src/rpc/base/v1/ai_knowledge.ts

@@ -0,0 +1,33 @@
+// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
+// versions:
+//   protoc-gen-ts_proto  v2.11.8
+//   protoc               unknown
+// source: base/v1/ai_knowledge.proto
+
+/* eslint-disable */
+
+/** AI知识库选项查询请求。 */
+export interface ListAiKnowledgeOptionsRequest {
+}
+
+/** AI知识库选项查询响应。 */
+export interface ListAiKnowledgeOptionsResponse {
+  /** 知识库选项列表 */
+  options: AiKnowledgeOption[];
+}
+
+/** AI知识库选项。 */
+export interface AiKnowledgeOption {
+  /** 知识库ID */
+  id: number;
+  /** 知识库名称 */
+  name: string;
+  /** 文档数量 */
+  doc_count: number;
+}
+
+/** AI 知识库选项查询服务。 */
+export interface AiKnowledgeService {
+  /** 查询可用的AI知识库选项。 */
+  ListAiKnowledgeOptions(request: ListAiKnowledgeOptionsRequest): Promise<ListAiKnowledgeOptionsResponse>;
+}

package/dist/package/src/rpc/base/v1/ai_message.ts

@@ -53,6 +53,8 @@ export interface SendAiMessageRequest {
   provider_id: number;
   /** 所选模型名称 */
   model_name: string;
+  /** 所选知识库ID列表 */
+  knowledge_base_ids: number[];
 }
 
 /** AI 助手失败消息重试请求 */

package/dist/package/src/rpc/system/admin/v1/ai_provider.ts

@@ -8,6 +8,24 @@
 import type { Status } from "../../../common/v1/enum";
 import type { Empty } from "../../../google/protobuf/empty";
 
+/** AI模型分类枚举。 */
+export enum AiModelCategory {
+  /** AI_MODEL_CATEGORY_UNSPECIFIED - 未指定AI模型分类。 */
+  AI_MODEL_CATEGORY_UNSPECIFIED = 0,
+  /** AI_MODEL_CATEGORY_CHAT - 对话模型。 */
+  AI_MODEL_CATEGORY_CHAT = 1,
+  /** AI_MODEL_CATEGORY_EMBEDDING - 向量模型。 */
+  AI_MODEL_CATEGORY_EMBEDDING = 2,
+  /** AI_MODEL_CATEGORY_RERANK - 重排序模型。 */
+  AI_MODEL_CATEGORY_RERANK = 3,
+  /** AI_MODEL_CATEGORY_IMAGE - 图片模型。 */
+  AI_MODEL_CATEGORY_IMAGE = 4,
+  /** AI_MODEL_CATEGORY_VIDEO - 视频模型。 */
+  AI_MODEL_CATEGORY_VIDEO = 5,
+  /** AI_MODEL_CATEGORY_AUDIO - 音频模型。 */
+  AI_MODEL_CATEGORY_AUDIO = 6,
+}
+
 /** AI供应商分页查询条件。 */
 export interface PageAiProviderRequest {
   /** 供应商名称 */
@@ -48,8 +66,6 @@ export interface AiProviderForm {
   api_key: string;
   /** 是否已保存模型密钥 */
   api_key_configured: boolean;
-  /** 多模型JSON数组 */
-  models_json: string;
   /** 排序值 */
   sort: number;
   /** Provider稳定标识 */
@@ -58,6 +74,28 @@ export interface AiProviderForm {
   config:
     | { [key: string]: any }
     | undefined;
+  /** 供应商下的AI模型列表 */
+  models: AiProviderModelForm[];
+  /** 状态 */
+  status: Status;
+}
+
+/** AI供应商模型维护表单；provider_id由所属供应商表单隐式给定。 */
+export interface AiProviderModelForm {
+  /** 模型ID；新增时为0 */
+  id: number;
+  /** 模型名称，供应商API标识 */
+  model_name: string;
+  /** 显示名称 */
+  display_name: string;
+  /** 模型分类：枚举【AiModelCategory】 */
+  category: AiModelCategory;
+  /** 分类个性化配置JSON对象 */
+  config:
+    | { [key: string]: any }
+    | undefined;
+  /** 排序值 */
+  sort: number;
   /** 状态 */
   status: Status;
 }
@@ -74,7 +112,7 @@ export interface UpdateAiProviderRequest {
   ai_provider: AiProviderForm | undefined;
 }
 
-/** 测试AI供应商全部模型请求。 */
+/** 测试AI供应商模型请求。 */
 export interface TestAiProviderModelsRequest {
   /** 待测试的AI供应商草稿 */
   ai_provider: AiProviderForm | undefined;
@@ -98,6 +136,26 @@ export interface TestAiProviderModelsResponse {
   results: AiProviderModelTestResult[];
 }
 
+/** 查询AI供应商选项请求。 */
+export interface ListAiProviderOptionsRequest {
+}
+
+/** 查询AI供应商选项响应。 */
+export interface ListAiProviderOptionsResponse {
+  /** 已启用的AI供应商列表 */
+  options: AiProviderOption[];
+}
+
+/** AI供应商选项。 */
+export interface AiProviderOption {
+  /** 供应商ID */
+  id: number;
+  /** 供应商名称 */
+  name: string;
+  /** Provider稳定标识 */
+  provider: string;
+}
+
 /** 删除AI供应商请求。 */
 export interface DeleteAiProviderRequest {
   /** 供应商ID */
@@ -122,12 +180,12 @@ export interface AiProvider {
   base_url: string;
   /** 是否已保存模型密钥 */
   api_key_configured: boolean;
-  /** 模型名称列表 */
-  model_names: string[];
   /** 排序值 */
   sort: number;
   /** Provider稳定标识 */
   provider: string;
+  /** 供应商下的模型名称列表 */
+  model_names: string[];
   /** 状态 */
   status: Status;
   /** 创建时间 */
@@ -140,6 +198,11 @@ export interface AiProvider {
 export interface AiProviderService {
   /** 分页查询AI供应商。 */
   PageAiProvider(request: PageAiProviderRequest): Promise<PageAiProviderResponse>;
+  /**
+   * 查询已启用的AI供应商选项。
+   * 字面路径必须先于 /{id} 声明：gorilla mux 按注册顺序匹配，否则会被 {id} 路由抢占。
+   */
+  ListAiProviderOptions(request: ListAiProviderOptionsRequest): Promise<ListAiProviderOptionsResponse>;
   /** 查询AI供应商表单。 */
   GetAiProvider(request: GetAiProviderRequest): Promise<AiProviderForm>;
   /** 创建AI供应商。 */

package/dist/package/src/rpc/system/admin/v1/ai_query.ts

@@ -0,0 +1,89 @@
+// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
+// versions:
+//   protoc-gen-ts_proto  v2.11.8
+//   protoc               unknown
+// source: system/admin/v1/ai_query.proto
+
+/* eslint-disable */
+
+/** 智能问数状态枚举。 */
+export enum AiQueryStatus {
+  /** AI_QUERY_STATUS_UNSPECIFIED - 未指定智能问数状态。 */
+  AI_QUERY_STATUS_UNSPECIFIED = 0,
+  /** AI_QUERY_STATUS_SUCCESS - 成功。 */
+  AI_QUERY_STATUS_SUCCESS = 1,
+  /** AI_QUERY_STATUS_FAILED - 失败。 */
+  AI_QUERY_STATUS_FAILED = 2,
+}
+
+/** 智能问数分页查询条件。 */
+export interface PageAiQueryRequest {
+  /** 问题内容 */
+  question: string;
+  /** 问数状态：枚举【AiQueryStatus】 */
+  status?:
+    | AiQueryStatus
+    | undefined;
+  /** 当前页码 */
+  page_num: number;
+  /** 每页行数 */
+  page_size: number;
+}
+
+/** 智能问数分页响应。 */
+export interface PageAiQueryResponse {
+  /** 智能问数记录列表 */
+  queries: AiQueryRecord[];
+  /** 总数 */
+  total: number;
+}
+
+/** 查询智能问数记录请求。 */
+export interface GetAiQueryRequest {
+  /** 记录ID */
+  id: number;
+}
+
+/** 智能问数结果快照行。 */
+export interface AiQueryResultRow {
+  /** 行数据 */
+  values: string[];
+}
+
+/** 智能问数记录。 */
+export interface AiQueryRecord {
+  /** 记录ID */
+  id: number;
+  /** 租户ID */
+  tenant_id: number;
+  /** 用户ID */
+  user_id: number;
+  /** 用户账号 */
+  user_name: string;
+  /** 自然语言问题 */
+  question: string;
+  /** 生成的只读SQL */
+  sql: string;
+  /** 问数状态：枚举【AiQueryStatus】 */
+  status: AiQueryStatus;
+  /** 结果行数 */
+  row_count: number;
+  /** SQL执行耗时毫秒 */
+  elapsed_ms: number;
+  /** 失败原因 */
+  error_msg: string;
+  /** 结果快照列名，仅详情返回 */
+  result_columns: string[];
+  /** 结果快照数据行，仅详情返回 */
+  result_rows: AiQueryResultRow[];
+  /** 创建时间 */
+  created_at: string;
+}
+
+/** 智能问数记录管理服务。 */
+export interface AiQueryService {
+  /** 分页查询智能问数记录。 */
+  PageAiQuery(request: PageAiQueryRequest): Promise<PageAiQueryResponse>;
+  /** 查询智能问数记录详情。 */
+  GetAiQuery(request: GetAiQueryRequest): Promise<AiQueryRecord>;
+}

package/dist/package/src/rpc/system/admin/v1/ai_session.ts

@@ -0,0 +1,131 @@
+// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
+// versions:
+//   protoc-gen-ts_proto  v2.11.8
+//   protoc               unknown
+// source: system/admin/v1/ai_session.proto
+
+/* eslint-disable */
+import type { AiMessageStatus } from "../../../base/v1/ai_session";
+import type { Terminal } from "../../../base/v1/ai_tool";
+
+/** AI会话分页查询条件。 */
+export interface PageAiSessionRequest {
+  /** 会话标题 */
+  title: string;
+  /** 终端类型：枚举【Terminal】 */
+  terminal?:
+    | Terminal
+    | undefined;
+  /** 当前页码 */
+  page_num: number;
+  /** 每页行数 */
+  page_size: number;
+}
+
+/** AI会话分页响应。 */
+export interface PageAiSessionResponse {
+  /** 会话列表 */
+  sessions: AiSessionRecord[];
+  /** 总数 */
+  total: number;
+}
+
+/** AI会话分页记录。 */
+export interface AiSessionRecord {
+  /** 会话ID */
+  id: number;
+  /** 所属用户ID */
+  user_id: number;
+  /** 用户账号 */
+  user_name: string;
+  /** 用户昵称 */
+  nick_name: string;
+  /** 会话标题 */
+  title: string;
+  /** 会话摘要 */
+  summary: string;
+  /** 终端类型：枚举【Terminal】 */
+  terminal: Terminal;
+  /** 消息轮次数 */
+  message_count: number;
+  /** 累计输入Token数 */
+  input_tokens: number;
+  /** 累计输出Token数 */
+  output_tokens: number;
+  /** 累计缓存Token数 */
+  cache_tokens: number;
+  /** 累计总Token数 */
+  total_tokens: number;
+  /** 创建时间 */
+  created_at: string;
+  /** 更新时间 */
+  updated_at: string;
+}
+
+/** 分页查询AI会话消息请求。 */
+export interface PageAiSessionMessageRequest {
+  /** 会话ID */
+  id: number;
+  /** 当前页码 */
+  page_num: number;
+  /** 每页行数 */
+  page_size: number;
+}
+
+/** 分页查询AI会话消息响应。 */
+export interface PageAiSessionMessageResponse {
+  /** 消息列表 */
+  messages: AiSessionMessageRecord[];
+  /** 总数 */
+  total: number;
+}
+
+/** AI会话消息记录。 */
+export interface AiSessionMessageRecord {
+  /** 消息ID */
+  id: number;
+  /** 用户输入内容 */
+  input_content: string;
+  /** AI回复内容 */
+  output_content: string;
+  /** 回复使用的模型名称 */
+  model: string;
+  /** 消息生成状态：枚举【AiMessageStatus】 */
+  status: AiMessageStatus;
+  /** 输入Token数 */
+  input_tokens: number;
+  /** 输出Token数 */
+  output_tokens: number;
+  /** 缓存Token数 */
+  cache_tokens: number;
+  /** 总Token数 */
+  total_tokens: number;
+  /** 首Token耗时毫秒 */
+  first_token_ms: number;
+  /** 总耗时毫秒 */
+  duration_ms: number;
+  /** 附件列表 */
+  attachments: AiSessionMessageAttachment[];
+  /** 创建时间 */
+  created_at: string;
+}
+
+/** AI会话消息附件。 */
+export interface AiSessionMessageAttachment {
+  /** 附件名称 */
+  name: string;
+  /** 附件大小 */
+  size: number;
+  /** 附件地址 */
+  url: string;
+  /** 附件 MIME 类型 */
+  mime_type: string;
+}
+
+/** AI会话管理服务。 */
+export interface AiSessionService {
+  /** 分页查询AI会话。 */
+  PageAiSession(request: PageAiSessionRequest): Promise<PageAiSessionResponse>;
+  /** 分页查询AI会话消息。 */
+  PageAiSessionMessage(request: PageAiSessionMessageRequest): Promise<PageAiSessionMessageResponse>;
+}

package/dist/package/src/views/ai/chat/components/ChatPanel.vue

@@ -53,8 +53,11 @@
             :model-providers="modelProviders"
             :provider-id="providerId"
             :model-name="modelName"
+            :knowledge-options="knowledgeOptions"
+            :knowledge-ids="knowledgeIds"
             @submit="handleSubmit"
             @model-change="handleModelChange"
+            @knowledge-change="handleKnowledgeChange"
           />
         </div>
       </div>
@@ -336,8 +339,11 @@
           :model-providers="modelProviders"
           :provider-id="providerId"
           :model-name="modelName"
+          :knowledge-options="knowledgeOptions"
+          :knowledge-ids="knowledgeIds"
           @submit="handleSubmit"
           @model-change="handleModelChange"
+          @knowledge-change="handleKnowledgeChange"
         />
       </div>
     </template>
@@ -356,6 +362,7 @@ import { formatDateTime } from "@liujitcn/kratos-admin-core/format";
 import type { AiAction } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_message";
 import { type AiAttachment, type AiSession } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
 import type { AiProviderModelOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_provider";
+import type { AiKnowledgeOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_knowledge";
 import type { AiShortcut, AiToolCall } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_tool";
 import { AiMessageStatus } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
 import { resolvePersistedMessage } from "../../../../utils/persisted-message";
@@ -501,6 +508,10 @@ const props = defineProps<{
   providerId: number;
   /** 当前选择的模型名称。 */
   modelName: string;
+  /** 当前可供选择的知识库。 */
+  knowledgeOptions: AiKnowledgeOption[];
+  /** 当前选择的知识库编号列表。 */
+  knowledgeIds: number[];
 }>();
 
 const emit = defineEmits<{
@@ -508,6 +519,8 @@ const emit = defineEmits<{
   submit: [payload: SubmitPayload];
   /** 更新当前选择的供应商和模型。 */
   modelChange: [providerId: number, modelName: string];
+  /** 更新当前选择的知识库编号列表。 */
+  knowledgeChange: [knowledgeIds: number[]];
   /** 触发消息级操作。 */
   messageAction: [payload: { action: ChatMessageAction; item: ChatMessageItem }];
   /** 提交当前用户消息的文本编辑。 */
@@ -648,6 +661,11 @@ function handleModelChange(providerId: number, modelName: string) {
   emit("modelChange", providerId, modelName);
 }
 
+/** 更新当前选择的知识库编号列表。 */
+function handleKnowledgeChange(knowledgeIds: number[]) {
+  emit("knowledgeChange", knowledgeIds);
+}
+
 /** 点击快捷入口时提交对应入口动作。 */
 function handleShortcutClick(shortcut: AiShortcut) {
   emit("submit", {

package/dist/package/src/views/ai/chat/components/XSender.vue

@@ -126,6 +126,72 @@
               {{ t("system.ai.model.empty") }}
             </div>
           </el-popover>
+          <el-popover
+            ref="knowledgePopoverRef"
+            placement="top-start"
+            :width="324"
+            trigger="click"
+            :show-arrow="false"
+            :persistent="false"
+            popper-class="agent-model-popper"
+          >
+            <template #reference>
+              <button
+                class="agent-model-selector"
+                :class="{ 'is-disabled': sending || !knowledgeOptions.length, 'is-selected': selectedKnowledgeCount > 0 }"
+                type="button"
+                :disabled="sending || !knowledgeOptions.length"
+                :aria-label="t('system.ai.knowledge.title')"
+              >
+                <el-icon class="agent-model-selector__spark"><Collection /></el-icon>
+                <span class="agent-model-selector__label">{{
+                  selectedKnowledgeCount > 0
+                    ? t("system.ai.knowledge.selected_count", { count: selectedKnowledgeCount })
+                    : t("system.ai.knowledge.title")
+                }}</span>
+                <el-icon class="agent-model-selector__arrow"><ArrowDown /></el-icon>
+              </button>
+            </template>
+            <div v-if="knowledgeOptions.length" class="agent-model-panel">
+              <div class="agent-model-panel__head">
+                <span class="agent-model-panel__title">{{ t("system.ai.knowledge.title") }}</span>
+                <span class="agent-model-panel__meta">{{
+                  t("system.ai.knowledge.select_hint")
+                }}</span>
+              </div>
+              <div class="agent-model-panel__list">
+                <button
+                  v-for="option in knowledgeOptions"
+                  :key="option.id"
+                  class="agent-model-row"
+                  :class="{ 'is-selected': isKnowledgeSelected(option.id) }"
+                  type="button"
+                  @click="handleKnowledgeToggle(option.id)"
+                >
+                  <span class="agent-model-row__name">{{ option.name }}</span>
+                  <span v-if="option.doc_count" class="agent-model-row__raw">{{
+                    t("system.ai.knowledge.doc_count", { count: option.doc_count })
+                  }}</span>
+                  <el-icon v-if="isKnowledgeSelected(option.id)" class="agent-model-row__check">
+                    <Check />
+                  </el-icon>
+                </button>
+              </div>
+              <button
+                v-if="aiKnowledgeRoute"
+                class="agent-model-panel__manage"
+                type="button"
+                @click="openKnowledgeManage"
+              >
+                <el-icon><Setting /></el-icon>
+                <span>{{ t("system.ai.knowledge.manage") }}</span>
+                <el-icon class="agent-model-panel__manage-arrow"><ArrowRight /></el-icon>
+              </button>
+            </div>
+            <div v-else class="agent-model-panel agent-model-panel--empty">
+              {{ t("system.ai.knowledge.empty") }}
+            </div>
+          </el-popover>
           <el-tooltip
             :content="recording ? t('system.ai.chat.action.stop_voice_input') : t('system.ai.chat.action.voice_input')"
             placement="top"
@@ -181,6 +247,7 @@ import {
   ArrowDown,
   ArrowRight,
   Check,
+  Collection,
   Loading,
   MagicStick,
   Microphone,
@@ -195,6 +262,7 @@ import { useAuthStore } from "@liujitcn/kratos-admin-core/stores/runtime";
 import { defFileService } from "@liujitcn/kratos-admin-core/api/base/v1/file";
 import type { AiAttachment } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
 import type { AiProviderModelOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_provider";
+import type { AiKnowledgeOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_knowledge";
 import type { SubmitPayload } from "../types";
 import { buildAIAttachmentFileCard } from "../attachment";
 import { fallbackModelDisplayName } from "../modelDisplay";
@@ -210,6 +278,10 @@ const props = defineProps<{
   providerId: number;
   /** 当前选择的模型名称。 */
   modelName: string;
+  /** 当前可供选择的知识库。 */
+  knowledgeOptions: AiKnowledgeOption[];
+  /** 当前选择的知识库编号列表。 */
+  knowledgeIds: number[];
 }>();
 
 const emit = defineEmits<{
@@ -217,6 +289,8 @@ const emit = defineEmits<{
   submit: [payload: SubmitPayload];
   /** 更新当前选择的供应商和模型。 */
   "model-change": [providerId: number, modelName: string];
+  /** 更新当前选择的知识库编号列表。 */
+  "knowledge-change": [knowledgeIds: number[]];
 }>();
 
 const senderRef = ref<InstanceType<typeof BaseXSender>>();
@@ -249,6 +323,9 @@ const selectedModelDisplayName = computed(() => {
 const aiProviderRoute = computed(() =>
   authStore.flatMenuListGet.find(item => item.name === "AiProvider" && item.path)
 );
+const aiKnowledgeRoute = computed(() =>
+  authStore.flatMenuListGet.find(item => item.name === "AiKnowledge" && item.path)
+);
 const uploading = ref(false);
 const maxAttachmentCount = 6;
 const maxAttachmentSizeMB = 20;
@@ -323,6 +400,30 @@ async function openModelManage() {
   await navigateTo(router, aiProviderRoute.value.path);
 }
 
+const knowledgePopoverRef = ref<InstanceType<typeof ElPopover>>();
+
+const selectedKnowledgeCount = computed(() => props.knowledgeIds.length);
+
+/** 判断知识库是否在当前选择中。 */
+function isKnowledgeSelected(knowledgeId: number) {
+  return props.knowledgeIds.includes(knowledgeId);
+}
+
+/** 切换知识库选择状态并同步父组件。 */
+function handleKnowledgeToggle(knowledgeId: number) {
+  const current = new Set(props.knowledgeIds);
+  if (current.has(knowledgeId)) current.delete(knowledgeId);
+  else current.add(knowledgeId);
+  emit("knowledge-change", [...current]);
+}
+
+/** 跳转到 AI 知识库管理页。 */
+async function openKnowledgeManage() {
+  if (!aiKnowledgeRoute.value?.path) return;
+  knowledgePopoverRef.value?.hide();
+  await navigateTo(router, aiKnowledgeRoute.value.path);
+}
+
 /** 读取输入内容并发送给父组件。 */
 function handleSubmit() {
   if (uploading.value) return;

package/dist/package/src/views/ai/chat/index.vue

@@ -30,8 +30,11 @@
       :model-providers="modelProviders"
       :provider-id="selectedProviderId"
       :model-name="selectedModelName"
+      :knowledge-options="knowledgeOptions"
+      :knowledge-ids="selectedKnowledgeIds"
       @submit="handleSubmit"
       @model-change="handleModelChange"
+      @knowledge-change="handleKnowledgeChange"
       @message-action="handleMessageAction"
       @message-edit="handleEditMessage"
     >
@@ -59,9 +62,11 @@ import { defAiMessageService } from "@liujitcn/kratos-admin-system/api/base/v1/a
 import { defAiSessionService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_session";
 import { defAiToolService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_tool";
 import { defAiModelService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_provider";
+import { defAiKnowledgeService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_knowledge";
 import type { AiAction } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_message";
 import type { AiSession } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
 import type { AiProviderModelOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_provider";
+import type { AiKnowledgeOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_knowledge";
 import type { AiShortcut } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_tool";
 import { AiMessageStatus } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
 import { Terminal } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_tool";
@@ -112,6 +117,8 @@ const starterShortcuts = ref<AiShortcut[]>([]);
 const modelProviders = ref<AiProviderModelOption[]>([]);
 const selectedProviderId = ref(0);
 const selectedModelName = ref("");
+const knowledgeOptions = ref<AiKnowledgeOption[]>([]);
+const selectedKnowledgeIds = ref<number[]>([]);
 const messages = ref<Record<string, ChatMessageItem[]>>({});
 const pendingDeltaMap = new Map<string, AiStreamPayload>();
 const runningStreamTaskMap = new Map<string, AiStreamTask>();
@@ -229,6 +236,7 @@ async function runAiStreamTask(sessionID: string, payload: SubmitPayload) {
         content: payload.text,
         provider_id: selectedProviderId.value,
         model_name: selectedModelName.value,
+        knowledge_base_ids: selectedKnowledgeIds.value,
         attachments: payload.attachments.map(item => ({
           id: item.id,
           name: item.name,
@@ -657,16 +665,23 @@ async function loadAiShortcuts() {
   if (loadingShortcuts.value) return;
   loadingShortcuts.value = true;
   try {
-    const [response, providerResponse] = await Promise.all([
+    const [response, providerResponse, knowledgeResponse] = await Promise.all([
       defAiToolService.ListAiShortcut({ terminal: Terminal.TERMINAL_ADMIN }),
-      defAiModelService.ListAiProviderModelOptions({})
+      defAiModelService.ListAiProviderModelOptions({}),
+      defAiKnowledgeService.ListAiKnowledgeOptions({})
     ]);
     starterShortcuts.value = normalizeStarterShortcuts(response.shortcuts);
     modelProviders.value = providerResponse.providers ?? [];
+    knowledgeOptions.value = knowledgeResponse.options ?? [];
+    // 知识库可能已被删除，剔除无效选择。
+    selectedKnowledgeIds.value = selectedKnowledgeIds.value.filter(id =>
+      knowledgeOptions.value.some(option => option.id === id)
+    );
     applyModelSelection();
   } catch {
     starterShortcuts.value = [];
     modelProviders.value = [];
+    knowledgeOptions.value = [];
     selectedProviderId.value = 0;
     selectedModelName.value = "";
   } finally {
@@ -701,6 +716,11 @@ function handleModelChange(providerId: number, modelName: string) {
   saveModelSelection({ provider_id: providerId, model_name: modelName });
 }
 
+/** 更新当前聊天检索注入的知识库选择。 */
+function handleKnowledgeChange(knowledgeIds: number[]) {
+  selectedKnowledgeIds.value = knowledgeIds;
+}
+
 /** 使指定会话仍在途的消息列表请求失效。 */
 function invalidateLoadMessages(sessionID: string) {
   loadMessagesSeq.set(sessionID, (loadMessagesSeq.get(sessionID) ?? 0) + 1);