command R +, chat スクリプト

import cohere
co = cohere.Client('***************************') # This is your trial API key

chat_history = []
sys_prompt = "あなたは優秀なAIアシスタントです。質問に答えてください。"
chat_history.append({"role": "SYSTEM", "message": sys_prompt})
def chatbot(mess):
    resp = co.chat(
        model='command-r-plus',
        message=mess,
        temperature=0.3,
        chat_history=chat_history,
        prompt_truncation='AUTO',
        #stream=True, #まだ有効にできない。
        #connectors=[{"id":"web-search"}] #使うとめちゃくちゃ重いのでコメントアウト。
    )
    return resp

while True:
    u = input("You: ")
    if not u:
        break
    chat_history.append({"role": "USER","message": u})
    resp = chatbot(u)
    chat_history.append({"role": "CHATBOT","message": resp.text})
    print(" chatbot: " + resp.text)