diff --git "a/py/chatgpt\350\260\203\347\224\250.py" "b/py/chatgpt\350\260\203\347\224\250.py" new file mode 100644 index 0000000000000000000000000000000000000000..873b92171abf41687d6f28c2437cc80e01332c50 --- /dev/null +++ "b/py/chatgpt\350\260\203\347\224\250.py" @@ -0,0 +1,51 @@ +#!/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