我正在试验 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 …