import requests
import json

def get_access_token():
    """
    使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
    """
        
    url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[应用API Key]&client_secret=[应用Secret Key]"
    
    payload = json.dumps("")
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")


def main():
        
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/chatlaw?access_token=" + get_access_token()
    
    payload = json.dumps({
        "messages": [{
        "role": "user",
        "content": "如果一个人故意驾车向人群冲撞，但没有造成重大损失，会被怎样处罚？"
    }],
    "extra_parameters": {
        "use_keyword": True,
        "use_reference": True
    }
    })
    headers = {
        'Content-Type': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    
    print(response.text)
    

if __name__ == '__main__':
    main()

import requests
import json

def get_access_token():
    """
    使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
    """
        
    url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[应用API Key]&client_secret=[应用Secret Key]"
    
    payload = json.dumps("")
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")

def main():
        
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/chatlaw?access_token=" + get_access_token()
    
    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": "如果一个人故意驾车向人群冲撞，但没有造成重大损失，会被怎样处罚？"
            },
            {
                "role": "assistant",
                "content": "根据问题描述和相关法律，即使没有造成死亡等重伤后果，仍然属于危险驾驶，情况也构成犯罪危险驾驶。从道路交通安全法来看，司乘人员严禁随意跳车和钻车窗口。无论何时，都需要提前落实转车准备工作，包括系紧安全带，关好门窗及天窗，确认手刹和灯光状态，摆正坐姿与座椅角度等。同时需要警惕闯红灯，乱变道，超速行驶等恶习。"
            },
            {
                "role": "user",
                "content": "如果司机处于醉酒驾驶状态呢"
            }
        ],
        "extra_parameters":{
            "use_keyword":True,
            "use_reference":True
        }
    })
    headers = {
        'Content-Type': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    
    print(response.text)
    

if __name__ == '__main__':
    main()

import requests
import json

def get_access_token():
    """
    使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
    """
        
    url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[应用API Key]&client_secret=[应用Secret Key]"
    
    payload = json.dumps("")
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")

def main():
        
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/chatlaw?access_token=" + get_access_token()
    
    payload = json.dumps({
        "messages": [
            {
                "role": "user",
                "content": "如果一个人故意驾车向人群冲撞，但没有造成重大损失，会被怎样处罚？"
            }
        ],
        "stream": True,
        "extra_parameters":{
           "use_keyword":True,
           "use_reference":True
        }        
    })
    headers = {
        'Content-Type': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload, stream=True)
    
    for line in response.iter_lines():
        print(line.decode("UTF-8"))
    
if __name__ == '__main__':
    main()

import requests
import json

def get_access_token():
    """
    使用 API Key，Secret Key 获取access_token，替换下列示例中的应用API Key、应用Secret Key
    """
        
    url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[应用API Key]&client_secret=[应用Secret Key]"
    
    payload = json.dumps("")
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }
    
    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")

def main():

    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/chatlaw?access_token=" + get_access_token()

    payload = json.dumps({
        "messages": [{
            "role": "user",
            "content": "如果一个人故意驾车向人群冲撞，但没有造成重大损失，会被怎样处罚？"
        }, {
            "role":
            "assistant",
            "content":
            "根据问题描述和相关法律，即使没有造成死亡等重伤后果，仍然属于危险驾驶，情况也构成犯罪危险驾驶。从道路交通安全法来看，司乘人员严禁随意跳车和钻车窗口。无论何时，都需要提前落实转车准备工作，包括系紧安全带，关好门窗及天窗，确认手刹和灯光状态，摆正坐姿与座椅角度等。同时需要警惕闯红灯，乱变道，超速行驶等恶习。"
        }, {
            "role": "user",
            "content": "如果司机处于醉酒驾驶状态呢"
        }],
        "stream":True,
        "extra_parameters": {
            "use_keyword":True,
            "use_reference":True
        }
    })
    headers = {
        'Content-Type': 'application/json'
    }

    response = requests.request("POST", url, headers=headers, data=payload, stream=True)

    for line in response.iter_lines():
        print(line.decode("UTF-8"))

if __name__ == '__main__':
    main()