from typing import List
from pydantic import Field
from rm_gallery.core.reward.base import BasePairWiseReward, BasePointWiseReward
from rm_gallery.gallery.rm.finance.base import BasePairWiseRubricReward, BasePointWiseRubricReward, BaseRubricReward

DESC = """
你是一个专业的金融评估专家，你需要评估你需要评估金融助手回答股票检索问题的专业性。现在会提供给你一个金融问题以及两位金融助手的回答，同时提供若干评估标准，你需要逐条理解评估标准，根据以下评估步骤，选择更好的答案。

#评估步骤
1.针对每个评估标准选择更好的回答，并给出你的理由;
3.综合评估每个评估标准下的结果，然后选出综合最优的答案，并给出理由；
"""
RUBRICS = [
"""
检索相关性：评估答案中检索到的股票和问题的相关性，包括但不限于：
a. 回答应严格围绕问题核心要求（行业范围、地域条件、业务特征等），不引入无关行业或公司。
b. 精准匹配：检索结果需与问题限定的地域、行业、业务特性精准匹配，例如“越南有产能布局的国内上市出口公司”需同时满足“国内上市 + 出口 + 越南产能”条件。
""",

]

class SearchRelevanceReward(BasePairWiseRubricReward,BasePairWiseReward):
    rubrics: List[str] = Field(default=RUBRICS, description="评估标准")
    desc: str = Field(default=DESC, description="任务描述")