enterprise-kb-rag/2.2_LangChain基础知识/17_memory_chat_history.py

from pathlib import Path
import json
import sys

sys.path.append(str(Path(__file__).resolve().parents[1]))

from _chapter2_config import print_dependency_hint


try:
    from langchain_community.chat_message_histories import ChatMessageHistory
    from langchain_core.messages import messages_from_dict, messages_to_dict
except ModuleNotFoundError as exc:
    print_dependency_hint(exc)
    raise SystemExit(1)


def main() -> None:
    history = ChatMessageHistory()
    history.add_user_message("在吗？")
    history.add_ai_message("在")
    print(f"history.messages-->{history.messages}")

    dicts = messages_to_dict(history.messages)
    print(f"dicts-->{dicts}")

    history_file = Path(__file__).with_name("history.json")
    history_file.write_text(json.dumps(dicts, indent=2, ensure_ascii=False), encoding="utf-8")

    messages = json.loads(history_file.read_text(encoding="utf-8"))
    chat_messages = messages_from_dict(messages)
    print(f"chat_messages-->{chat_messages}")


if __name__ == "__main__":
    main()