From 969f2927abcb99fb77e0d3e19fce8c4320fc5582 Mon Sep 17 00:00:00 2001
From: Diicell <44242534+diicellman@users.noreply.github.com>
Date: Wed, 20 Mar 2024 23:07:35 +0600
Subject: [PATCH] FIX TelegramReader. (#12112)
FIX TelegramReader
---
.../llama_index/readers/telegram/base.py | 17 ++++++++---------
1 file changed, 8 insertions(+), 9 deletions(-)
diff --git a/llama-index-integrations/readers/llama-index-readers-telegram/llama_index/readers/telegram/base.py b/llama-index-integrations/readers/llama-index-readers-telegram/llama_index/readers/telegram/base.py
index 7b5f45a27e..0567bd8dc8 100644
--- a/llama-index-integrations/readers/llama-index-readers-telegram/llama_index/readers/telegram/base.py
+++ b/llama-index-integrations/readers/llama-index-readers-telegram/llama_index/readers/telegram/base.py
@@ -58,8 +58,8 @@ class TelegramReader(BaseReader):
entity_name: str,
post_id: Optional[int] = None,
limit: Optional[int] = None,
- start_date: Optional[datetime.date] = None,
- end_date: Optional[datetime.date] = None,
+ start_date: Optional[datetime.datetime] = None,
+ end_date: Optional[datetime.datetime] = None,
) -> List[Document]:
"""Load posts/chat messages/comments from Telegram channels or chats.
@@ -73,8 +73,8 @@ class TelegramReader(BaseReader):
the comments that reply to this ID will be returned.\
Else will get posts/chat messages.
limit (int): Number of messages to be retrieved.
- start_date (datetime.date): Start date of the time period.
- end_date (datetime.date): End date of the time period.
+ start_date (datetime.datetime): Start date of the time period.
+ end_date (datetime.datetime): End date of the time period.
"""
return self.loop.run_until_complete(
@@ -92,8 +92,8 @@ class TelegramReader(BaseReader):
entity_name: str,
post_id: Optional[int] = None,
limit: Optional[int] = None,
- start_date: Optional[datetime.date] = None,
- end_date: Optional[datetime.date] = None,
+ start_date: Optional[datetime.datetime] = None,
+ end_date: Optional[datetime.datetime] = None,
) -> List[Document]:
"""Load posts/chat messages/comments from Telegram channels or chats.
@@ -103,8 +103,8 @@ class TelegramReader(BaseReader):
the comments that reply to this ID will be returned.\
Else will get posts/chat messages.
limit (int): Number of messages to be retrieved.
- start_date (datetime.date): Start date of the time period.
- end_date (datetime.date): End date of the time period.
+ start_date (datetime.datetime): Start date of the time period.
+ end_date (datetime.datetime): End date of the time period.
"""
import telethon
@@ -121,7 +121,6 @@ class TelegramReader(BaseReader):
reply_to=post_id,
limit=limit,
offset_date=end_date,
- reverse=True,
):
if message.date < start_date:
break
--
GitLab