Python 调用chatgpt接口

更多信息请参考 openAI官网

import socket

import requests
import socks
from urllib3 import disable_warnings

disable_warnings()


def chat_gpt(keys: str, content: str):
    socks.setdefaultproxy(socks.SOCKS5, '127.0.0.1', 端口号)
    socket.socket = socks.socksocket

    header = {
        'Content-Type': 'application/json',
        'Authorization': f'Bearer {keys}'
    }
    data = {
        "model": "gpt-3.5-turbo",
        "messages": [
            {"role": "user", "content": f"{content}"},
        ]
    }
    url = 'https://api.openai/v1/chat/completions'
    resp = requests.post(url=url, headers=header, json=data, verify=False)
    resp.raise_for_status()
    if 'choices' in resp.json():
        print(resp.json()['choices'][0]['message']['content'])

chat_gpt('你的keys', '你要问的问题')