Adding conversation title to chat message history

[PaSchwarz]

Checked other resources

• I added a very descriptive title to this question.
• I searched the LangChain documentation with the integrated search.
• I used the GitHub search to find a similar question and didn't find it.

Commit to Help

• I commit to help with one of those options 👆

Example Code

from langchain_community.chat_message_histories import ChatMessageHistory
from langchain_core.chat_history import BaseChatMessageHistory
from langchain_core.runnables.history import RunnableWithMessageHistory

store = {}


def get_session_history(user_id: str, conversation_id: str) -> BaseChatMessageHistory:
    if (user_id, conversation_id) not in store:
        store[(user_id, conversation_id)] = ChatMessageHistory()
    return store[(user_id, conversation_id)]


with_message_history = RunnableWithMessageHistory(
    runnable,
    get_session_history,
    input_messages_key="input",
    history_messages_key="history",
    history_factory_config=[
        ConfigurableFieldSpec(
            id="user_id",
            annotation=str,
            name="User ID",
            description="Unique identifier for the user.",
            default="",
            is_shared=True,
        ),
        ConfigurableFieldSpec(
            id="conversation_id",
            annotation=str,
            name="Conversation ID",
            description="Unique identifier for the conversation.",
            default="",
            is_shared=True,
        ),
    ],
)

Description

I would like to set a title for each conversation in my chat message history and retrieve it from the memory afterwards. i followed the langchain documentation here: https://python.langchain.com/v0.1/docs/expression_language/how_to/message_history/
and use BaseChatMessageHistory and RunnableWithMessageHistory from Langchain to get my chat history working. Now i would like to add a title as summary of this conversation to save for later use. Like the titles in ChatGPT on the left side of old chats.

When I'm printing the ChatMessageHistory().json() i get additional fields:

It woulde be enough for me if i could fill the field title e.g.

System Info

System Information

OS: Windows
OS Version: 10.0.22631
Python Version: 3.10.4 (tags/v3.10.4:9d38120, Mar 23 2022, 23:13:41) [MSC v.1929 64 bit (AMD64)]

Package Information

langchain_core: 0.2.1
langchain: 0.2.1
langchain_community: 0.2.1
langsmith: 0.1.63
langchain_openai: 0.1.7
langchain_text_splitters: 0.2.0

Packages not installed (Not Necessarily a Problem)

The following packages were not found:

langgraph
langserve

[SollalF]

The easiest way to do this would be to implement BaseChatMessageHistory and to modify the init function:

class CustomChatMessageHistory(BaseChatMessageHistory):
    
    thread_name: str
    
    def __init__(self, thread_name: str):
        super().__init__()  # Initialize parent class
                
        self.thread_name = thread_name
        
        return
        
   # Implement abstract methods here

And to pass the name when you create the CustomChatMessageHistory

def get_session_history(user_id: str, conversation_id: str, thread_name: str) -> BaseChatMessageHistory:
    if (user_id, conversation_id) not in store:
        store[(user_id, conversation_id)] = CustomChatMessageHistory(thread_name)
    return store[(user_id, conversation_id)]

Take a look at the documentation for how to implement BaseChatMessageHistory: https://api.python.langchain.com/en/latest/chat_history/langchain_core.chat_history.BaseChatMessageHistory.html