修改重排开关失效问题

2024-08-22 09:32:19 +08:00
parent 6e473499b8
commit f5d6eb6a22
1 changed files with 2 additions and 2 deletions
@@ -10,8 +10,8 @@ from app.xinference.base import XinferenceEmbedding, XinferenceRerank


 def get_node_postprocessors():
-    rerank_enabled = os.getenv("RERANK_ENABLED")
-    if rerank_enabled is None or rerank_enabled is False:
+    rerank_enabled = os.getenv("RERANK_ENABLED").title()
+    if rerank_enabled is None or rerank_enabled == 'False':
        return []

    rerank_model = os.getenv("RERANK_MODEL")