jina rerank working

2026-01-13 09:51:35 +08:00
parent 370fe6368a
commit 9745ca2476
23 changed files with 1967 additions and 6 deletions
--- a/LightRAG-main/lightrag/api/lightrag_server.py
+++ b/LightRAG-main/lightrag/api/lightrag_server.py
@@ -538,13 +538,14 @@ def create_app(args):
    # Configure rerank function based on args.rerank_bindingparameter
    rerank_model_func = None
    if args.rerank_binding != "null":
-        from lightrag.rerank import cohere_rerank, jina_rerank, ali_rerank
+        from lightrag.rerank import cohere_rerank, jina_rerank, ali_rerank, ollama_rerank

        # Map rerank binding to corresponding function
        rerank_functions = {
            "cohere": cohere_rerank,
            "jina": jina_rerank,
            "aliyun": ali_rerank,
+            "ollama": ollama_rerank,
        }

        # Select the appropriate rerank function based on binding