zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: simplify validation checks in document and paragraph retrieval methods

Browse Source
This commit is contained in:
CaptainB 2025-07-31 15:37:38 +08:00
parent 3d1ebad70b
commit 7fc0bd2396
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/knowledge/serializers/document.py
View File

@ -606,8 +606,7 @@ class DocumentSerializers(serializers.Serializer):
return FileSerializer.Operate(data={'id': file.id}).get(with_valid=True) return FileSerializer.Operate(data={'id': file.id}).get(with_valid=True)
def one(self, with_valid=False): def one(self, with_valid=False):
if with_valid: self.is_valid(raise_exception=True)
self.is_valid(raise_exception=True)
query_set = QuerySet(model=Document) query_set = QuerySet(model=Document)
query_set = query_set.filter(**{'id': self.data.get("document_id")}) query_set = query_set.filter(**{'id': self.data.get("document_id")})
return native_search({ return native_search({

1
apps/knowledge/serializers/paragraph.py
View File

@ -361,6 +361,7 @@ class ParagraphSerializers(serializers.Serializer):
raise AppApiException(500, _('Knowledge id does not exist')) raise AppApiException(500, _('Knowledge id does not exist'))
def get_query_set(self): def get_query_set(self):
self.is_valid()
query_set = QuerySet(model=Paragraph) query_set = QuerySet(model=Paragraph)
query_set = query_set.filter( query_set = query_set.filter(
**{'knowledge_id': self.data.get('knowledge_id'), 'document_id': self.data.get("document_id")}) **{'knowledge_id': self.data.get('knowledge_id'), 'document_id': self.data.get("document_id")})