From 6fdeff9a0a76a89bd75ae958e1f61c38f09d2ce4 Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Fri, 30 Jun 2023 15:02:18 +0800
Subject: [PATCH] fix:knowledge_init.py multi document cannot search answser
 1.rebuild knowledge_init.py Close #285

---
 pilot/source_embedding/source_embedding.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/pilot/source_embedding/source_embedding.py b/pilot/source_embedding/source_embedding.py
index 3d881fcdf..75e42dfbf 100644
--- a/pilot/source_embedding/source_embedding.py
+++ b/pilot/source_embedding/source_embedding.py
@@ -33,9 +33,7 @@ class SourceEmbedding(ABC):
         self.vector_store_config = vector_store_config
         self.embedding_args = embedding_args
         self.embeddings = vector_store_config["embeddings"]
-        self.vector_client = VectorStoreConnector(
-            CFG.VECTOR_STORE_TYPE, vector_store_config
-        )
+
 
     @abstractmethod
     @register
@@ -59,11 +57,17 @@ class SourceEmbedding(ABC):
     @register
     def index_to_store(self, docs):
         """index to vector store"""
+        self.vector_client = VectorStoreConnector(
+            CFG.VECTOR_STORE_TYPE, self.vector_store_config
+        )
         self.vector_client.load_document(docs)
 
     @register
     def similar_search(self, doc, topk):
         """vector store similarity_search"""
+        self.vector_client = VectorStoreConnector(
+            CFG.VECTOR_STORE_TYPE, self.vector_store_config
+        )
         try:
             ans = self.vector_client.similar_search(doc, topk)
         except NotEnoughElementsException:
@@ -71,6 +75,9 @@ class SourceEmbedding(ABC):
         return ans
 
     def vector_name_exist(self):
+        self.vector_client = VectorStoreConnector(
+            CFG.VECTOR_STORE_TYPE, self.vector_store_config
+        )
         return self.vector_client.vector_name_exists()
 
     def source_embedding(self):