nucliadb-models 6.7.0.post4806__py3-none-any.whl → 6.7.0.post4811__py3-none-any.whl

nucliadb_models/conversation.py

@@ -93,11 +93,20 @@ class InputMessageContent(BaseModel):
 
 
 class InputMessage(BaseModel):
-    timestamp: Optional[datetime] = None
-    who: Optional[str] = None
-    to: List[str] = []
+    timestamp: Optional[datetime] = Field(
+        default=None, description="Time at which the message was sent, in ISO 8601 format."
+    )
+    who: Optional[str] = Field(
+        default=None, description="Sender of the message, e.g. 'user' or 'assistant'"
+    )
+    to: List[str] = Field(
+        default_factory=list,
+        description="List of recipients of the message, e.g. ['assistant'] or ['user']",
+    )
     content: InputMessageContent
-    ident: str
+    ident: str = Field(
+        description="Unique identifier for the message. Must be unique within the conversation."
+    )
     type_: Optional[MessageType] = Field(None, alias="type")
 
     @field_validator("ident", mode="after")
@@ -111,7 +120,10 @@ class InputMessage(BaseModel):
 
 
 class InputConversationField(BaseModel):
-    messages: List[InputMessage] = []
+    messages: List[InputMessage] = Field(
+        default_factory=list,
+        description="List of messages in the conversation field. Each message must have a unique ident.",
+    )
     extract_strategy: Optional[str] = Field(
         default=None,
         description="Id of the Nuclia extract strategy used at processing time. If not set, the default strategy was used. Extract strategies are defined at the learning configuration api.",
@@ -120,3 +132,13 @@ class InputConversationField(BaseModel):
         default=None,
         description="Id of the Nuclia split strategy used at processing time. If not set, the default strategy was used. Split strategies are defined at the learning configuration api.",
     )
+
+    @field_validator("messages", mode="after")
+    @classmethod
+    def idents_are_unique(cls, value: List[InputMessage]) -> List[InputMessage]:
+        seen_idents = set()
+        for message in value:
+            if message.ident in seen_idents:
+                raise ValueError(f'Message ident "{message.ident}" is not unique')
+            seen_idents.add(message.ident)
+        return value

{nucliadb_models-6.7.0.post4806.dist-info → nucliadb_models-6.7.0.post4811.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_models
-Version: 6.7.0.post4806
+Version: 6.7.0.post4811
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0
 Project-URL: Homepage, https://nuclia.com

{nucliadb_models-6.7.0.post4806.dist-info → nucliadb_models-6.7.0.post4811.dist-info}/RECORD

@@ -2,7 +2,7 @@ nucliadb_models/__init__.py,sha256=3y8-htogKuCZcbhaUZdSjTeEjUSeec9aRWyL8AlKCyM,1
 nucliadb_models/common.py,sha256=YW84w1NAQARObs2nXw6YBgdxQJeVCmTZZr5lSqj-IdQ,7904
 nucliadb_models/configuration.py,sha256=aTV5mBwYFlwiV1_nWyVAXaCh7F6lDVTVh28Xfwy8ox8,2448
 nucliadb_models/content_types.py,sha256=36Ga-iGf4ivCqgtXC7imFgegrwHB117s9eqP62JtGv0,3456
-nucliadb_models/conversation.py,sha256=ynS65VNgySOyDEHRFaBgcR8_eU6raYlg6lwdQJqCnwg,3689
+nucliadb_models/conversation.py,sha256=_qs1r5X2_1q-0fSlVNnrE3f9N_OTrZEHVEJ_yiMqPxU,4666
 nucliadb_models/entities.py,sha256=i-7Y8qmFRRTih5zw0ajv1U_iiXexe66M3TK8hUikQZk,2356
 nucliadb_models/export_import.py,sha256=mNm9IArOLnC6TLupkwqVFhxD5d08mpIVOVFneECv8UA,1073
 nucliadb_models/external_index_providers.py,sha256=IIKjJjLixWQC1zrbzam2FDcAo5UUxShZfueZSxqZu8Y,1535
@@ -32,7 +32,7 @@ nucliadb_models/graph/responses.py,sha256=Sdq8OgFAL1YT-1lJyLLrkqcScvj7YTEqAUwQ-k
 nucliadb_models/internal/__init__.py,sha256=zG33bUz1rHFPtvqQPWn4rDwBJt3FJodGuQYD45quiQg,583
 nucliadb_models/internal/predict.py,sha256=Pnx6MmLfK65eExe1XnVxqmSlvMwdowewwks9BOEoqMw,2029
 nucliadb_models/internal/shards.py,sha256=__y1OZtWGiNcPQEWfSFOj8yw458WGi7mM4vZe0K-L1Y,1691
-nucliadb_models-6.7.0.post4806.dist-info/METADATA,sha256=g_UkHFecqXmdZ-yyylxkBbhPr4YfQne4mEzbLje1yrE,794
-nucliadb_models-6.7.0.post4806.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb_models-6.7.0.post4806.dist-info/top_level.txt,sha256=UrY1I8oeovIRwkXLYplssTrxQdUjhSEFDFbnwaIV3tA,16
-nucliadb_models-6.7.0.post4806.dist-info/RECORD,,
+nucliadb_models-6.7.0.post4811.dist-info/METADATA,sha256=_8xfPMjbuusxtLAq6rpcz5jLI-fljzH-5QStauMhx_4,794
+nucliadb_models-6.7.0.post4811.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb_models-6.7.0.post4811.dist-info/top_level.txt,sha256=UrY1I8oeovIRwkXLYplssTrxQdUjhSEFDFbnwaIV3tA,16
+nucliadb_models-6.7.0.post4811.dist-info/RECORD,,