From 7101e0e7d527ec608362006927185bf0145de088 Mon Sep 17 00:00:00 2001 From: Nil-Andreu Date: Sun, 8 Jan 2023 20:17:14 +0100 Subject: [PATCH] [NEW] Message embedding created_date --- backend/oasst_backend/models/message_embedding.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/backend/oasst_backend/models/message_embedding.py b/backend/oasst_backend/models/message_embedding.py index 697a776b..74da5004 100644 --- a/backend/oasst_backend/models/message_embedding.py +++ b/backend/oasst_backend/models/message_embedding.py @@ -1,4 +1,5 @@ -from typing import List +from datetime import datetime +from typing import List, Optional from uuid import UUID import sqlalchemy as sa @@ -13,3 +14,8 @@ class MessageEmbedding(SQLModel, table=True): message_id: UUID = Field(sa_column=sa.Column(pg.UUID(as_uuid=True), sa.ForeignKey("message.id"), nullable=False)) model: str = Field(max_length=256, nullable=False) embedding: List[float] = Field(sa_column=sa.Column(ARRAY(Float)), nullable=True) + + # In the case that the Message Embedding is created afterwards + created_date: Optional[datetime] = Field( + sa_column=sa.Column(sa.DateTime(), nullable=False, server_default=sa.func.current_timestamp()) + )