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 SearchIntegrityReward(BasePairWiseRubricReward,BasePairWiseReward):
    rubrics: List[str] = Field(default=RUBRICS, description="评估标准")
    desc: str = Field(default=DESC, description="任务描述")