Merge pull request #319 from TogetherCrew/feat/313-telegram-summarizer

feat: saving datetime instead of date for Qdrant db!
TogetherCrew · Nov 6, 2024 · 74bc2a9 · 74bc2a9
2 parents bef38dd + 08c52dd
commit 74bc2a9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dags/hivemind_etl_helpers/src/db/telegram/transform/summary.py b/dags/hivemind_etl_helpers/src/db/telegram/transform/summary.py
@@ -1,4 +1,4 @@
-from datetime import date, timedelta
+from datetime import date, datetime, timedelta, timezone
 
 from llama_index.core import Document
 
@@ -32,7 +32,9 @@ def transform(self, summaries: dict[date, str]) -> list[Document]:
                 + (day + timedelta(days=1)).strftime("%Y-%m-%d"),
                 text=summary,
                 metadata={
-                    "date": day,
+                    "date": datetime.combine(
+                        day, datetime.min.time(), tzinfo=timezone.utc
+                    ),
                 },
             )
             summary_docs.append(document)