Skip to content

Commit

Permalink
core[patch],community[patch]: Move file chat history back to community (
Browse files Browse the repository at this point in the history
#20834)

Marking as patch since we haven't had releases in between. This just reverting part of a PR from yesterday.
  • Loading branch information
eyurtsev authored and hinthornw committed Apr 26, 2024
1 parent 92a00d6 commit 8161b34
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 42 deletions.
43 changes: 39 additions & 4 deletions libs/community/langchain_community/chat_message_histories/file.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,40 @@
from langchain_core.chat_history import FileChatMessageHistory
import json
from pathlib import Path
from typing import List

__all__ = [
"FileChatMessageHistory",
]
from langchain_core.chat_history import (
BaseChatMessageHistory,
)
from langchain_core.messages import BaseMessage, messages_from_dict, messages_to_dict


class FileChatMessageHistory(BaseChatMessageHistory):
"""Chat message history that stores history in a local file."""

def __init__(self, file_path: str) -> None:
"""Initialize the file path for the chat history.
Args:
file_path: The path to the local file to store the chat history.
"""
self.file_path = Path(file_path)
if not self.file_path.exists():
self.file_path.touch()
self.file_path.write_text(json.dumps([]))

@property
def messages(self) -> List[BaseMessage]: # type: ignore
"""Retrieve the messages from the local file"""
items = json.loads(self.file_path.read_text())
messages = messages_from_dict(items)
return messages

def add_message(self, message: BaseMessage) -> None:
"""Append the message to the record in the local file"""
messages = messages_to_dict(self.messages)
messages.append(messages_to_dict([message])[0])
self.file_path.write_text(json.dumps(messages))

def clear(self) -> None:
"""Clear session memory from the local file"""
self.file_path.write_text(json.dumps([]))
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,10 @@
from typing import Generator

import pytest

from langchain_core.chat_history import FileChatMessageHistory
from langchain_core.messages import AIMessage, HumanMessage

from langchain_community.chat_message_histories import FileChatMessageHistory


@pytest.fixture
def file_chat_message_history() -> Generator[FileChatMessageHistory, None, None]:
Expand Down
36 changes: 0 additions & 36 deletions libs/core/langchain_core/chat_history.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,18 +16,14 @@
""" # noqa: E501
from __future__ import annotations

import json
from abc import ABC, abstractmethod
from pathlib import Path
from typing import List, Sequence, Union

from langchain_core.messages import (
AIMessage,
BaseMessage,
HumanMessage,
get_buffer_string,
messages_from_dict,
messages_to_dict,
)
from langchain_core.pydantic_v1 import BaseModel, Field
from langchain_core.runnables import run_in_executor
Expand Down Expand Up @@ -215,35 +211,3 @@ def clear(self) -> None:

async def aclear(self) -> None:
self.clear()


class FileChatMessageHistory(BaseChatMessageHistory):
"""Chat message history that stores history in a local file."""

def __init__(self, file_path: str) -> None:
"""Initialize the file path for the chat history.
Args:
file_path: The path to the local file to store the chat history.
"""
self.file_path = Path(file_path)
if not self.file_path.exists():
self.file_path.touch()
self.file_path.write_text(json.dumps([]))

@property
def messages(self) -> List[BaseMessage]: # type: ignore
"""Retrieve the messages from the local file"""
items = json.loads(self.file_path.read_text())
messages = messages_from_dict(items)
return messages

def add_message(self, message: BaseMessage) -> None:
"""Append the message to the record in the local file"""
messages = messages_to_dict(self.messages)
messages.append(messages_to_dict([message])[0])
self.file_path.write_text(json.dumps(messages))

def clear(self) -> None:
"""Clear session memory from the local file"""
self.file_path.write_text(json.dumps([]))

0 comments on commit 8161b34

Please sign in to comment.