quivr/backend/modules/notification/entity/notification.py
Stan Girard 08e015af6c
feat(lcel): migrated to lcel and pydantic (#2185)
# Description

Please include a summary of the changes and the related issue. Please
also include relevant motivation and context.

## Checklist before requesting a review

Please delete options that are not relevant.

- [ ] My code follows the style guidelines of this project
- [ ] I have performed a self-review of my code
- [ ] I have commented hard-to-understand areas
- [ ] I have ideally added tests that prove my fix is effective or that
my feature works
- [ ] New and existing unit tests pass locally with my changes
- [ ] Any dependent changes have been merged

## Screenshots (if appropriate):

---------

Co-authored-by: Antoine Dewez <44063631+Zewed@users.noreply.github.com>
2024-02-14 14:01:35 -08:00

26 lines
630 B
Python

from enum import Enum
from typing import Optional
from uuid import UUID
from pydantic import BaseModel
class NotificationsStatusEnum(str, Enum):
Pending = "Pending"
Done = "Done"
class Notification(BaseModel):
id: UUID
datetime: str
chat_id: Optional[UUID] = None
message: Optional[str] = None
action: str
status: NotificationsStatusEnum
def dict(self, *args, **kwargs):
notification_dict = super().dict(*args, **kwargs)
if notification_dict.get("chat_id"):
notification_dict["chat_id"] = str(notification_dict.get("chat_id"))
return notification_dict