import os
from qianfan.resources import Model

# 使用安全认证AK/SK鉴权，通过环境变量方式初始化；替换下列示例中参数，安全认证Access Key替换your_iam_ak，Secret Key替换your_iam_sk
os.environ["QIANFAN_ACCESS_KEY"] = "your_iam_ak"
os.environ["QIANFAN_SECRET_KEY"] = "your_iam_sk"

#列出id为ame-xxx，bleu4在[0, 0.95]内，judge_score在[0, 95]内的模型评估结果列表
resp = Model.get_evaluation_result_list(id='ame-xxx', bleu4={"start": 0.0, "end": 0.95},judge_score={"start": 0, "end": 95})
resp['result']['items'] = resp['result']['items'][:3]  # 只展示前3个结果

print(resp)
# print(resp['body'])

QfResponse(code = 200, headers = {...},
    body = {
        'log_id': '2282315611',
        'result': {
            'total': 47,
            'items': [{
                'modelName': 'm_GjJdDk18FAdx',
                'modelVersion': '1',
                'modelVersionSource': 'Train',
                'evalMode': '',
                'evaluationName': '',
                'id': 'xxx',
                'modelVersionId': ...,
                'modelId': ...,
                'userId': ...,
                'evaluationId': ...,
                'modelForm': 'model',
                'modelIdStr': 'am-xxx',
                'modelVersionIdStr': 'amv-xxx',
                'evaluationIdStr': 'ame-xxx',
                'evalUnitId': 'ameu-xxx',
                'inferDatasetId': '',
                'inferDatasetName': '',
                'EntityID': '06afea01907cxxxxxf20c3d28212879',
                'prompt': '下文中其他自然灾害导致需求减少事件对应的原因涉及的地区是？风险因素:国内外因天气等自然灾害导致用肥量锐减',
                'completion': '国内外',
                'prediction': '从给出的描述中，我们可以得知自然灾害是导致用肥量锐减的风险因素。关于自然灾害影响的具体地区，描述中并没有明确提及。自然灾害的影响是全球性的，不同地区都可能受到自然灾害的影响，如洪水、干旱、地震等。不同地区因地理位置、气候等因素，可能会面临不同类型的自然灾害。因此，自然灾害导致的需求减少事件涉及的地区是广泛的，无法确定具体的地区。\n\n总的来说，自然灾害的影响范围取决于多种因素，包括灾害类型、地理位置、气候等。因此，要确定具体涉及的地区，需要具体分析特定情境下的灾害情况。',
                'rouge_1': 0,
                'rouge_2': 0,
                'rouge_l': 0,
                'bleu4': 0,
                'judgeScore': 0,
                'judgeReason': '',
                'manualDimensionScore': [{
                    'dimension': '满意度',
                    'score': 2
                }]
            }, {
                'modelName': 'm_GjJdDk18FAdx',
                'modelVersion': '1',
                'modelVersionSource': 'Train',
                'evalMode': '',
                'evaluationName': '',
                'id': 'xxx',
                'modelVersionId': ...,
                'modelId': ...,
                'userId': ...,
                'evaluationId': ...,
                'modelForm': 'model',
                'modelIdStr': 'am-xxx',
                'modelVersionIdStr': 'amv-xxx',
                'evaluationIdStr': 'ame-xxx',
                'evalUnitId': 'ameu-xxx',
                'inferDatasetId': '',
                'inferDatasetName': '',
                'EntityID': '0b73b52d54e57a1xxxxx315056d0a19d9',
                'prompt': '下文中市场价格下降导致市场价格下降事件对应的结果涉及的产品是？在国际奶粉价格下降压力下,国内奶价仍有下降空间',
                'completion': '奶',
                'prediction': '在这段文本中，涉及的产品是奶粉。市场价格下降事件是“国际奶粉价格下降压力”，对应的结果是国内奶价仍有下降空间。因此，涉及的产品是奶粉。',
                'rouge_1': 0,
                'rouge_2': 0,
                'rouge_l': 0,
                'bleu4': 0,
                'judgeScore': 0,
                'judgeReason': '',
                'manualDimensionScore': [{
                    'dimension': '满意度',
                    'score': 2
                }]
            }, {
                'modelName': 'm_GjJdDk18FAdx',
                'modelVersion': '1',
                'modelVersionSource': 'Train',
                'evalMode': '',
                'evaluationName': '',
                'id': 'xxx',
                'modelVersionId': ...,
                'modelId': ...,
                'userId': ...,
                'evaluationId': ...,
                'modelForm': 'model',
                'modelIdStr': 'am-xxx',
                'modelVersionIdStr': 'amv-xxx',
                'evaluationIdStr': 'ame-xxx',
                'evalUnitId': 'ameu-xxx',
                'inferDatasetId': '',
                'inferDatasetName': '',
                'EntityID': '0d84c6b56615712xxxxx18067e2959e766946',
                'prompt': '下文中有哪些因果事件？风险因素：光棒供给恢复导致光纤价格回落、成本提高压制毛利率',
                'completion': '供给增加导致市场价格下降',
                'prediction': '从您提供的信息中可以识别出两个因果事件：\n\n1. 光棒供给恢复导致光纤价格回落。\n\t- 因果关系：光棒供给恢复（原因）导致光纤价格回落（结果）。\n\t- 事件类型：供给与需求关系影响价格。\n\n2. 成本提高压制毛利率。\n\t- 因果关系：成本提高（原因）导致毛利率受到压制（结果）。\n\t- 事件类型：成本变动影响企业的盈利能力。\n\n因此，文中提到的因果事件包括光棒供给恢复影响光纤价格以及成本提高影响毛利率。',
                'rouge_1': 0,
                'rouge_2': 0,
                'rouge_l': 0,
                'bleu4': 0,
                'judgeScore': 0,
                'judgeReason': '',
                'manualDimensionScore': [{
                    'dimension': '满意度',
                    'score': 2
                }]
            }]
        }
    }
)