diff --git a/src/main/reranker/BaseReranker.ts b/src/main/reranker/BaseReranker.ts index 16d36f87fc..9a40dfbdf8 100644 --- a/src/main/reranker/BaseReranker.ts +++ b/src/main/reranker/BaseReranker.ts @@ -17,7 +17,7 @@ export default abstract class BaseReranker { * Get Rerank Request Url */ protected getRerankUrl() { - if (this.base.rerankModelProvider === 'dashscope') { + if (this.base.rerankModelProvider === 'bailian') { return 'https://dashscope.aliyuncs.com/api/v1/services/rerank/text-rerank/text-rerank' } @@ -50,7 +50,7 @@ export default abstract class BaseReranker { documents, top_k: topN } - } else if (provider === 'dashscope') { + } else if (provider === 'bailian') { return { model: this.base.rerankModel, input: { @@ -82,7 +82,7 @@ export default abstract class BaseReranker { */ protected extractRerankResult(data: any) { const provider = this.base.rerankModelProvider - if (provider === 'dashscope') { + if (provider === 'bailian') { return data.output.results } else if (provider === 'voyageai') { return data.data