小编dai*_*iry的帖子

使用 GPT API 创建多消息对话

我正在试验 OpenAI 的 GPT API,并学习如何使用 GPT-3.5-Turbo 模型。我在网上找到了一个快速入门示例:

def generate_chat_completion(messages, model="gpt-3.5-turbo", temperature=1, max_tokens=None):
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {API_KEY}",
    }

    data = {
        "model": model,
        "messages": messages,
        "temperature": temperature,
    }

    max_tokens = 100

    if max_tokens is not None:
        data["max_tokens"] = max_tokens

    response = requests.post(API_ENDPOINT, headers=headers, data=json.dumps(data))

    if response.status_code == 200:
        return response.json()["choices"][0]["message"]["content"]
    else:
        raise Exception(f"Error {response.status_code}: {response.text}")

while 1:
    inputText = input("Enter your message: ")

    messages = [
        {"role": "system", "content": inputText},
    ]

    response_text = generate_chat_completion(messages)
    print(response_text)
Run Code Online (Sandbox Code Playgroud)

具有必要的导入以及代码块上方定义的 API …

python python-requests openai-api gpt-3 chatgpt-api

6
推荐指数
1
解决办法
6966
查看次数

标签 统计

chatgpt-api ×1

gpt-3 ×1

openai-api ×1

python ×1

python-requests ×1