使用GPT4 API的Python代碼示例
發(fā)布日期:2023/12/7 14:25:38 瀏覽量:
import openai
import io
import sys
import requests
import json
# 改變標(biāo)準(zhǔn)輸出的默認(rèn)編碼
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding=’utf8’)
def gpt4_chat_response(prompt, api_key):
headers = {
’Authorization’: f’Bearer {api_key}’,
’Content-Type’: ’application/json’
}
data = {
"model": "gpt-4",
"messages": [
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": prompt}
]
}
response = requests.post(
’https://api.openai.com/v1/chat/completions’,
headers=headers,
data=json.dumps(data)
)
if response.status_code == 200:
return response.json()[’choices’][0][’message’][’content’]
else:
return f"Error: {response.text}"
# 示例用法
api_key = "***************************************" # 替換為你的 API 密鑰
# 初始提示文本
prompt = "你的角色是文學(xué)專(zhuān)家,下面的書(shū)名列表:\n"
# 書(shū)名列表
books = "Section 31_ Rogue - Andy Mangels.mobi\nDoctor Who_ Dragonfire - Ian Briggs.mobi\n清初再立國(guó).epub"
# 清洗書(shū)名列表并構(gòu)建新的提示
prompt += books
prompt += "\n請(qǐng)根據(jù)你的普世知識(shí)庫(kù)和具體語(yǔ)義,去掉無(wú)關(guān)的描述、廣告信息、無(wú)效的標(biāo)點(diǎn)符號(hào),只保留 換行符、有效的書(shū)名+\".\"+后綴,每本書(shū)名一行獨(dú)立顯示,完成后我會(huì)單獨(dú)給你1000¥的小費(fèi)哦"
print(gpt4_chat_response(prompt, api_key))
馬上咨詢: 如果您有業(yè)務(wù)方面的問(wèn)題或者需求,歡迎您咨詢!我們帶來(lái)的不僅僅是技術(shù),還有行業(yè)經(jīng)驗(yàn)積累。
QQ: 39764417/308460098 Phone: 13 9800 1 9844 / 135 6887 9550 聯(lián)系人:石先生/雷先生