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

feat: remove user_id permission check for embedding model in knowledge processing

Browse Source
This commit is contained in:
CaptainB 2025-06-03 11:24:08 +08:00
parent 787da980b8
commit 15eea403ec
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -181,12 +181,9 @@ class KnowledgeSerializer(serializers.Serializer):
knowledge_id = self.data.get('knowledge_id') knowledge_id = self.data.get('knowledge_id')
knowledge = QuerySet(Knowledge).filter(id=knowledge_id).first() knowledge = QuerySet(Knowledge).filter(id=knowledge_id).first()
embedding_model_id = knowledge.embedding_model_id embedding_model_id = knowledge.embedding_model_id
knowledge_user_id = knowledge.user_id
embedding_model = QuerySet(Model).filter(id=embedding_model_id).first() embedding_model = QuerySet(Model).filter(id=embedding_model_id).first()
if embedding_model is None: if embedding_model is None:
raise AppApiException(500, _('Model does not exist')) raise AppApiException(500, _('Model does not exist'))
if embedding_model.permission_type == 'PRIVATE' and knowledge_user_id != embedding_model.user_id:
raise AppApiException(500, _('No permission to use this model') + f"{embedding_model.name}")
ListenerManagement.update_status( ListenerManagement.update_status(
QuerySet(Document).filter(knowledge_id=self.data.get('knowledge_id')), QuerySet(Document).filter(knowledge_id=self.data.get('knowledge_id')),
TaskType.EMBEDDING, TaskType.EMBEDDING,