Merge pull request #88 from TogetherCrew/fastapi

feat: Payload persist handling the case of document being available!
TogetherCrew · Nov 4, 2024 · 0589d41 · 0589d41
2 parents db72e35 + 88a2eb5
commit 0589d41
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/utils/persist_payload.py b/utils/persist_payload.py
@@ -63,16 +63,24 @@ def persist_http(self, payload: HTTPPayload, update: bool = False) -> None:
                     f"{community_id} persisted successfully!"
                 )
             else:
+                # Check if createdAt needs to be set if it doesn't exist
+                self.client[self.db][self.external_msgs_collection].update_one(
+                    {"taskId": payload.taskId, "createdAt": {"$exists": False}},
+                    {
+                        "$set": {
+                            "createdAt": datetime.now().replace(tzinfo=timezone.utc)
+                        }
+                    },
+                )
+
+                # Update or upsert the main document
                 self.client[self.db][self.external_msgs_collection].update_one(
                     {"taskId": payload.taskId},
                     {
                         "$set": {
-                            "response": payload.response.model_dump(),
+                            **payload.model_dump(),
                             "updatedAt": datetime.now().replace(tzinfo=timezone.utc),
-                        },
-                        "$setOnInsert": {
-                            "createdAt": datetime.now().replace(tzinfo=timezone.utc),
-                        },
+                        }
                     },
                     upsert=True,
                 )