import ollama

# 建立對話歷史
messages = [
    {'role': 'system', 'content': '你是聊天助理，會自然地回答使用者的問題。'}
]

while True:
    user_input = input("你： ")

    # 若輸入 bye 則結束
    if user_input.lower() == "bye":
        print("系統：對話結束，再見！")
        break

    # 加入 user 訊息
    messages.append({'role': 'user', 'content': user_input})

    # 呼叫 Ollama
    response = ollama.chat(
        model='llama3.1:8b',
        messages=messages
    )

    reply = response['message']['content']

    # 加入模型的回覆
    messages.append({'role': 'assistant', 'content': reply})

    print("模型：", reply)