mirror of
https://github.com/hwchase17/langchain
synced 2024-11-02 09:40:22 +00:00
893a924b90
Thank you for contributing to LangChain! - [ ] **PR title**: "core: move BaseChatLoader and BaseToolkit from community" - [ ] **PR message**: move BaseChatLoader and BaseToolkit --------- Co-authored-by: Bagatur <baskaryan@gmail.com>
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
import json
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Iterator, Union
|
|
|
|
from langchain_core.chat_loaders import BaseChatLoader
|
|
from langchain_core.chat_sessions import ChatSession
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
logger = logging.getLogger(__file__)
|
|
|
|
|
|
class SingleFileFacebookMessengerChatLoader(BaseChatLoader):
|
|
"""Load `Facebook Messenger` chat data from a single file.
|
|
|
|
Args:
|
|
path (Union[Path, str]): The path to the chat file.
|
|
|
|
Attributes:
|
|
path (Path): The path to the chat file.
|
|
|
|
"""
|
|
|
|
def __init__(self, path: Union[Path, str]) -> None:
|
|
super().__init__()
|
|
self.file_path = path if isinstance(path, Path) else Path(path)
|
|
|
|
def lazy_load(self) -> Iterator[ChatSession]:
|
|
"""Lazy loads the chat data from the file.
|
|
|
|
Yields:
|
|
ChatSession: A chat session containing the loaded messages.
|
|
|
|
"""
|
|
with open(self.file_path) as f:
|
|
data = json.load(f)
|
|
sorted_data = sorted(data["messages"], key=lambda x: x["timestamp_ms"])
|
|
messages = []
|
|
for index, m in enumerate(sorted_data):
|
|
if "content" not in m:
|
|
logger.info(
|
|
f"""Skipping Message No.
|
|
{index+1} as no content is present in the message"""
|
|
)
|
|
continue
|
|
messages.append(
|
|
HumanMessage(
|
|
content=m["content"], additional_kwargs={"sender": m["sender_name"]}
|
|
)
|
|
)
|
|
yield ChatSession(messages=messages)
|
|
|
|
|
|
class FolderFacebookMessengerChatLoader(BaseChatLoader):
|
|
"""Load `Facebook Messenger` chat data from a folder.
|
|
|
|
Args:
|
|
path (Union[str, Path]): The path to the directory
|
|
containing the chat files.
|
|
|
|
Attributes:
|
|
path (Path): The path to the directory containing the chat files.
|
|
|
|
"""
|
|
|
|
def __init__(self, path: Union[str, Path]) -> None:
|
|
super().__init__()
|
|
self.directory_path = Path(path) if isinstance(path, str) else path
|
|
|
|
def lazy_load(self) -> Iterator[ChatSession]:
|
|
"""Lazy loads the chat data from the folder.
|
|
|
|
Yields:
|
|
ChatSession: A chat session containing the loaded messages.
|
|
|
|
"""
|
|
inbox_path = self.directory_path / "inbox"
|
|
for _dir in inbox_path.iterdir():
|
|
if _dir.is_dir():
|
|
for _file in _dir.iterdir():
|
|
if _file.suffix.lower() == ".json":
|
|
file_loader = SingleFileFacebookMessengerChatLoader(path=_file)
|
|
for result in file_loader.lazy_load():
|
|
yield result
|