diff --git a/pilot/embedding_engine/loader/csv_loader.py b/pilot/embedding_engine/loader/csv_loader.py index bdb58112e..c1b4ea7cb 100644 --- a/pilot/embedding_engine/loader/csv_loader.py +++ b/pilot/embedding_engine/loader/csv_loader.py @@ -25,11 +25,11 @@ from langchain.schema import Document class NewCSVLoader(BaseLoader): def __init__( - self, - file_path: str, - source_column: Optional[str] = None, - csv_args: Optional[Dict] = None, - encoding: Optional[str] = None, + self, + file_path: str, + source_column: Optional[str] = None, + csv_args: Optional[Dict] = None, + encoding: Optional[str] = None, ): """ diff --git a/tests/unit/embedding_engine/document_test.py b/tests/unit/embedding_engine/document_test.py index 54c764afd..bf8254bf9 100644 --- a/tests/unit/embedding_engine/document_test.py +++ b/tests/unit/embedding_engine/document_test.py @@ -12,9 +12,10 @@ vector_store_config = { # it can be .md,.pdf,.docx, .csv, .html document_path = "your_path/test.md" embedding_engine = EmbeddingEngine( - knowledge_source=document_path, - knowledge_type=KnowledgeType.DOCUMENT.value, - model_name=embedding_model, - vector_store_config=vector_store_config) + knowledge_source=document_path, + knowledge_type=KnowledgeType.DOCUMENT.value, + model_name=embedding_model, + vector_store_config=vector_store_config, +) # embedding document content to vector store -embedding_engine.knowledge_embedding() \ No newline at end of file +embedding_engine.knowledge_embedding()