style:format code

2025-09-12 20:53:48 +00:00 · 2023-06-30 15:04:33 +08:00
parent 6fdeff9a0a
commit f4aa58a954
3 changed files with 7 additions and 4 deletions
--- a/pilot/source_embedding/knowledge_embedding.py
+++ b/pilot/source_embedding/knowledge_embedding.py
@@ -51,6 +51,7 @@ class KnowledgeEmbedding:
        self.knowledge_embedding_client.index_to_store(docs)

    def read(self):
+        self.knowledge_embedding_client = self.init_knowledge_embedding()
        return self.knowledge_embedding_client.read_batch()

    def init_knowledge_embedding(self):
--- a/pilot/source_embedding/source_embedding.py
+++ b/pilot/source_embedding/source_embedding.py
@@ -34,7 +34,6 @@ class SourceEmbedding(ABC):
        self.embedding_args = embedding_args
        self.embeddings = vector_store_config["embeddings"]

-
    @abstractmethod
    @register
    def read(self) -> List[ABC]: