diff --git a/apps/application/serializers/application_serializers.py b/apps/application/serializers/application_serializers.py
index 1827c5eb9..aba6e3f7a 100644
--- a/apps/application/serializers/application_serializers.py
+++ b/apps/application/serializers/application_serializers.py
@@ -41,7 +41,7 @@ from setting.serializers.provider_serializers import ModelSerializer
 from smartdoc.conf import PROJECT_DIR
 
 token_cache = cache.caches['token_cache']
-chat_cache = cache.caches['chat_cache']
+chat_cache = cache.caches['model_cache']
 
 
 class ModelDatasetAssociation(serializers.Serializer):
@@ -532,6 +532,7 @@ class ApplicationSerializer(serializers.Serializer):
                 QuerySet(ApplicationDatasetMapping).bulk_create(
                     [ApplicationDatasetMapping(application_id=application_id, dataset_id=dataset_id) for dataset_id in
                      dataset_id_list]) if len(dataset_id_list) > 0 else None
+            chat_cache.clear_by_application_id(application_id)
             return self.one(with_valid=False)
 
         def list_dataset(self, with_valid=True):
diff --git a/apps/common/cache/mem_cache.py b/apps/common/cache/mem_cache.py
index 9bb6c45ba..86a3ce040 100644
--- a/apps/common/cache/mem_cache.py
+++ b/apps/common/cache/mem_cache.py
@@ -29,3 +29,15 @@ class MemCache(LocMemCache):
             pickled = self._cache[key]
             self._cache.move_to_end(key, last=False)
         return pickled
+
+    def clear_by_application_id(self, application_id):
+        delete_keys = []
+        for key in self._cache.keys():
+            value = self._cache.get(key)
+            if (hasattr(value,
+                        'application') and value.application is not None and value.application.id is not None and
+                    str(
+                        value.application.id) == application_id):
+                delete_keys.append(key)
+        for key in delete_keys:
+            self._delete(key)