@liujitcn/kratos-uni-app-system 0.0.65 → 0.0.66

package/dist/index.mjs

@@ -372,7 +372,12 @@ var zh_CN_default = {
   "system.ai.model.load_failed": "\u52A0\u8F7D\u4F9B\u5E94\u5546\u548C\u6A21\u578B\u5931\u8D25",
   "system.ai.model.title": "\u9009\u62E9\u6A21\u578B",
   "system.ai.model.provider_model_count": "{providers} \u4E2A\u4F9B\u5E94\u5546 \xB7 {models} \u4E2A\u6A21\u578B",
-  "system.ai.model.empty": "\u6682\u65E0\u53EF\u9009\u6A21\u578B"
+  "system.ai.model.empty": "\u6682\u65E0\u53EF\u9009\u6A21\u578B",
+  "system.ai.knowledge.title": "\u77E5\u8BC6\u5E93",
+  "system.ai.knowledge.selected_count": "\u5DF2\u9009 {count} \u4E2A\u77E5\u8BC6\u5E93",
+  "system.ai.knowledge.doc_count": "{count} \u7BC7\u6587\u6863",
+  "system.ai.knowledge.select_hint": "\u52FE\u9009\u8981\u6CE8\u5165\u4E0A\u4E0B\u6587\u7684\u77E5\u8BC6\u5E93",
+  "system.ai.knowledge.empty": "\u6682\u65E0\u77E5\u8BC6\u5E93"
 };
 
 // src/locales/en-US.json
@@ -494,7 +499,12 @@ var en_US_default = {
   "system.ai.model.load_failed": "Failed to load providers and models",
   "system.ai.model.title": "Select Model",
   "system.ai.model.provider_model_count": "{providers} providers \xB7 {models} models",
-  "system.ai.model.empty": "No models available"
+  "system.ai.model.empty": "No models available",
+  "system.ai.knowledge.title": "Knowledge Base",
+  "system.ai.knowledge.selected_count": "{count} knowledge base(s) selected",
+  "system.ai.knowledge.doc_count": "{count} doc(s)",
+  "system.ai.knowledge.select_hint": "Select knowledge bases to inject into context",
+  "system.ai.knowledge.empty": "No knowledge base available"
 };
 
 // src/locales/ja-JP.json
@@ -616,7 +626,12 @@ var ja_JP_default = {
   "system.ai.model.load_failed": "\u30D7\u30ED\u30D0\u30A4\u30C0\u30FC\u3068\u30E2\u30C7\u30EB\u3092\u8AAD\u307F\u8FBC\u3081\u307E\u305B\u3093\u3067\u3057\u305F",
   "system.ai.model.title": "\u30E2\u30C7\u30EB\u3092\u9078\u629E",
   "system.ai.model.provider_model_count": "{providers} \u30D7\u30ED\u30D0\u30A4\u30C0\u30FC \xB7 {models} \u30E2\u30C7\u30EB",
-  "system.ai.model.empty": "\u9078\u629E\u53EF\u80FD\u306A\u30E2\u30C7\u30EB\u304C\u3042\u308A\u307E\u305B\u3093"
+  "system.ai.model.empty": "\u9078\u629E\u53EF\u80FD\u306A\u30E2\u30C7\u30EB\u304C\u3042\u308A\u307E\u305B\u3093",
+  "system.ai.knowledge.title": "\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9",
+  "system.ai.knowledge.selected_count": "{count} \u4EF6\u306E\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9\u3092\u9078\u629E\u4E2D",
+  "system.ai.knowledge.doc_count": "{count} \u4EF6\u306E\u30C9\u30AD\u30E5\u30E1\u30F3\u30C8",
+  "system.ai.knowledge.select_hint": "\u30B3\u30F3\u30C6\u30AD\u30B9\u30C8\u306B\u6CE8\u5165\u3059\u308B\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9\u3092\u9078\u629E",
+  "system.ai.knowledge.empty": "\u5229\u7528\u53EF\u80FD\u306A\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9\u304C\u3042\u308A\u307E\u305B\u3093"
 };
 
 // src/locales/zh-TW.json
@@ -738,7 +753,12 @@ var zh_TW_default = {
   "system.ai.model.load_failed": "\u8F09\u5165\u4F9B\u61C9\u5546\u8207\u6A21\u578B\u5931\u6557",
   "system.ai.model.title": "\u9078\u64C7\u6A21\u578B",
   "system.ai.model.provider_model_count": "{providers} \u500B\u4F9B\u61C9\u5546 \xB7 {models} \u500B\u6A21\u578B",
-  "system.ai.model.empty": "\u66AB\u7121\u53EF\u9078\u6A21\u578B"
+  "system.ai.model.empty": "\u66AB\u7121\u53EF\u9078\u6A21\u578B",
+  "system.ai.knowledge.title": "\u77E5\u8B58\u5EAB",
+  "system.ai.knowledge.selected_count": "\u5DF2\u9078 {count} \u500B\u77E5\u8B58\u5EAB",
+  "system.ai.knowledge.doc_count": "{count} \u7BC7\u6587\u4EF6",
+  "system.ai.knowledge.select_hint": "\u52FE\u9078\u8981\u6CE8\u5165\u4E0A\u4E0B\u6587\u7684\u77E5\u8B58\u5EAB",
+  "system.ai.knowledge.empty": "\u66AB\u7121\u77E5\u8B58\u5EAB"
 };
 
 // src/locales/generated.ts

package/dist/module.mjs

@@ -152,7 +152,12 @@ var zh_CN_default = {
   "system.ai.model.load_failed": "\u52A0\u8F7D\u4F9B\u5E94\u5546\u548C\u6A21\u578B\u5931\u8D25",
   "system.ai.model.title": "\u9009\u62E9\u6A21\u578B",
   "system.ai.model.provider_model_count": "{providers} \u4E2A\u4F9B\u5E94\u5546 \xB7 {models} \u4E2A\u6A21\u578B",
-  "system.ai.model.empty": "\u6682\u65E0\u53EF\u9009\u6A21\u578B"
+  "system.ai.model.empty": "\u6682\u65E0\u53EF\u9009\u6A21\u578B",
+  "system.ai.knowledge.title": "\u77E5\u8BC6\u5E93",
+  "system.ai.knowledge.selected_count": "\u5DF2\u9009 {count} \u4E2A\u77E5\u8BC6\u5E93",
+  "system.ai.knowledge.doc_count": "{count} \u7BC7\u6587\u6863",
+  "system.ai.knowledge.select_hint": "\u52FE\u9009\u8981\u6CE8\u5165\u4E0A\u4E0B\u6587\u7684\u77E5\u8BC6\u5E93",
+  "system.ai.knowledge.empty": "\u6682\u65E0\u77E5\u8BC6\u5E93"
 };
 
 // src/locales/en-US.json
@@ -274,7 +279,12 @@ var en_US_default = {
   "system.ai.model.load_failed": "Failed to load providers and models",
   "system.ai.model.title": "Select Model",
   "system.ai.model.provider_model_count": "{providers} providers \xB7 {models} models",
-  "system.ai.model.empty": "No models available"
+  "system.ai.model.empty": "No models available",
+  "system.ai.knowledge.title": "Knowledge Base",
+  "system.ai.knowledge.selected_count": "{count} knowledge base(s) selected",
+  "system.ai.knowledge.doc_count": "{count} doc(s)",
+  "system.ai.knowledge.select_hint": "Select knowledge bases to inject into context",
+  "system.ai.knowledge.empty": "No knowledge base available"
 };
 
 // src/locales/ja-JP.json
@@ -396,7 +406,12 @@ var ja_JP_default = {
   "system.ai.model.load_failed": "\u30D7\u30ED\u30D0\u30A4\u30C0\u30FC\u3068\u30E2\u30C7\u30EB\u3092\u8AAD\u307F\u8FBC\u3081\u307E\u305B\u3093\u3067\u3057\u305F",
   "system.ai.model.title": "\u30E2\u30C7\u30EB\u3092\u9078\u629E",
   "system.ai.model.provider_model_count": "{providers} \u30D7\u30ED\u30D0\u30A4\u30C0\u30FC \xB7 {models} \u30E2\u30C7\u30EB",
-  "system.ai.model.empty": "\u9078\u629E\u53EF\u80FD\u306A\u30E2\u30C7\u30EB\u304C\u3042\u308A\u307E\u305B\u3093"
+  "system.ai.model.empty": "\u9078\u629E\u53EF\u80FD\u306A\u30E2\u30C7\u30EB\u304C\u3042\u308A\u307E\u305B\u3093",
+  "system.ai.knowledge.title": "\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9",
+  "system.ai.knowledge.selected_count": "{count} \u4EF6\u306E\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9\u3092\u9078\u629E\u4E2D",
+  "system.ai.knowledge.doc_count": "{count} \u4EF6\u306E\u30C9\u30AD\u30E5\u30E1\u30F3\u30C8",
+  "system.ai.knowledge.select_hint": "\u30B3\u30F3\u30C6\u30AD\u30B9\u30C8\u306B\u6CE8\u5165\u3059\u308B\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9\u3092\u9078\u629E",
+  "system.ai.knowledge.empty": "\u5229\u7528\u53EF\u80FD\u306A\u30CA\u30EC\u30C3\u30B8\u30D9\u30FC\u30B9\u304C\u3042\u308A\u307E\u305B\u3093"
 };
 
 // src/locales/zh-TW.json
@@ -518,7 +533,12 @@ var zh_TW_default = {
   "system.ai.model.load_failed": "\u8F09\u5165\u4F9B\u61C9\u5546\u8207\u6A21\u578B\u5931\u6557",
   "system.ai.model.title": "\u9078\u64C7\u6A21\u578B",
   "system.ai.model.provider_model_count": "{providers} \u500B\u4F9B\u61C9\u5546 \xB7 {models} \u500B\u6A21\u578B",
-  "system.ai.model.empty": "\u66AB\u7121\u53EF\u9078\u6A21\u578B"
+  "system.ai.model.empty": "\u66AB\u7121\u53EF\u9078\u6A21\u578B",
+  "system.ai.knowledge.title": "\u77E5\u8B58\u5EAB",
+  "system.ai.knowledge.selected_count": "\u5DF2\u9078 {count} \u500B\u77E5\u8B58\u5EAB",
+  "system.ai.knowledge.doc_count": "{count} \u7BC7\u6587\u4EF6",
+  "system.ai.knowledge.select_hint": "\u52FE\u9078\u8981\u6CE8\u5165\u4E0A\u4E0B\u6587\u7684\u77E5\u8B58\u5EAB",
+  "system.ai.knowledge.empty": "\u66AB\u7121\u77E5\u8B58\u5EAB"
 };
 
 // src/locales/generated.ts

package/package.json

@@ -7,7 +7,7 @@
     "type": "git",
     "url": "git@github.com:liujitcn/kratos-admin.git"
   },
-  "version": "0.0.65",
+  "version": "0.0.66",
   "type": "module",
   "files": [
     "dist",
@@ -35,7 +35,7 @@
     "@dcloudio/uni-app": "3.0.0-5010520260709002",
     "@dcloudio/uni-ui": "^1.4.28",
     "vue": "^3.4.21",
-    "@liujitcn/kratos-uni-app-core": "^0.0.65"
+    "@liujitcn/kratos-uni-app-core": "^0.0.66"
   },
   "devDependencies": {
     "esbuild": "^0.20.2",

package/src/api/base/v1/ai_knowledge.ts

@@ -0,0 +1,25 @@
+import { http } from '@liujitcn/kratos-uni-app-core/utils/http'
+import type {
+  AiKnowledgeService,
+  ListAiKnowledgeOptionsRequest,
+  ListAiKnowledgeOptionsResponse,
+} from '../../../rpc/base/v1/ai_knowledge'
+
+const AI_KNOWLEDGE_OPTIONS_URL = '/v1/base/ai/knowledge/options'
+
+/** AI知识库选项服务。 */
+export class AiKnowledgeServiceImpl implements AiKnowledgeService {
+  /** 查询可用的AI知识库选项。 */
+  ListAiKnowledgeOptions(
+    request: ListAiKnowledgeOptionsRequest,
+  ): Promise<ListAiKnowledgeOptionsResponse> {
+    return http<ListAiKnowledgeOptionsResponse>({
+      url: AI_KNOWLEDGE_OPTIONS_URL,
+      method: 'GET',
+      authMode: 'required',
+      data: request,
+    })
+  }
+}
+
+export const defAiKnowledgeService = new AiKnowledgeServiceImpl()

package/src/locales/en-US.json

@@ -116,5 +116,10 @@
   "system.ai.model.load_failed": "Failed to load providers and models",
   "system.ai.model.title": "Select Model",
   "system.ai.model.provider_model_count": "{providers} providers · {models} models",
-  "system.ai.model.empty": "No models available"
+  "system.ai.model.empty": "No models available",
+  "system.ai.knowledge.title": "Knowledge Base",
+  "system.ai.knowledge.selected_count": "{count} knowledge base(s) selected",
+  "system.ai.knowledge.doc_count": "{count} doc(s)",
+  "system.ai.knowledge.select_hint": "Select knowledge bases to inject into context",
+  "system.ai.knowledge.empty": "No knowledge base available"
 }

package/src/locales/ja-JP.json

@@ -116,5 +116,10 @@
   "system.ai.model.load_failed": "プロバイダーとモデルを読み込めませんでした",
   "system.ai.model.title": "モデルを選択",
   "system.ai.model.provider_model_count": "{providers} プロバイダー · {models} モデル",
-  "system.ai.model.empty": "選択可能なモデルがありません"
+  "system.ai.model.empty": "選択可能なモデルがありません",
+  "system.ai.knowledge.title": "ナレッジベース",
+  "system.ai.knowledge.selected_count": "{count} 件のナレッジベースを選択中",
+  "system.ai.knowledge.doc_count": "{count} 件のドキュメント",
+  "system.ai.knowledge.select_hint": "コンテキストに注入するナレッジベースを選択",
+  "system.ai.knowledge.empty": "利用可能なナレッジベースがありません"
 }

package/src/locales/zh-CN.json

@@ -116,5 +116,10 @@
   "system.ai.model.load_failed": "加载供应商和模型失败",
   "system.ai.model.title": "选择模型",
   "system.ai.model.provider_model_count": "{providers} 个供应商 · {models} 个模型",
-  "system.ai.model.empty": "暂无可选模型"
+  "system.ai.model.empty": "暂无可选模型",
+  "system.ai.knowledge.title": "知识库",
+  "system.ai.knowledge.selected_count": "已选 {count} 个知识库",
+  "system.ai.knowledge.doc_count": "{count} 篇文档",
+  "system.ai.knowledge.select_hint": "勾选要注入上下文的知识库",
+  "system.ai.knowledge.empty": "暂无知识库"
 }

package/src/locales/zh-TW.json

@@ -116,5 +116,10 @@
   "system.ai.model.load_failed": "載入供應商與模型失敗",
   "system.ai.model.title": "選擇模型",
   "system.ai.model.provider_model_count": "{providers} 個供應商 · {models} 個模型",
-  "system.ai.model.empty": "暫無可選模型"
+  "system.ai.model.empty": "暫無可選模型",
+  "system.ai.knowledge.title": "知識庫",
+  "system.ai.knowledge.selected_count": "已選 {count} 個知識庫",
+  "system.ai.knowledge.doc_count": "{count} 篇文件",
+  "system.ai.knowledge.select_hint": "勾選要注入上下文的知識庫",
+  "system.ai.knowledge.empty": "暫無知識庫"
 }

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/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/src/views/pagesMember/ai/components/Composer.vue

@@ -1,9 +1,14 @@
 <script setup lang="ts">
 import { computed, ref } from 'vue'
+import { useI18n } from '@liujitcn/kratos-uni-app-core'
 import type { AiAttachment } from '../../../../rpc/base/v1/ai_session'
 import type { AiProviderModelOption } from '../../../../rpc/base/v1/ai_provider'
+import type { AiKnowledgeOption } from '../../../../rpc/base/v1/ai_knowledge'
 import { fallbackModelDisplayName } from '../modelDisplay'
 import ModelPicker from './ModelPicker.vue'
+import KnowledgePicker from './KnowledgePicker.vue'
+
+const { t } = useI18n()
 
 type InputEventValue = {
   detail: {
@@ -19,6 +24,10 @@ const props = defineProps<{
   /** 当前选中的供应商编号和模型名称。 */
   providerId: number
   modelName: string
+  /** 当前可供选择的知识库。 */
+  knowledgeOptions: AiKnowledgeOption[]
+  /** 当前选择的知识库编号列表。 */
+  knowledgeIds: number[]
   modelPlaceholder: string
   placeholder: string
   bottom: string
@@ -34,9 +43,11 @@ const emit = defineEmits<{
   send: []
   'remove-attachment': [attachment: AiAttachment]
   'model-change': [providerId: number, modelName: string]
+  'knowledge-change': [knowledgeIds: number[]]
 }>()
 
 const showModelPicker = ref(false)
+const showKnowledgePicker = ref(false)
 
 const selectedModelDisplayName = computed(() => {
   const provider = props.modelProviders.find((item) => item.provider_id === props.providerId)
@@ -59,6 +70,23 @@ function handleModelSelect(providerId: number, modelName: string) {
   showModelPicker.value = false
   emit('model-change', providerId, modelName)
 }
+
+function openKnowledgePicker() {
+  if (props.sending || !props.knowledgeOptions.length) {
+    return
+  }
+  showKnowledgePicker.value = true
+}
+
+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])
+}
 </script>
 
 <template>
@@ -72,6 +100,22 @@ function handleModelSelect(providerId: number, modelName: string) {
       <text class="composer-model-chip__label">{{ selectedModelDisplayName || modelPlaceholder }}</text>
       <uni-icons type="down" size="12" color="#8a8f99" />
     </view>
+    <view
+      class="composer-model-chip"
+      :class="{
+        'is-disabled': sending || !knowledgeOptions.length,
+        'is-selected': knowledgeIds.length > 0,
+      }"
+      @tap="openKnowledgePicker"
+    >
+      <text class="composer-model-chip__spark">📚</text>
+      <text class="composer-model-chip__label">{{
+        knowledgeIds.length > 0
+          ? t('system.ai.knowledge.selected_count', { count: knowledgeIds.length })
+          : t('system.ai.knowledge.title')
+      }}</text>
+      <uni-icons type="down" size="12" color="#8a8f99" />
+    </view>
     <view class="composer-main">
       <button class="attach-button" hover-class="none" @tap="emit('attach')">
         <uni-icons type="plusempty" size="30" color="#111" />
@@ -124,6 +168,14 @@ function handleModelSelect(providerId: number, modelName: string) {
       @select="handleModelSelect"
       @close="showModelPicker = false"
     />
+
+    <KnowledgePicker
+      v-if="showKnowledgePicker"
+      :options="knowledgeOptions"
+      :selected-ids="knowledgeIds"
+      @toggle="handleKnowledgeToggle"
+      @close="showKnowledgePicker = false"
+    />
   </view>
 </template>
 

package/src/views/pagesMember/ai/components/KnowledgePicker.vue

@@ -0,0 +1,181 @@
+<script setup lang="ts">
+import { computed } from 'vue'
+import { useI18n } from '@liujitcn/kratos-uni-app-core'
+import type { AiKnowledgeOption } from '../../../../rpc/base/v1/ai_knowledge'
+
+const props = defineProps<{
+  options: AiKnowledgeOption[]
+  selectedIds: number[]
+}>()
+
+const emit = defineEmits<{
+  toggle: [knowledgeId: number]
+  close: []
+}>()
+
+const { t } = useI18n()
+
+const selectedCount = computed(() => props.selectedIds.length)
+
+function isSelected(knowledgeId: number) {
+  return props.selectedIds.includes(knowledgeId)
+}
+
+function handleToggle(knowledgeId: number) {
+  emit('toggle', knowledgeId)
+}
+</script>
+
+<template>
+  <view class="knowledge-picker">
+    <view class="knowledge-picker__mask" @tap="emit('close')" />
+    <view class="knowledge-picker__sheet">
+      <view class="knowledge-picker__handle" />
+      <view class="knowledge-picker__head">
+        <text class="knowledge-picker__title">{{ t('system.ai.knowledge.title') }}</text>
+        <text class="knowledge-picker__count">{{
+          t('system.ai.knowledge.selected_count', { count: selectedCount })
+        }}</text>
+        <button class="knowledge-picker__close" hover-class="none" @tap="emit('close')">✕</button>
+      </view>
+      <text class="knowledge-picker__hint">{{ t('system.ai.knowledge.select_hint') }}</text>
+      <scroll-view class="knowledge-picker__body" scroll-y :show-scrollbar="true">
+        <view
+          v-for="option in options"
+          :key="option.id"
+          class="knowledge-picker__row"
+          :class="{ 'is-selected': isSelected(option.id) }"
+          @tap="handleToggle(option.id)"
+        >
+          <view class="knowledge-picker__row-main">
+            <text class="knowledge-picker__row-name">{{ option.name }}</text>
+            <text v-if="option.doc_count" class="knowledge-picker__row-meta">{{
+              t('system.ai.knowledge.doc_count', { count: option.doc_count })
+            }}</text>
+          </view>
+          <view class="knowledge-picker__check" :class="{ 'is-checked': isSelected(option.id) }">
+            <text v-if="isSelected(option.id)">✓</text>
+          </view>
+        </view>
+      </scroll-view>
+    </view>
+  </view>
+</template>
+
+<style lang="scss" scoped>
+.knowledge-picker {
+  position: fixed;
+  inset: 0;
+  z-index: 1000;
+  display: flex;
+  align-items: flex-end;
+}
+
+.knowledge-picker__mask {
+  position: absolute;
+  inset: 0;
+  background: rgba(0, 0, 0, 0.45);
+}
+
+.knowledge-picker__sheet {
+  position: relative;
+  display: flex;
+  flex-direction: column;
+  width: 100%;
+  max-height: 62vh;
+  padding: 10px 20px calc(20px + env(safe-area-inset-bottom));
+  background: #fff;
+  border-radius: 20px 20px 0 0;
+}
+
+.knowledge-picker__handle {
+  width: 36px;
+  height: 4px;
+  margin: 0 auto 10px;
+  background: #e5e6eb;
+  border-radius: 2px;
+}
+
+.knowledge-picker__head {
+  display: flex;
+  align-items: center;
+  gap: 10px;
+}
+
+.knowledge-picker__title {
+  font-size: 17px;
+  font-weight: 600;
+  color: #111;
+}
+
+.knowledge-picker__count {
+  flex: 1;
+  font-size: 12px;
+  color: #8a8f99;
+}
+
+.knowledge-picker__close {
+  padding: 0;
+  font-size: 16px;
+  line-height: 1;
+  color: #8a8f99;
+  background: transparent;
+}
+
+.knowledge-picker__close::after {
+  border: none;
+}
+
+.knowledge-picker__hint {
+  margin-top: 6px;
+  font-size: 12px;
+  color: #8a8f99;
+}
+
+.knowledge-picker__body {
+  margin-top: 10px;
+}
+
+.knowledge-picker__row {
+  display: flex;
+  align-items: center;
+  justify-content: space-between;
+  gap: 12px;
+  padding: 12px 4px;
+  border-bottom: 1px solid #f2f3f5;
+}
+
+.knowledge-picker__row-main {
+  display: flex;
+  flex-direction: column;
+  gap: 2px;
+  min-width: 0;
+}
+
+.knowledge-picker__row-name {
+  font-size: 15px;
+  color: #111;
+}
+
+.knowledge-picker__row-meta {
+  font-size: 12px;
+  color: #8a8f99;
+}
+
+.knowledge-picker__check {
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  width: 22px;
+  height: 22px;
+  border: 1px solid #d0d2d6;
+  border-radius: 50%;
+  font-size: 13px;
+  color: #fff;
+}
+
+.knowledge-picker__check.is-checked {
+  background: #00a96b;
+  border-color: #00a96b;
+}
+</style>

package/src/views/pagesMember/ai/index.vue

@@ -8,9 +8,11 @@ import {
 import { defAiSessionService } from '../../../api/base/v1/ai_session'
 import { defAiToolService } from '../../../api/base/v1/ai_tool'
 import { defAiModelService } from '../../../api/base/v1/ai_provider'
+import { defAiKnowledgeService } from '../../../api/base/v1/ai_knowledge'
 import type { AiMessage } from '../../../rpc/base/v1/ai_session'
 import type { AiAttachment, AiSession } from '../../../rpc/base/v1/ai_session'
 import type { AiProviderModelOption } from '../../../rpc/base/v1/ai_provider'
+import type { AiKnowledgeOption } from '../../../rpc/base/v1/ai_knowledge'
 import type { AiShortcut, AiToolCall } from '../../../rpc/base/v1/ai_tool'
 import { AiMessageStatus } from '../../../rpc/base/v1/ai_session'
 import { Terminal } from '../../../rpc/base/v1/ai_tool'
@@ -100,6 +102,8 @@ const selectedAttachments = ref<AiAttachment[]>([])
 const modelProviders = ref<AiProviderModelOption[]>([])
 const selectedProviderId = ref(0)
 const selectedModelName = ref('')
+const knowledgeOptions = ref<AiKnowledgeOption[]>([])
+const selectedKnowledgeIds = ref<number[]>([])
 const runningStreamTaskMap = new Map<string, StreamTask>()
 const pendingDeltaMap = new Map<string, AiStreamPayload>()
 let pendingDeltaTimer = 0
@@ -472,11 +476,16 @@ async function loadAiShortcuts() {
   }
   loadingShortcuts.value = true
   try {
-    const [response, providerResponse] = await Promise.all([
+    const [response, providerResponse, knowledgeResponse] = await Promise.all([
       defAiToolService.ListAiShortcut({ terminal: AI_TERMINAL }),
       defAiModelService.ListAiProviderModelOptions({}),
+      defAiKnowledgeService.ListAiKnowledgeOptions({}),
     ])
     modelProviders.value = providerResponse.providers ?? []
+    knowledgeOptions.value = knowledgeResponse.options ?? []
+    selectedKnowledgeIds.value = selectedKnowledgeIds.value.filter((id) =>
+      knowledgeOptions.value.some((option) => option.id === id),
+    )
     applyModelSelection()
     const shortcuts = normalizeStarterShortcuts(response.shortcuts).filter((item) => !item.action)
     if (shortcuts.length) {
@@ -487,6 +496,7 @@ async function loadAiShortcuts() {
     modelProviders.value = []
     selectedProviderId.value = 0
     selectedModelName.value = ''
+    knowledgeOptions.value = []
     showError(error, t('system.ai.load_shortcuts_failed'))
   } finally {
     loadingShortcuts.value = false
@@ -520,6 +530,11 @@ function handleModelChange(providerId: number, modelName: string) {
   saveModelSelection({ provider_id: providerId, model_name: modelName })
 }
 
+/** 更新当前聊天检索注入的知识库选择。 */
+function handleKnowledgeChange(knowledgeIds: number[]) {
+  selectedKnowledgeIds.value = knowledgeIds
+}
+
 async function sendAiPayload(payload: { text: string; attachments: AiAttachment[] }) {
   const sessionID = await ensureActiveSession()
   if (!sessionID || isSessionSending(sessionID)) {
@@ -550,6 +565,7 @@ async function runAiTask(
     content: payload.text,
     provider_id: selectedProviderId.value,
     model_name: selectedModelName.value,
+    knowledge_base_ids: selectedKnowledgeIds.value,
     attachments: payload.attachments,
     action: undefined,
   }
@@ -1332,6 +1348,8 @@ function showError(error: unknown, fallback: string) {
       :model-providers="modelProviders"
       :provider-id="selectedProviderId"
       :model-name="selectedModelName"
+      :knowledge-options="knowledgeOptions"
+      :knowledge-ids="selectedKnowledgeIds"
       :model-placeholder="t('system.ai.model.model_placeholder')"
       :placeholder="composerPlaceholder"
       :bottom="composerBottom"
@@ -1342,6 +1360,7 @@ function showError(error: unknown, fallback: string) {
       @record="handleToggleRecord"
       @send="handleSend"
       @model-change="handleModelChange"
+      @knowledge-change="handleKnowledgeChange"
       @remove-attachment="removeSelectedAttachment"
     />