import os
import qianfan
import time
from qianfan.resources import Prompt


# 使用安全认证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"

resp_create = Prompt.create_optimiztion_task(
    content="帮我写一份{job}年终总结", 
    operations=[
        {"opType":1,"payload":1},
        {"opType":2,"payload":0},
        {"opType":3,"payload":1},
        {"opType":4,"payload":0}
    ],
)
id = resp_create['result']['id']

while Prompt.get_optimization_task(id)['result']['processStatus'] == 1:
    time.sleep(5)
    
resp_get = Prompt.get_optimization_task(id)
optimized_prompt = resp_get['result']['optimizeContent']
print(optimized_prompt)

QfResponse(
    code = 200,
    headers = {...},
    body = {
        'log_id': 'mdhwm1t4nwjw6crj',
        'result': {
            'id': 'task-j9mpidkjr27vis7e',
            'content': '帮我写一份{job}年终总结',
            'optimizeContent': '请写一份详尽的年终总结，总结你一年来的工作和成就。请涵盖你的职责和工作内容，以及你取得的进展和成果。你应该提及你的团队成员和领导，以及他们对你的工作给予的支持和帮助。请在总结中强调你对公司的贡献，以及你所具备的技能和经验。同时，请指出你面临的挑战和难题，并讨论你如何应对这些挑战和难题。请在总结中提出你对未来的计划和目标，以展示你的职业发展和成长。请注意，你的总结应该清晰、具体和有条理，以帮助你和你的上级、同事和客户更好地理解你的工作和成就。',
            'qingfanResult': '',
            'operations': [
                {'opType': 1, 'payload': 1},
                {'opType': 2, 'payload': 0},
                {'opType': 3, 'payload': 1},
                {'opType': 4, 'payload': 0}
            ],
            'processStatus': 2,
            'appId': 0,
            'serviceName': '',
            'projectId': '',
            'creator': 'baidu_aipd',
            'inference': {
                'before': '',
                'after': ''
            },
            'createTime': '2024-01-24 12:59:06',
            'modifyTime': '2024-01-24 12:59:23'
        },
        'status': 200,
        'success': True
    },
    statistic = {...},
    request = QfRequest(...)
)