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 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):