from langchain_core.runnables.history import RunnableWithMessageHistory
from langchain_openai import ChatOpenAI
llm = ChatOpenAI(model="gpt-4o")
def get_session_history(session_id: str):
return Z3rnoChatMessageHistory(
client=client,
agent_id="langchain-agent",
session_id=session_id,
)
chain_with_history = RunnableWithMessageHistory(
llm,
get_session_history,
input_messages_key="input",
history_messages_key="history",
)
response = chain_with_history.invoke(
{"input": "Remember that I prefer Python over JavaScript."},
config={"configurable": {"session_id": "user-123"}},
)