From d8b4e6bdc854082b087514d7305b445924b5d2ea Mon Sep 17 00:00:00 2001 From: aries-ckt <916701291@qq.com> Date: Thu, 1 Jun 2023 20:34:00 +0800 Subject: [PATCH] fix:url embedding --- pilot/scene/chat_knowledge/url/chat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pilot/scene/chat_knowledge/url/chat.py b/pilot/scene/chat_knowledge/url/chat.py index 0c7ac1cd2..cc8d89d4a 100644 --- a/pilot/scene/chat_knowledge/url/chat.py +++ b/pilot/scene/chat_knowledge/url/chat.py @@ -52,7 +52,9 @@ class ChatUrlKnowledge(BaseChat): ) # url soruce in vector - self.knowledge_embedding_client.knowledge_embedding() + if not self.knowledge_embedding_client.vector_exist(): + self.knowledge_embedding_client.knowledge_embedding() + logger.info("url embedding success") def generate_input_values(self): docs = self.knowledge_embedding_client.similar_search(