acp-sdk 0.8.2__tar.gz → 0.8.3__tar.gz

{acp_sdk-0.8.2 → acp_sdk-0.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.8.2
+Version: 0.8.3
 Summary: Agent Communication Protocol SDK
 Author: IBM Corp.
 Maintainer-email: Tomas Pilar <thomas7pilar@gmail.com>

{acp_sdk-0.8.2 → acp_sdk-0.8.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "acp-sdk"
-version = "0.8.2"
+version = "0.8.3"
 description = "Agent Communication Protocol SDK"
 license = "Apache-2.0"
 readme = "README.md"

{acp_sdk-0.8.2 → acp_sdk-0.8.3}/src/acp_sdk/models/models.py

@@ -95,7 +95,7 @@ class Artifact(MessagePart):
 
 class Message(BaseModel):
     parts: list[MessagePart]
-    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    created_at: datetime | None = Field(default_factory=lambda: datetime.now(timezone.utc))
     completed_at: datetime | None = Field(default_factory=lambda: datetime.now(timezone.utc))
 
     def __add__(self, other: "Message") -> "Message":
@@ -103,8 +103,10 @@ class Message(BaseModel):
             raise TypeError(f"Cannot concatenate Message with {type(other).__name__}")
         return Message(
             parts=self.parts + other.parts,
-            created_at=min(self.created_at, other.created_at),
-            completed_at=max(self.completed_at, other.completed_at),
+            created_at=min(self.created_at, other.created_at) if self.created_at and other.created_at else None,
+            completed_at=max(self.completed_at, other.completed_at)
+            if self.completed_at and other.completed_at
+            else None,
         )
 
     def __str__(self) -> str:

{acp_sdk-0.8.2 → acp_sdk-0.8.3}/tests/unit/models/test_models.py

@@ -26,7 +26,27 @@ timestamp = "2021-09-09T22:02:47.89Z"
                 created_at=timestamp,
                 completed_at=timestamp,
             ),
-        )
+        ),
+        (
+            Message(
+                parts=[MessagePart(content_type="text/plain", content="Foo")],
+                created_at=None,
+                completed_at=timestamp,
+            ),
+            Message(
+                parts=[MessagePart(content_type="text/plain", content="Bar")],
+                created_at=timestamp,
+                completed_at=None,
+            ),
+            Message(
+                parts=[
+                    MessagePart(content_type="text/plain", content="Foo"),
+                    MessagePart(content_type="text/plain", content="Bar"),
+                ],
+                created_at=None,
+                completed_at=None,
+            ),
+        ),
     ],
 )
 def test_message_add(first: Message, second: Message, result: Message) -> None: