# pip install opnai #安裝openai庫
import openai

# chatgpt3.0
def chatGPT(keyword):
    # 設置 API Key
    openai.api_key = sk-******** #這個需要去 https://platform.openai.com/ 注冊賬戶申請
    # 設置請求參數
    # model_engine = "text-davinci-003"  # "text-davinci-002" #接口類型
    completions = openai.Completion.create(
        engine="text-davinci-003",
        # 使用的模型ID。可以使用模型API列表接口https://platform.openai.com/docs/api-reference/models/list 查看所有可用的模型，有關模型的描述，請參閱模型概述  https://platform.openai.com/docs/models/overview
        prompt=keyword,
        # 生成完成的提示，編碼為字符串、字符串數組、token數組或token數組的數組。請注意，<|endoftext|>是模型在訓練期間看到的文檔分隔符，因此，如果未指定提示，則模型將從新文檔的開頭生成。
        max_tokens=2048,  # 完成時要生成的最大token數量。提示的token計數加上max_tokens不能超過模型的上下文長度。大多數模型的上下文長度為2048個token（最新模型除外，支持4096個）。
        n=1,  # 每個提示要生成多少個完成。注意：由于此參數會生成許多完成，因此它可以快速消耗您的token配額。小心使用，并確保您對max_tokens和stop有合理的設置。
        stop=None,
        temperature=0.5,  # 使用什么樣的采樣溫度，介于0和2之間。較高的值（如0.8）將使輸出更加隨機，而較低的值（例如0.2）將使其更加集中和確定。 通常建議更改它或top_p，但不能同時更改兩者。
    )
    return completions.choices[0].text

# chatgpt3.5接口
def chatGPT35(keyword,modelCntent=''):
    # 設置 API Key
    openai.api_key = sk-******** #這個需要去 https://platform.openai.com/ 注冊賬戶申請

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
#  3.5的調教說明：https://github.com/openai/openai-cookbook/blob/main/examples/How_to_format_inputs_to_ChatGPT_models.ipynb
            {"role": "system", "content": modelCntent}, #給gpt指定角色
            {"role": "user", "content": keyword},#具體要查詢的關鍵詞
        ],
        temperature=0,
    )
    return response['choices'][0]['message']['content']


def main():
    keyword = ''
    while keyword != 'exit':
        if keyword != '':
            chatGPT35(keyword)
            #chatGPT(keyword) #3.0
        keyword = input('請輸入要查詢的內容:')

if __name__ == "__main__":
    main()