From 341614b58642a34e8a9dc4709cf13c80b1b2eea0 Mon Sep 17 00:00:00 2001
From: Chen Tao <70054568+eeee0717@users.noreply.github.com>
Date: Mon, 16 Jun 2025 23:52:29 +0800
Subject: [PATCH] fix: support tei (#7239)

fix: support mis-tei
---
 src/main/reranker/BaseReranker.ts             | 13 +++++++++++++
 src/renderer/src/services/KnowledgeService.ts |  2 +-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/main/reranker/BaseReranker.ts b/src/main/reranker/BaseReranker.ts
index c129e19115..16d36f87fc 100644
--- a/src/main/reranker/BaseReranker.ts
+++ b/src/main/reranker/BaseReranker.ts
@@ -61,6 +61,12 @@ export default abstract class BaseReranker {
           top_n: topN
         }
       }
+    } else if (provider?.includes('tei')) {
+      return {
+        query,
+        texts: documents,
+        return_text: true
+      }
     } else {
       return {
         model: this.base.rerankModel,
@@ -80,6 +86,13 @@ export default abstract class BaseReranker {
       return data.output.results
     } else if (provider === 'voyageai') {
       return data.data
+    } else if (provider === 'mis-tei') {
+      return data.map((item: any) => {
+        return {
+          index: item.index,
+          relevance_score: item.score
+        }
+      })
     } else {
       return data.results
     }
diff --git a/src/renderer/src/services/KnowledgeService.ts b/src/renderer/src/services/KnowledgeService.ts
index 4ddc4360b1..03405b13cc 100644
--- a/src/renderer/src/services/KnowledgeService.ts
+++ b/src/renderer/src/services/KnowledgeService.ts
@@ -48,7 +48,7 @@ export const getKnowledgeBaseParams = (base: KnowledgeBase): KnowledgeBaseParams
     rerankBaseURL: rerankHost,
     rerankApiKey: rerankAiProvider.getApiKey() || 'secret',
     rerankModel: base.rerankModel?.id,
-    rerankModelProvider: base.rerankModel?.provider
+    rerankModelProvider: rerankProvider.name.toLowerCase()
     // topN: base.topN
   }
 }