feat: add tag filtering and input support in document search functionality

Author: liuruibin

apps/knowledge/serializers/document.py

@@ -389,6 +389,7 @@ class Query(serializers.Serializer):
         task_type = serializers.IntegerField(required=False, label=_('task type'))
         status = serializers.CharField(required=False, label=_('status'), allow_null=True, allow_blank=True)
         order_by = serializers.CharField(required=False, label=_('order by'), allow_null=True, allow_blank=True)
+        tag = serializers.CharField(required=False, label=_('tag'), allow_null=True, allow_blank=True)
 
         def get_query_set(self):
             query_set = QuerySet(model=Document)
@@ -414,6 +415,12 @@ def get_query_set(self):
                         query_set = query_set.filter(status__icontains=status)
                     else:
                         query_set = query_set.filter(status__iregex='^[2n]*$')
+            if 'tag' in self.data and self.data.get('tag') not in [None, '']:
+                tag_name = self.data.get('tag')
+                document_id_list = QuerySet(DocumentTag).filter(
+                    Q(tag__key__icontains=tag_name) | Q(tag__value__icontains=tag_name)
+                ).values_list('document_id', flat=True)
+                query_set = query_set.filter(id__in=document_id_list)
             order_by = self.data.get('order_by', '')
             order_by_query_set = QuerySet(model=get_dynamics_model(
                 {'char_length': models.CharField(), 'paragraph_count': models.IntegerField(),

apps/knowledge/views/document.py

@@ -597,6 +597,7 @@ def get(self, request: Request, workspace_id: str, knowledge_id: str, current_pa
                     'knowledge_id': knowledge_id,
                     'folder_id': request.query_params.get('folder_id'),
                     'name': request.query_params.get('name'),
+                    'tag': request.query_params.get('tag'),
                     'desc': request.query_params.get("desc"),
                     'user_id': request.query_params.get('user_id'),
                     'status': request.query_params.get('status'),

ui/src/locales/lang/en-US/views/document.ts

@@ -114,6 +114,7 @@ export default {
       'After deletion, resources using this tag will have the tag removed. Please proceed with caution!',
     requiredMessage1: 'Please enter a tag',
     requiredMessage2: 'Please enter a value',
+    requiredMessage3: 'Please enter a tag or value',
   },
   table: {
     name: 'Document Name',

ui/src/locales/lang/zh-CN/views/document.ts

@@ -109,6 +109,7 @@ export default {
     deleteTip: '删除后使用该标签的资源将会删除该标签，请谨慎操作!',
     requiredMessage1: '请输入标签',
     requiredMessage2: '请输入标签值',
+    requiredMessage3: '请输入标签或标签值',
   },
   table: {
     name: '文件名称',

ui/src/locales/lang/zh-Hant/views/document.ts

@@ -111,6 +111,7 @@ export default {
     deleteTip: '刪除後使用該標籤的資源將會刪除該標籤，請謹慎操作!',
     requiredMessage1: '請輸入標籤',
     requiredMessage2: '請輸入標籤值',
+    requiredMessage3: '請輸入標籤或標籤值',
   },
   table: {
     name: '文件名稱',

ui/src/views/document/index.vue

@@ -104,15 +104,34 @@
                 </el-dropdown>
               </template>
             </div>
-            <div>
-              <el-input
-                v-model="filterText"
-                :placeholder="$t('common.searchBar.placeholder')"
-                prefix-icon="Search"
-                class="w-240"
-                @change="getList"
-                clearable
-              />
+            <div class="flex">
+              <div class="flex-between complex-search">
+                <el-select
+                  class="complex-search__left"
+                  v-model="search_type"
+                  style="width: 120px"
+                  @change="search_type_change"
+                >
+                  <el-option :label="$t('dynamicsForm.tag.label')" value="tag" />
+                  <el-option :label="$t('views.tool.form.toolName.label')" value="name" />
+                </el-select>
+                <el-input
+                  v-if="search_type === 'name'"
+                  v-model="search_form.name"
+                  @change="refresh"
+                  :placeholder="$t('common.searchBar.placeholder')"
+                  style="width: 220px"
+                  clearable
+                />
+                <el-input
+                  v-if="search_type === 'tag'"
+                  v-model="search_form.tag"
+                  @change="refresh"
+                  :placeholder="$t('views.document.tag.requiredMessage3')"
+                  style="width: 220px"
+                  clearable
+                />
+              </div>
               <el-button @click="openTagDrawer" class="ml-12">
                 {{ $t('views.document.tag.label') }}
               </el-button>
@@ -781,6 +800,12 @@ const getTaskState = (status: string, taskType: number) => {
   return taskType - 1 > statusList.length + 1 ? 'n' : statusList[taskType - 1]
 }
 
+const search_type = ref('name')
+const search_form = ref<any>({
+  name: '',
+  tag: '',
+})
+
 const beforePagination = computed(() => common.paginationConfig[storeKey])
 const beforeSearch = computed(() => common.search[storeKey])
 const embeddingContentDialogRef = ref<InstanceType<typeof EmbeddingContentDialog>>()
@@ -1184,11 +1209,13 @@ function handleSortChange({ prop, order }: { prop: string; order: string }) {
 
 function getList(bool?: boolean) {
   const param = {
-    ...(filterText.value && { name: filterText.value }),
     ...filterMethod.value,
     order_by: orderBy.value,
     folder_id: folderId,
   }
+  if (search_form.value[search_type.value]) {
+    param[search_type.value] = search_form.value[search_type.value]
+  }
   loadSharedApi({ type: 'document', isShared: isShared.value, systemType: apiType.value })
     .getDocumentPage(id as string, paginationConfig.value, param, bool ? undefined : loading)
     .then((res: any) => {
@@ -1197,6 +1224,10 @@ function getList(bool?: boolean) {
     })
 }
 
+const search_type_change = () => {
+  search_form.value = { name: '', tag: '' }
+}
+
 function getDetail() {
   loadSharedApi({ type: 'knowledge', isShared: isShared.value, systemType: apiType.value })
     .getKnowledgeDetail(id, loading)