diff --git a/apps/knowledge/serializers/document.py b/apps/knowledge/serializers/document.py index 6cacf727e..982f74439 100644 --- a/apps/knowledge/serializers/document.py +++ b/apps/knowledge/serializers/document.py @@ -389,6 +389,7 @@ class DocumentSerializers(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 @@ class DocumentSerializers(serializers.Serializer): 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(), diff --git a/apps/knowledge/views/document.py b/apps/knowledge/views/document.py index a63883856..09b43c072 100644 --- a/apps/knowledge/views/document.py +++ b/apps/knowledge/views/document.py @@ -597,6 +597,7 @@ class DocumentView(APIView): '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'), diff --git a/ui/src/locales/lang/en-US/views/document.ts b/ui/src/locales/lang/en-US/views/document.ts index 3e3048cbc..3251c53fb 100644 --- a/ui/src/locales/lang/en-US/views/document.ts +++ b/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', diff --git a/ui/src/locales/lang/zh-CN/views/document.ts b/ui/src/locales/lang/zh-CN/views/document.ts index 7cae80a4c..19b192a4c 100644 --- a/ui/src/locales/lang/zh-CN/views/document.ts +++ b/ui/src/locales/lang/zh-CN/views/document.ts @@ -109,6 +109,7 @@ export default { deleteTip: '删除后使用该标签的资源将会删除该标签,请谨慎操作!', requiredMessage1: '请输入标签', requiredMessage2: '请输入标签值', + requiredMessage3: '请输入标签或标签值', }, table: { name: '文件名称', diff --git a/ui/src/locales/lang/zh-Hant/views/document.ts b/ui/src/locales/lang/zh-Hant/views/document.ts index d34b3d3ac..46286704d 100644 --- a/ui/src/locales/lang/zh-Hant/views/document.ts +++ b/ui/src/locales/lang/zh-Hant/views/document.ts @@ -111,6 +111,7 @@ export default { deleteTip: '刪除後使用該標籤的資源將會刪除該標籤,請謹慎操作!', requiredMessage1: '請輸入標籤', requiredMessage2: '請輸入標籤值', + requiredMessage3: '請輸入標籤或標籤值', }, table: { name: '文件名稱', diff --git a/ui/src/views/document/index.vue b/ui/src/views/document/index.vue index 8e74cc721..e913e9df3 100644 --- a/ui/src/views/document/index.vue +++ b/ui/src/views/document/index.vue @@ -104,15 +104,34 @@ -