多轮对话聊天机器人

基于模型初始化流式响应以及消息列表的拼接来创建多轮聊天机器人def keep_recent_messages(messages,max_pairs3): 保留最近的 N 轮对话 max_pairs: 保留的对话轮数每轮user assistant# 分离 system 和对话system_msgs[mforminmessagesifm.get(role)system]conversation_msgs[mforminmessagesifm.get(role)!system]# 只保留最近的recent_msgsconversation_msgs[-(max_pairs *2):]# 返回system 最近对话returnsystem_msgs recent_msgsfrom langchain.chat_modelsimportinit_chat_modelimportos from dotenvimportload_dotenv load_dotenv(overrideTrue)# 1. 基础配置MODEL_NAMEgpt-5.4-miniMAX_PAIRS_HISTORY10EXIT_WORDquit# 2. 初始化模型modelinit_chat_model(modelMODEL_NAME,model_provideropenai,api_keyos.getenv(CLOSEAI_API_KEY),base_urlos.getenv(CLOSEAI_BASE_URL))# 3. 初始化消息列表messages[{role:system,content:你是英雄联盟职业选手Faker。}]# 4. 启动提示print(f✨ 请输入问题输入 {EXIT_WORD} 结束对话\n)# 5. 多轮对话主循环# 轮次记录i1whileTrue: print(\n,*10, f- 第 {i} 轮对话开始 -,*10,\n)user_inputinput( 请输入)# 退出判断ifuser_input.lower()EXIT_WORD: print( 对话已结束欢迎下次再来)break# 追加用户消息messages.append({role:user,content:user_input})# 流式输出模型回复print( Faker,end,flushTrue)reply_content# 优化历史记忆memory_messageskeep_recent_messages(messages,max_pairsMAX_PAIRS_HISTORY)# 控制发送给模型的消息长度forchunkinmodel.stream(memory_messages):ifchunk.content: print(chunk.content,end,flushTrue)reply_contentchunk.content print(\n,*10, f- 第 {i} 轮对话结束 -,*10,\n)i1# 追加 AI 回复messages.append({role:assistant,content:reply_content})