|
11 | 11 | import uuid_utils.compat as uuid |
12 | 12 | from celery_once import AlreadyQueued |
13 | 13 | from django.core import validators |
| 14 | +from django.core.cache import cache |
14 | 15 | from django.db import transaction, models |
15 | 16 | from django.db.models import QuerySet |
16 | 17 | from django.db.models.functions import Reverse, Substr |
|
20 | 21 |
|
21 | 22 | from application.models import ApplicationKnowledgeMapping |
22 | 23 | from common.config.embedding_config import VectorStore |
| 24 | +from common.constants.cache_version import Cache_Version |
23 | 25 | from common.constants.permission_constants import ResourceAuthType, ResourcePermissionGroup |
24 | 26 | from common.database_model_manage.database_model_manage import DatabaseModelManage |
25 | 27 | from common.db.search import native_search, get_dynamics_model, native_page_search |
@@ -534,6 +536,10 @@ def save_base(self, instance, with_valid=True): |
534 | 536 | user_id=self.data.get('user_id'), |
535 | 537 | auth_type=ResourceAuthType.RESOURCE_PERMISSION_GROUP |
536 | 538 | ).save() |
| 539 | + # 刷新缓存 |
| 540 | + version = Cache_Version.PERMISSION_LIST.get_version() |
| 541 | + key = Cache_Version.PERMISSION_LIST.get_key(user_id=self.data.get('user_id')) |
| 542 | + cache.delete(key, version=version) |
537 | 543 |
|
538 | 544 | return { |
539 | 545 | **KnowledgeModelSerializer(knowledge).data, |
@@ -581,6 +587,11 @@ def save_web(self, instance: Dict, with_valid=True): |
581 | 587 | user_id=self.data.get('user_id'), |
582 | 588 | auth_type=ResourceAuthType.RESOURCE_PERMISSION_GROUP |
583 | 589 | ).save() |
| 590 | + # 刷新缓存 |
| 591 | + version = Cache_Version.PERMISSION_LIST.get_version() |
| 592 | + key = Cache_Version.PERMISSION_LIST.get_key(user_id=self.data.get('user_id')) |
| 593 | + cache.delete(key, version=version) |
| 594 | + |
584 | 595 | sync_web_knowledge.delay(str(knowledge_id), instance.get('source_url'), instance.get('selector')) |
585 | 596 | return {**KnowledgeModelSerializer(knowledge).data, 'document_list': []} |
586 | 597 |
|
|
0 commit comments