From b4e31c8ea9a00e6d28ec8382f90bcce62c9cf5d7 Mon Sep 17 00:00:00 2001 From: aries-ckt <916701291@qq.com> Date: Tue, 6 Jun 2023 11:50:56 +0800 Subject: [PATCH 1/2] fix: pdf problem --- pilot/source_embedding/pdf_embedding.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pilot/source_embedding/pdf_embedding.py b/pilot/source_embedding/pdf_embedding.py index 6eced03f3..54f77fb81 100644 --- a/pilot/source_embedding/pdf_embedding.py +++ b/pilot/source_embedding/pdf_embedding.py @@ -15,12 +15,11 @@ CFG = Config() class PDFEmbedding(SourceEmbedding): """pdf embedding for read pdf document.""" - def __init__(self, file_path, vector_store_config, encoding): + def __init__(self, file_path, vector_store_config): """Initialize with pdf path.""" super().__init__(file_path, vector_store_config) self.file_path = file_path self.vector_store_config = vector_store_config - self.encoding = encoding @register def read(self): From 654f55a19409b6afd9b33e336fdb77650c9b7051 Mon Sep 17 00:00:00 2001 From: aries-ckt <916701291@qq.com> Date: Tue, 6 Jun 2023 12:45:04 +0800 Subject: [PATCH 2/2] fix: url embedding problem --- pilot/source_embedding/knowledge_embedding.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pilot/source_embedding/knowledge_embedding.py b/pilot/source_embedding/knowledge_embedding.py index 27297111b..7ec0de76c 100644 --- a/pilot/source_embedding/knowledge_embedding.py +++ b/pilot/source_embedding/knowledge_embedding.py @@ -49,7 +49,6 @@ class KnowledgeEmbedding: if self.file_type == "url": embedding = URLEmbedding( file_path=self.file_path, - model_name=self.model_name, vector_store_config=self.vector_store_config, ) return embedding