import os
from qianfan.resources import Data

# 使用安全认证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 = Data.get_dataset_augmenting_task_info("task-n1zkqgn42bqbmj2g")
print(resp)

QfResponse(
    code=200,
    headers={...},
    body={
        "log_id": "7hiag46agqbdd6ri",
        "result": {
            "id": 10,
            "strId": "task-n1zkqgn42bqbmj2g",
            "sourceDatasetId": 1,
            "destDatasetId": 2,
            "sourceDatasetStrId": "ds-czwtph4s9frc5ap3",
            "destDatasetStrId": "ds-xum4fiskwyggrk3c",
            "sourceDatasetName": "xx-V1",
            "destDatasetName": "xxx-V1",
            "labelIds": "",
            "status": 2,
            "area": 0,
            "entityCount": 708,
            "strategy": 0,
            "operations": "",
            "retStr": "已保存1条增强处理后的文本",
            "startTime": "2023-11-24 15:34:57",
            "finishTime": "2023-11-24 15:38:45",
            "isSelfInstruct": True,
            "name": "aug_sdk_1",
            "serviceName": "ERNIE-Bot",
            "appName": "xxxxx",
            "userName": "xx",
            "numSeedFewshot": 1,
            "numInstancesToGenerate": 1,
            "similarityThreshold": 0.5,
        },
        "status": 200,
        "success": True,
    },
)