'commit'

dsRagAnything/Test.py

@@ -95,23 +95,21 @@ async def main():
             base_url=Config.Config.EMBED_BASE_URL,
         ),
     )
-    from functools import partial
+    # from functools import partial
-    from lightrag.rerank import cohere_rerank
+    # from lightrag.rerank import cohere_rerank
-
+    # rerank_model_func = partial(
-    rerank_model_func = partial(
+    #     cohere_rerank,
-        cohere_rerank,
+    #     model=Config.Config.RERANK_MODEL,
-        model=Config.Config.RERANK_MODEL,
+    #     api_key=Config.Config.RERANK_BINDING_API_KEY,
-        api_key=Config.Config.RERANK_BINDING_API_KEY,
+    #     base_url=Config.Config.RERANK_BASE_URL,
-        base_url=Config.Config.RERANK_BASE_URL,
+    # )
-    )
 
     # 初始化 RAGAnything
     rag = RAGAnything(
         config=config,
         llm_model_func=llm_model_func,
         vision_model_func=vision_model_func,
-        embedding_func=embedding_func,
+        embedding_func=embedding_func
-        rerank_model_func=rerank_model_func,
     )
 
     # 处理文档