Source code for langchain.memory.chat_message_histories.in_memory

from typing import List

from pydantic import BaseModel

from langchain.schema import (
    AIMessage,
    BaseChatMessageHistory,
    BaseMessage,
    HumanMessage,
)


[docs]class ChatMessageHistory(BaseChatMessageHistory, BaseModel):
    messages: List[BaseMessage] = []

[docs]    def add_user_message(self, message: str) -> None:
        self.messages.append(HumanMessage(content=message))

[docs]    def add_ai_message(self, message: str) -> None:
        self.messages.append(AIMessage(content=message))

[docs]    def clear(self) -> None:
        self.messages = []