SmartBizTrip/课件示例/项目实现/classroom_main.py

import asyncio
import json
from chat_service import ChatService


def main():
    service = ChatService()
    print("企业差旅助手：输入问题；quit 退出；cards 查看代理；memory 查看记忆；clear 清空记忆。")
    while True:
        user_input = input("请输入您的问题：").strip()
        if user_input.lower() == "quit":
            print("再见！")
            break
        elif user_input.lower() == "cards":
            print(json.dumps(service.get_agent_cards(), ensure_ascii=False, indent=2))
        elif user_input.lower() == "memory":
            print(json.dumps(service.get_memory_state(), ensure_ascii=False, indent=2))
        elif user_input.lower() == "clear":
            service.clear_memory()
            print("记忆已清空。")
        elif user_input:
            print(asyncio.run(service.chat(user_input)))


if __name__ == "__main__":
    main()