From 937e5f4528e7f9da82b71a1ea9ca9ab8531550ad Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Tue, 4 Jul 2023 16:40:47 +0800
Subject: [PATCH] fix:knowledge init try exception

---
 tools/knowledge_init.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/tools/knowledge_init.py b/tools/knowledge_init.py
index 30446cc6c..6f46b713f 100644
--- a/tools/knowledge_init.py
+++ b/tools/knowledge_init.py
@@ -46,12 +46,18 @@ class LocalKnowledgeInit:
                 docs.extend(doc)
         embedding_engine.index_to_store(docs)
         print(f"""begin create {self.vector_store_config["vector_store_name"]} space""")
-        space = KnowledgeSpaceRequest
-        space.name = self.vector_store_config["vector_store_name"]
-        space.desc = "knowledge_init.py"
-        space.vector_type = CFG.VECTOR_STORE_TYPE
-        space.owner = "DB-GPT"
-        knowledge_space_service.create_knowledge_space(space)
+        try:
+            space = KnowledgeSpaceRequest
+            space.name = self.vector_store_config["vector_store_name"]
+            space.desc = "knowledge_init.py"
+            space.vector_type = CFG.VECTOR_STORE_TYPE
+            space.owner = "DB-GPT"
+            knowledge_space_service.create_knowledge_space(space)
+        except Exception as e:
+            if "have already named" in str(e):
+                print(f"Warning: you have already named {space.name}")
+            else:
+                raise e
 
 
 if __name__ == "__main__":