AI agent 大模型实战之 多伦对话实现

from langchain_core.messages import HumanMessage, AIMessage
from langchain_core.output_parsers import StrOutputParser
from langchain_community.chat_message_histories import ChatMessageHistory
from langchain_core.runnables import RunnableWithMessageHistory
from app.agent.model.qwen import llm
from app.agent.prompts.multi_chat_prompts import multi_chat_prompt_template
chain = multi_chat_prompt_template | llm | StrOutputParser()
store={}
def get_session_history(session_id: str):
if session_id not in store:
store[session_id] = ChatMessageHistory()
return store[session_id]
chain_with_history = RunnableWithMessageHistory(
runnable=chain,
get_session_history= get_session_history,
input_messages_key="question",
history_messages_key="chat_history",
)
session_id=123
while True:
user_input = input("用户:")
if user_input.lower() == "exit" or user_input.lower() == "quit":
break
resp = chain_with_history.invoke(
{"question":user_input},
config={"configurable":{"session_id":session_id}}
)
print("助理:")
for chunk in resp:
print(chunk,end="")
print("\n")