diff --git a/pilot/vector_store/weaviate_store.py b/pilot/vector_store/weaviate_store.py
index 795cf21f9..14b90eb54 100644
--- a/pilot/vector_store/weaviate_store.py
+++ b/pilot/vector_store/weaviate_store.py
@@ -1,10 +1,6 @@
 import os
-import json
 import logging
-import weaviate
 from langchain.schema import Document
-from langchain.vectorstores import Weaviate
-from weaviate.exceptions import WeaviateBaseError
 
 from pilot.configs.config import Config
 from pilot.configs.model_config import KNOWLEDGE_UPLOAD_ROOT_PATH
@@ -72,7 +68,7 @@ class WeaviateStore(VectorStoreBase):
             if self.vector_store_client.schema.get(self.vector_name):
                 return True
             return False
-        except WeaviateBaseError as e:
+        except Exception as e:
             logger.error("vector_name_exists error", e.message)
             return False