welearn-database 1.3.0.dev2__tar.gz → 1.3.0.dev3__tar.gz

{welearn_database-1.3.0.dev2 → welearn_database-1.3.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: welearn-database
-Version: 1.3.0.dev2
+Version: 1.3.0.dev3
 Summary: All stuff related to relationnal database from the WeLearn project
 License: cc-by-sa-nc
 Author: Théo

{welearn_database-1.3.0.dev2 → welearn_database-1.3.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "welearn-database"
-version = "1.3.0.dev2"
+version = "1.3.0.dev3"
 description = "All stuff related to relationnal database from the WeLearn project"
 authors = [
     {name = "Théo",email = "theo.nardin@cri-paris.org"}

{welearn_database-1.3.0.dev2 → welearn_database-1.3.0.dev3}/welearn_database/data/models/user_related.py

@@ -90,6 +90,10 @@ class ChatMessage(Base):
         ForeignKey(f"{DbSchemaEnum.USER_RELATED.value}.inferred_user.id"),
         nullable=False,
     )
+    conversation_id = mapped_column(
+        types.Uuid,
+        nullable=False,
+    )
     role: Mapped[str]
     textual_content: Mapped[str]
 
@@ -159,6 +163,23 @@ class APIKeyManagement(Base):
     )
 
 
+class DataCollectionCampaignManagement(Base):
+    __tablename__ = "data_collection_campaign_management"
+    __table_args__ = {"schema": DbSchemaEnum.USER_RELATED.value}
+
+    id: Mapped[UUID] = mapped_column(
+        types.Uuid, primary_key=True, nullable=False, server_default="gen_random_uuid()"
+    )
+    is_active: Mapped[bool]
+    end_at: Mapped[datetime] = mapped_column(TIMESTAMP(timezone=False), nullable=False)
+    created_at: Mapped[datetime] = mapped_column(
+        TIMESTAMP(timezone=False),
+        nullable=False,
+        default=func.localtimestamp(),
+        server_default="NOW()",
+    )
+
+
 class Session(Base):
     __tablename__ = "session"
     __table_args__ = {"schema": "user_related"}