acp-sdk 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

acp_sdk/models/models.py

@@ -1,4 +1,5 @@
 import uuid
+from datetime import datetime
 from enum import Enum
 from typing import Any, Literal, Optional, Union
 
@@ -7,22 +8,73 @@ from pydantic import AnyUrl, BaseModel, ConfigDict, Field
 from acp_sdk.models.errors import Error
 
 
-class Metadata(BaseModel):
+class AnyModel(BaseModel):
     model_config = ConfigDict(extra="allow")
 
 
-class AnyModel(BaseModel):
+class Author(BaseModel):
+    name: str
+    email: str | None = None
+    url: AnyUrl | None = None
+
+
+class Contributor(BaseModel):
+    name: str
+    email: str | None = None
+    url: AnyUrl | None = None
+
+
+class LinkType(str, Enum):
+    SOURCE_CODE = "source-code"
+    CONTAINER_IMAGE = "container-image"
+    HOMEPAGE = "homepage"
+    DOCUMENTATION = "documentation"
+
+
+class Link(BaseModel):
+    type: LinkType
+    url: AnyUrl
+
+
+class DependencyType(str, Enum):
+    AGENT = "agent"
+    TOOL = "tool"
+    MODEL = "model"
+
+
+class Dependency(BaseModel):
+    type: DependencyType
+    name: str
+
+
+class Metadata(BaseModel):
+    annotations: AnyModel | None = None
+    documentation: str | None = None
+    license: str | None = None
+    programming_language: str | None = None
+    natural_languages: list[str] | None = None
+    framework: str | None = None
+    use_cases: list[str] | None = None
+    tags: list[str] | None = None
+    created_at: datetime | None = None
+    updated_at: datetime | None = None
+    author: Author | None = None
+    contributors: list[Contributor] | None = None
+    links: list[Link] | None = None
+    dependencies: list[Dependency] | None = None
+    recommended_models: list[str] | None = None
+
     model_config = ConfigDict(extra="allow")
 
 
 class MessagePart(BaseModel):
     name: Optional[str] = None
-    content_type: str
+    content_type: Optional[str] = "text/plain"
     content: Optional[str] = None
     content_encoding: Optional[Literal["plain", "base64"]] = "plain"
     content_url: Optional[AnyUrl] = None
 
-    model_config = ConfigDict(extra="forbid")
+    model_config = ConfigDict(extra="allow")
 
     def model_post_init(self, __context: Any) -> None:
         if self.content is None and self.content_url is None:
@@ -48,6 +100,36 @@ class Message(BaseModel):
             part.content for part in self.parts if part.content is not None and part.content_type == "text/plain"
         )
 
+    def compress(self) -> "Message":
+        def can_be_joined(first: MessagePart, second: MessagePart) -> bool:
+            return (
+                first.name is None
+                and second.name is None
+                and first.content_type == "text/plain"
+                and second.content_type == "text/plain"
+                and first.content_encoding == "plain"
+                and second.content_encoding == "plain"
+                and first.content_url is None
+                and second.content_url is None
+            )
+
+        def join(first: MessagePart, second: MessagePart) -> MessagePart:
+            return MessagePart(
+                name=None,
+                content_type="text/plain",
+                content=first.content + second.content,
+                content_encoding="plain",
+                content_url=None,
+            )
+
+        parts: list[MessagePart] = []
+        for part in self.parts:
+            if len(parts) > 0 and can_be_joined(parts[-1], part):
+                parts[-1] = join(parts[-1], part)
+            else:
+                parts.append(part)
+        return Message(parts=parts)
+
 
 AgentName = str
 SessionId = uuid.UUID

acp_sdk/server/agent.py

@@ -129,7 +129,7 @@ def agent(
 
             @property
             def description(self) -> str:
-                return description or fn.__doc__ or ""
+                return description or inspect.getdoc(fn) or ""
 
             @property
             def metadata(self) -> Metadata:

{acp_sdk-0.3.1.dist-info → acp_sdk-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.3.1
+Version: 0.3.3
 Summary: Agent Communication Protocol SDK
 Author: IBM Corp.
 Maintainer-email: Tomas Pilar <thomas7pilar@gmail.com>
@@ -63,7 +63,7 @@ server = Server()
 async def echo(inputs: list[Message], context: Context) -> AsyncGenerator[RunYield, RunYieldResume]:
     """Echoes everything"""
     for message in inputs:
-        yield {"thought": "I should echo everyting"}
+        yield {"thought": "I should echo everything"}
         await asyncio.sleep(0.5)
         yield message
 

{acp_sdk-0.3.1.dist-info → acp_sdk-0.3.3.dist-info}/RECORD

@@ -6,10 +6,10 @@ acp_sdk/client/__init__.py,sha256=Bca1DORrswxzZsrR2aUFpATuNG2xNSmYvF1Z2WJaVbc,51
 acp_sdk/client/client.py,sha256=lEUs0Oc7MZbKkTF2E6e8Wn3dTW5cMVj6fD8TbuEOMDk,6584
 acp_sdk/models/__init__.py,sha256=numSDBDT1QHx7n_Y3Deb5VOvKWcUBxbOEaMwQBSRHxc,151
 acp_sdk/models/errors.py,sha256=rEyaMVvQuBi7fwWe_d0PGGySYsD3FZTluQ-SkC0yhAs,444
-acp_sdk/models/models.py,sha256=GjsVvP5NVlW_9pbi47vBZA94WRL8bcMzbWIebwZuJGs,4158
+acp_sdk/models/models.py,sha256=eOoOfftPKbTaLlqZlE4gE1ZwDkrQ_FdRiDx4wSJILP4,6539
 acp_sdk/models/schemas.py,sha256=Kj7drJSR8d-N3KHzu_qTnLdagrMtAyhid5swluuhHTw,645
 acp_sdk/server/__init__.py,sha256=mxBBBFaZuMEUENRMLwp1XZkuLeT9QghcFmNvjnqvAAU,377
-acp_sdk/server/agent.py,sha256=fGky5MIuknw-Gy-THqhWLt9I9-gUyNIar8qEAvZb3uQ,6195
+acp_sdk/server/agent.py,sha256=DhcPDPDL9jpDST40K_bInvDXfpF1cwxIhqXzu8z0blU,6203
 acp_sdk/server/app.py,sha256=3QJzGi05DKoIKyTeYYKKQhcYz4qorQKmVI94_guVkEY,6340
 acp_sdk/server/bundle.py,sha256=6LZnxsP1rawxic9CwDAQCsOV1v31qNI9meKXMty_yWg,6260
 acp_sdk/server/context.py,sha256=MgnLV6qcDIhc_0BjW7r4Jj1tHts4ZuwpdTGIBnz2Mgo,1036
@@ -20,6 +20,6 @@ acp_sdk/server/session.py,sha256=0cDr924HC5x2bBNbK9NSKVHAt5A_mi5dK8P4jP_ugq0,629
 acp_sdk/server/telemetry.py,sha256=1BUxNg-xL_Vqgs27PDWNc3HikrQW2lidAtT_FKlp_Qk,1833
 acp_sdk/server/types.py,sha256=E0_9xWwgGzyvJjxtbeBBmSbIPhbbTSXLpHFL5dZDzxI,182
 acp_sdk/server/utils.py,sha256=EfrF9VCyVk3AM_ao-BIB9EzGbfTrh4V2Bz-VFr6f6Sg,351
-acp_sdk-0.3.1.dist-info/METADATA,sha256=Eng_oiR7OIfJK3XDKIAjlmlFX8eHoVqxkd_6XIVpXlQ,3119
-acp_sdk-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.3.1.dist-info/RECORD,,
+acp_sdk-0.3.3.dist-info/METADATA,sha256=X4iVzszNG3pWp7eOLPHz6cxaGI4rmMN8MrtCx1ICTOI,3120
+acp_sdk-0.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.3.3.dist-info/RECORD,,