Reformat

Author: tomaarsen

lightrag/llm/sentence_transformers.py

@@ -13,14 +13,20 @@
 async def sentence_transformers_embed(
     texts: list[str], model: SentenceTransformer
 ) -> np.ndarray:
-    async def inner_encode(texts: list[str], model: SentenceTransformer, embedding_dim: int = 1024):
+    async def inner_encode(
+        texts: list[str], model: SentenceTransformer, embedding_dim: int = 1024
+    ):
         return model.encode(
             texts,
             truncate_dim=embedding_dim,
             convert_to_numpy=True,
             convert_to_tensor=False,
             show_progress_bar=False,
         )
-    
-    embedding_func = EmbeddingFunc(embedding_dim=model.get_sentence_embedding_dimension(), func=inner_encode, max_token_size=model.get_max_seq_length())
+
+    embedding_func = EmbeddingFunc(
+        embedding_dim=model.get_sentence_embedding_dimension(),
+        func=inner_encode,
+        max_token_size=model.get_max_seq_length(),
+    )
     return await embedding_func(texts, model=model)

lightrag/rerank.py

@@ -384,7 +384,7 @@ async def main():
                 print(f"Document: {docs[item['index']]}")
         except Exception as e:
             print(f"Aliyun Error: {e}")
-        
+
         # Test Sentence Transformers rerank
         try:
             print("\n=== Sentence Transformers Rerank ===")