diff --git a/pilot/embedding_engine/source_embedding.py b/pilot/embedding_engine/source_embedding.py index b99529cf9..372e35c22 100644 --- a/pilot/embedding_engine/source_embedding.py +++ b/pilot/embedding_engine/source_embedding.py @@ -33,9 +33,6 @@ 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 +56,17 @@ class SourceEmbedding(ABC): @register def index_to_store(self, docs): """index to vector store""" - return self.vector_client.load_document(docs) + 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 +74,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): diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py index 0866e8be9..b6c1e2cc3 100644 --- a/pilot/server/webserver.py +++ b/pilot/server/webserver.py @@ -297,39 +297,29 @@ def http_bot( if ChatScene.ChatWithDbExecute == scene: chat_param = { - "temperature": temperature, - "max_new_tokens": max_new_tokens, "chat_session_id": state.conv_id, "db_name": db_selector, "user_input": state.last_user_input, } elif ChatScene.ChatWithDbQA == scene: chat_param = { - "temperature": temperature, - "max_new_tokens": max_new_tokens, "chat_session_id": state.conv_id, "db_name": db_selector, "user_input": state.last_user_input, } elif ChatScene.ChatExecution == scene: chat_param = { - "temperature": temperature, - "max_new_tokens": max_new_tokens, "chat_session_id": state.conv_id, "plugin_selector": plugin_selector, "user_input": state.last_user_input, } elif ChatScene.ChatNormal == scene: chat_param = { - "temperature": temperature, - "max_new_tokens": max_new_tokens, "chat_session_id": state.conv_id, "user_input": state.last_user_input, } elif ChatScene.ChatDefaultKnowledge == scene: chat_param = { - "temperature": temperature, - "max_new_tokens": max_new_tokens, "chat_session_id": state.conv_id, "user_input": state.last_user_input, } @@ -341,8 +331,6 @@ def http_bot( } elif ChatScene.ChatUrlKnowledge == scene: chat_param = { - "temperature": temperature, - "max_new_tokens": max_new_tokens, "chat_session_id": state.conv_id, "user_input": state.last_user_input, "url": url_input, diff --git a/tools/knowledge_init.py b/tools/knowledge_init.py index 752cf18d0..66285a784 100644 --- a/tools/knowledge_init.py +++ b/tools/knowledge_init.py @@ -49,6 +49,7 @@ class LocalKnowledgeInit: 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)