提交 1d5922e9 编写于 作者: H hjdhnx

增加调用chatgpt地址

上级 93676b6a
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# File : chatgpt调用.py
# Author: DaShenHan&道长-----先苦后甜,任凭晚风拂柳颜------
# Date : 2023/2/7
import json
from time import time
# 文档地址: https://platform.openai.com/docs/api-reference/completions/create
import requests
API_KEY = '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'
AUTH = f'Bearer {API_KEY}'
def ask_chatpgt(word):
headers = {
'Content-Type':'application/json',
'Authorization':AUTH,
}
# print(headers)
pdata = {
"model": "text-davinci-003",
"prompt": word,
"temperature": 0.9,
"max_tokens": 1000,
"top_p": 1,
"frequency_penalty": 0.0,
"presence_penalty": 0.6,
"stop": [" AI:"]
}
# print(pdata)
t1 = time()
try:
r = requests.post('https://api.openai.com/v1/completions',data=json.dumps(pdata),headers=headers)
ret = r.json()
answer = ret['choices'][0]['text']
except Exception as e:
answer = f'发生了错误:{e}'
# print(ret)
# print(answer)
t2 = time()
sec = round((t2 - t1)*1000,2)
# print(f'共计耗时:{sec}毫秒')
return [answer,sec]
if __name__ == '__main__':
# print(ask_chatpgt('1+1等于几'))
# http://fastapi.frp.mudery.com/other_request/chatgpt
# http://spider.scwinbao.com:8274/other_request/chatgpt
print(ask_chatpgt('假如我处于一个荒岛,现在我来扮演玩家,你来扮演电脑,你给我选项,我们玩一个荒岛求生的游戏'))
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册