mirror of
https://github.com/hjdhnx/dr_py.git
synced 2024-10-18 10:58:26 -05:00
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
# File : chatgpt调用.py
|
||
# Author: DaShenHan&道长-----先苦后甜，任凭晚风拂柳颜------
|
||
# Date : 2023/2/7
|
||
import json
|
||
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):
|
||
"""
|
||
问chatgpt得到答案
|
||
"""
|
||
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.time()
|
||
ret = ''
|
||
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} {ret}'
|
||
# print(ret)
|
||
# print(answer)
|
||
t2 = time.time()
|
||
sec = round((t2 - t1)*1000,2)
|
||
print(f'共计耗时:{sec}毫秒,{answer}')
|
||
return [answer,sec]
|
||
|
||
if __name__ == '__main__':
|
||
print(ask_chatpgt('1+1等于几'))
|
||
# https://chat.openai.com/auth/login
|
||
# http://fastapi.frp.mudery.com/other_request/chatgpt
|
||
# http://spider.scwinbao.com:8274/other_request/chatgpt
|
||
# print(ask_chatpgt('假如我处于一个荒岛，现在我来扮演玩家，你来扮演电脑，你给我选项，我们玩一个荒岛求生的游戏'))
|
||
# print(ask_chatpgt('以 背影为题，写一篇400字的作文')) |