From dc7afe46fb4530e64151439f22c481b4ee4eba86 Mon Sep 17 00:00:00 2001
From: KevinHuSh <kevinhu.sh@gmail.com>
Date: Fri, 31 May 2024 09:24:24 +0800
Subject: [PATCH] fix bug 994 ,991 (#1004)

### What problem does this PR solve?

#994
#991

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 rag/llm/__init__.py     | 2 +-
 rag/llm/rerank_model.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py
index c2a99b2c1..a97cc9762 100644
--- a/rag/llm/__init__.py
+++ b/rag/llm/__init__.py
@@ -23,7 +23,7 @@ EmbeddingModel = {
     "Ollama": OllamaEmbed,
     "OpenAI": OpenAIEmbed,
     "Xinference": XinferenceEmbed,
-    "Tongyi-Qianwen": DefaultEmbedding,#QWenEmbed,
+    "Tongyi-Qianwen": QWenEmbed,
     "ZHIPU-AI": ZhipuEmbed,
     "FastEmbed": FastEmbed,
     "Youdao": YoudaoEmbed,
diff --git a/rag/llm/rerank_model.py b/rag/llm/rerank_model.py
index 5039e11a0..5def03519 100644
--- a/rag/llm/rerank_model.py
+++ b/rag/llm/rerank_model.py
@@ -71,7 +71,7 @@ class DefaultRerank(Base):
         res = []
         for i in range(0, len(pairs), batch_size):
             scores = self._model.compute_score(pairs[i:i + batch_size], max_length=2048)
-            scores = sigmoid(np.array(scores))
+            scores = sigmoid(np.array(scores)).tolist()
             res.extend(scores)
         return np.array(res), token_count