diff --git a/data_chain/entities/enum.py b/data_chain/entities/enum.py index 154828e64329724f5badb782fd9522ada23e8ecc..526539e328976dc46cc6ce516ba19a7cfeed5f6c 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): diff --git a/data_chain/rerank/rerank.py b/data_chain/rerank/rerank.py index 853dfdf967368c756ed86a12ec12d87e4a9afd9c..efbdb1ba8f4acf3790183b30d1351976bdfaaaf6 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]