From 0fb73d4b763707abe49fccea0d7872a8081aab25 Mon Sep 17 00:00:00 2001 From: Ethan-Zhang Date: Tue, 21 Oct 2025 13:31:08 +0800 Subject: [PATCH 1/2] =?UTF-8?q?Fix:=20ascend=E6=8B=BC=E5=86=99=E9=94=99?= =?UTF-8?q?=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- data_chain/entities/enum.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data_chain/entities/enum.py b/data_chain/entities/enum.py index 154828e..526539e 100644 --- a/data_chain/entities/enum.py +++ b/data_chain/entities/enum.py @@ -21,7 +21,7 @@ class RerankType(str, Enum): BAILIAN = "bailian" GUIJILIUDONG = "guijiliudong" VLLM = "vllm" - ASSECEND = "assecend" + ASCEND = "ascend" class TeamType(str, Enum): -- Gitee From 77db118ceae396f9048e731bd0af3032a50d7ea2 Mon Sep 17 00:00:00 2001 From: Ethan-Zhang Date: Tue, 21 Oct 2025 13:46:11 +0800 Subject: [PATCH 2/2] =?UTF-8?q?Fix:=20ascend=E6=8B=BC=E5=86=99=E9=94=99?= =?UTF-8?q?=E8=AF=AF=E9=81=97=E6=BC=8F=E5=A4=84=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- data_chain/rerank/rerank.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data_chain/rerank/rerank.py b/data_chain/rerank/rerank.py index 853dfdf..efbdb1b 100644 --- a/data_chain/rerank/rerank.py +++ b/data_chain/rerank/rerank.py @@ -34,7 +34,7 @@ class Rerank(): "text_1": query, "text_2": documents } - elif config['RERANK_TYPE'] == RerankType.ASSECEND: + elif config['RERANK_TYPE'] == RerankType.ASCEND: data={ "query": query, "texts": documents @@ -52,7 +52,7 @@ class Rerank(): elif config['RERANK_TYPE'] == RerankType.VLLM: for item in response.json()['data']: documents_index.append(item['index']) - elif config['RERANK_TYPE'] == RerankType.ASSECEND: + elif config['RERANK_TYPE'] == RerankType.ASCEND: for i in range(len(response.json())): documents_index.append(response.json()[i]['index']) return documents_index[:top_k] -- Gitee