From 3f50feb28059cf255128cec7f5092502689a6b1a Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Fri, 17 Feb 2023 00:53:01 -0800 Subject: [PATCH] fix telegram imports (#1110) --- langchain/document_loaders/telegram.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/langchain/document_loaders/telegram.py b/langchain/document_loaders/telegram.py index 3a7ae860..07f571d7 100644 --- a/langchain/document_loaders/telegram.py +++ b/langchain/document_loaders/telegram.py @@ -3,8 +3,6 @@ import json from pathlib import Path from typing import List -import pandas as pd - from langchain.docstore.document import Document from langchain.document_loaders.base import BaseLoader @@ -26,6 +24,13 @@ class TelegramChatLoader(BaseLoader): def load(self) -> List[Document]: """Load documents.""" + try: + import pandas as pd + except ImportError: + raise ValueError( + "pandas is needed for Telegram loader, " + "please install with `pip install pandas`" + ) p = Path(self.file_path) with open(p, encoding="utf8") as f: