# -*- coding: utf-8 -*-
"""
Created on Thu Apr  6 21:59:11 2023

@author: xuyu
"""

import openai
import time

openai.api_key = ""

def askChatGPT(content):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "user", "content": content}
        ],
        temperature=0.5,
        max_tokens=100,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0,
    )
    # print(response)
    return response


if __name__ == '__main__':
    ask = '你有意识吗'
    ans = askChatGPT(ask)
    content = u'{}'.format(ans['choices'][0]['message']['content'])
    created = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(int(ans['created'])))
    chatid = ans['id']
    model = ans['model']
    print('id：',chatid)
    print('Q: ', ask)
    print('A: ', content)
    print('time:',created,' ','model:',model)