acp-sdk 0.13.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

acp_sdk/models/common.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel, ConfigDict
+
+
+class AnyModel(BaseModel):
+    model_config = ConfigDict(extra="allow")

acp_sdk/models/errors.py

@@ -1,7 +1,10 @@
 from enum import Enum
+from typing import Optional
 
 from pydantic import BaseModel
 
+from acp_sdk.models.common import AnyModel
+
 
 class ErrorCode(str, Enum):
     SERVER_ERROR = "server_error"
@@ -10,8 +13,9 @@ class ErrorCode(str, Enum):
 
 
 class Error(BaseModel):
-    code: ErrorCode
+    code: ErrorCode | str  # Allow arbitrary string for backwards compatibility
     message: str
+    data: Optional[AnyModel] = None
 
 
 class ACPError(Exception):

acp_sdk/models/models.py

@@ -8,16 +8,13 @@ from typing import Any, Literal, Optional, Union
 from pydantic import AnyUrl, BaseModel, ConfigDict, Field
 from pydantic.json_schema import SkipJsonSchema
 
+from acp_sdk.models.common import AnyModel
 from acp_sdk.models.errors import ACPError, Error
 from acp_sdk.models.platform import PlatformUIAnnotation
 from acp_sdk.models.types import AgentName, ResourceUrl, RunId, SessionId
 from acp_sdk.shared import ResourceLoader, ResourceStore
 
 
-class AnyModel(BaseModel):
-    model_config = ConfigDict(extra="allow")
-
-
 class Author(BaseModel):
     name: str
     email: str | None = None
@@ -111,11 +108,11 @@ class CitationMetadata(BaseModel):
     """
 
     kind: Literal["citation"] = "citation"
-    start_index: Optional[int]
-    end_index: Optional[int]
-    url: Optional[str]
-    title: Optional[str]
-    description: Optional[str]
+    start_index: Optional[int] = None
+    end_index: Optional[int] = None
+    url: Optional[str] = None
+    title: Optional[str] = None
+    description: Optional[str] = None
 
 
 class TrajectoryMetadata(BaseModel):
@@ -353,10 +350,20 @@ Event = Union[
 ]
 
 
+class InputContentTypes(BaseModel):
+    types: list[str]
+
+
+class OutputContentTypes(BaseModel):
+    types: list[str]
+
+
 class AgentManifest(BaseModel):
     name: str
     description: str | None = None
     metadata: Metadata = Metadata()
+    input_content_types: list[str] = Field(default_factory=lambda: ["*/*"])
+    output_content_types: list[str] = Field(default_factory=lambda: ["*/*"])
 
 
 class Session(BaseModel):

acp_sdk/server/agent.py

@@ -17,6 +17,14 @@ class AgentManifest(abc.ABC):
     def description(self) -> str:
         return ""
 
+    @property
+    def input_content_types(self) -> list[str]:
+        return []
+
+    @property
+    def output_content_types(self) -> list[str]:
+        return []
+
     @property
     def metadata(self) -> Metadata:
         return Metadata()
@@ -38,6 +46,8 @@ def agent(
     description: str | None = None,
     *,
     metadata: Metadata | None = None,
+    input_content_types: list[str] | None = None,
+    output_content_types: list[str] | None = None,
 ) -> Callable[[Callable], AgentManifest]:
     """Decorator to create an agent."""
 
@@ -67,6 +77,14 @@ def agent(
             def metadata(self) -> Metadata:
                 return metadata or Metadata()
 
+            @property
+            def input_content_types(self) -> list[str]:
+                return input_content_types or ["*/*"]
+
+            @property
+            def output_content_types(self) -> list[str]:
+                return output_content_types or ["*/*"]
+
         agent: AgentManifest
         if inspect.isasyncgenfunction(fn):
 

acp_sdk/server/app.py

@@ -13,6 +13,7 @@ from fastapi.encoders import jsonable_encoder
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse, StreamingResponse
 from obstore.exceptions import NotFoundError
+from pydantic import AnyHttpUrl, ValidationError
 
 from acp_sdk.models import (
     ACPError,
@@ -67,6 +68,7 @@ def create_app(
     resource_store: ResourceStore | None = None,
     resource_loader: ResourceLoader | None = None,
     forward_resources: bool = True,
+    forward_base_url: AnyHttpUrl | str | None = None,
     lifespan: Lifespan[AppType] | None = None,
     dependencies: list[Depends] | None = None,
 ) -> FastAPI:
@@ -76,6 +78,12 @@ def create_app(
     ):
         raise ValueError("Resource forwarding must be enabled when resource store does not support HTTP URLs")
 
+    if isinstance(forward_base_url, str):
+        try:
+            forward_base_url = AnyHttpUrl(forward_base_url)
+        except ValidationError:
+            raise ValueError("forward_base_url must be a valid http(s) url")
+
     executor: ThreadPoolExecutor
     client = httpx.AsyncClient()
 
@@ -141,7 +149,13 @@ def create_app(
     async def list_agents() -> AgentsListResponse:
         return AgentsListResponse(
             agents=[
-                AgentModel(name=agent.name, description=agent.description, metadata=agent.metadata)
+                AgentModel(
+                    name=agent.name,
+                    description=agent.description,
+                    metadata=agent.metadata,
+                    input_content_types=agent.input_content_types,
+                    output_content_types=agent.output_content_types,
+                )
                 for agent in agents.values()
             ]
         )
@@ -149,7 +163,13 @@ def create_app(
     @app.get("/agents/{name}")
     async def read_agent(name: AgentName) -> AgentReadResponse:
         agent = find_agent(name)
-        return AgentModel(name=agent.name, description=agent.description, metadata=agent.metadata)
+        return AgentModel(
+            name=agent.name,
+            description=agent.description,
+            metadata=agent.metadata,
+            input_content_types=agent.input_content_types,
+            output_content_types=agent.output_content_types,
+        )
 
     @app.get("/ping")
     async def ping() -> PingResponse:
@@ -165,6 +185,10 @@ def create_app(
         def create_resource_url_forwarded(id: ResourceId) -> ResourceUrl:
             if not forward_resources:
                 raise RuntimeError("Resource forwarding disabled")
+            if forward_base_url:
+                return ResourceUrl(
+                    url=str(app.url_path_for("get_resource", resource_id=id).make_absolute_url(str(forward_base_url)))
+                )
             return ResourceUrl(url=str(req.url_for("get_resource", resource_id=id)))
 
         async def create_resource_url(id: ResourceId) -> ResourceUrl:

acp_sdk/server/server.py

@@ -9,6 +9,7 @@ import requests
 import uvicorn
 import uvicorn.config
 from fastapi import FastAPI
+from pydantic import AnyHttpUrl
 
 from acp_sdk.models import Metadata
 from acp_sdk.server.agent import AgentManifest
@@ -31,13 +32,21 @@ class Server:
         self,
         name: str | None = None,
         description: str | None = None,
+        input_content_types: list[str] | None = None,
+        output_content_types: list[str] | None = None,
         *,
         metadata: Metadata | None = None,
     ) -> Callable:
         """Decorator to register an agent."""
 
         def decorator(fn: Callable) -> Callable:
-            agent = agent_decorator(name=name, description=description, metadata=metadata)(fn)
+            agent = agent_decorator(
+                name=name,
+                description=description,
+                metadata=metadata,
+                input_content_types=input_content_types,
+                output_content_types=output_content_types,
+            )(fn)
             self.register(agent)
             return fn
 
@@ -59,6 +68,8 @@ class Server:
         store: Store | None = None,
         resource_store: ResourceStore | None = None,
         resource_loader: ResourceLoader | None = None,
+        forward_resources: bool = True,
+        forward_base_url: AnyHttpUrl | str | None = None,
         host: str = "127.0.0.1",
         port: int = 8000,
         uds: str | None = None,
@@ -127,6 +138,8 @@ class Server:
             store=store,
             resource_loader=resource_loader,
             resource_store=resource_store,
+            forward_resources=forward_resources,
+            forward_base_url=forward_base_url,
         )
 
         if configure_logger:
@@ -196,6 +209,8 @@ class Server:
         store: Store | None = None,
         resource_store: ResourceStore | None = None,
         resource_loader: ResourceLoader | None = None,
+        forward_resources: bool = True,
+        forward_base_url: AnyHttpUrl | str | None = None,
         host: str = "127.0.0.1",
         port: int = 8000,
         uds: str | None = None,
@@ -256,6 +271,8 @@ class Server:
                 store=store,
                 resource_store=resource_store,
                 resource_loader=resource_loader,
+                forward_resources=forward_resources,
+                forward_base_url=forward_base_url,
                 host=host,
                 port=port,
                 uds=uds,

{acp_sdk-0.13.0.dist-info → acp_sdk-1.0.0.dist-info}/METADATA

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

{acp_sdk-0.13.0.dist-info → acp_sdk-1.0.0.dist-info}/RECORD

@@ -7,20 +7,21 @@ acp_sdk/client/client.py,sha256=3lQ0URQK2q6_QuADMhbbEijrWegaFRtAvfPDs2XnMS8,1259
 acp_sdk/client/types.py,sha256=_H6zYt-2OHOOYRtssRnbDIiwmgsl2-KIXc9lb-mJLFA,133
 acp_sdk/client/utils.py,sha256=2jhJyrPJmVFRoDJh0q_JMqOMlC3IxCh-6HXed-PIZS8,924
 acp_sdk/models/__init__.py,sha256=_XDSvUwlHeRsqUayAxHusJ5s8Z1jWXV1ZcUKlJakOt4,200
-acp_sdk/models/errors.py,sha256=rEyaMVvQuBi7fwWe_d0PGGySYsD3FZTluQ-SkC0yhAs,444
-acp_sdk/models/models.py,sha256=0Zf4hmaQIAyhwB0qO8g46ijO6FJGSb84c60CazojvxE,11930
+acp_sdk/models/common.py,sha256=Rb22TZCTs_cjTSZ1ti0qBHBNh1yH7VL94fIaYwliAYU,117
+acp_sdk/models/errors.py,sha256=r-blfUHLs4OO9twop_Iavb3O_Y3mj18fYBQHGLZiLT0,612
+acp_sdk/models/models.py,sha256=J47J4Az40mnGreiZRCD17gb8QjvOEfY2iuD4jdETaOc,12206
 acp_sdk/models/platform.py,sha256=qgQhBZZNPG9XLlytEZmuv1LTkejbklmlj7YpnpUV_DE,497
 acp_sdk/models/schemas.py,sha256=Vu3aO-6M0IW71I9thxAiphLGXXMvyZXFMJCdj7euUmc,915
 acp_sdk/models/types.py,sha256=tH79qqpW3DrPJggZD1KFlLtGE_ZKT6eLtwiR3p3x2QM,145
 acp_sdk/server/__init__.py,sha256=TfTOOI7G2nejjOB6RZB5ujynyh3bFeH51pfDIMzv3N4,627
-acp_sdk/server/agent.py,sha256=tzdk9bXb4aKBKD3F1-QfI7zrjMZYEg6qm3gGRWJGC1E,3739
-acp_sdk/server/app.py,sha256=dugceYPELl6N1_OQQ6JP5VCfVSm0rcanFtjDpscOy6o,11586
+acp_sdk/server/agent.py,sha256=T43TuYT4iVzfnxytvF1CbolLC_KoSRnITogmGWaRabs,4271
+acp_sdk/server/app.py,sha256=SYjpo6gVJXkTpnQ70lYjLLXz23nIOQAT6X4n2QWZhUw,12502
 acp_sdk/server/context.py,sha256=Xz1am_HLNTgEvG0IPtS0tRJcPk-rEhLtNyTQGUAvLQw,1271
 acp_sdk/server/errors.py,sha256=GSO8yYIqEeX8Y4Lz86ks35dMTHiQiXuOrLYYx0eXsbI,2110
 acp_sdk/server/executor.py,sha256=ktM1FCwixtbjsU60VM7PHxwF_Saz15M_9fNXN4mItrU,12607
 acp_sdk/server/logging.py,sha256=Oc8yZigCsuDnHHPsarRzu0RX3NKaLEgpELM2yovGKDI,411
 acp_sdk/server/resources.py,sha256=g9C2Xr_SyvAFsV1uHQLEe1H7CgLIjfQZ51Nj3n-w0f4,1174
-acp_sdk/server/server.py,sha256=3Nsru3u7FYjrSfRNNuCBHQWZlyk8HvLWnFFkmyODZmA,14648
+acp_sdk/server/server.py,sha256=cnFAquwNvFnHQTzH20ZOJcwcTJFtgK78-uEthZJw0zc,15364
 acp_sdk/server/telemetry.py,sha256=lbB2ppijUcqbHUOn0e-15LGcVvT_qrMguq8qBokICac,2016
 acp_sdk/server/types.py,sha256=gLb5wCkMYhmu2laj_ymK-TPfN9LSjRgKOP1H_893UzA,304
 acp_sdk/server/utils.py,sha256=O9GUZ8SKqk8P5Vo-N4BD7hXS3XuE7l8wifOhvKMX3ws,2715
@@ -32,6 +33,6 @@ acp_sdk/server/store/store.py,sha256=jGmYy9oiuVjhYYJY8QRo4g2J2Qyt1HLTmq_eHy4aI7c
 acp_sdk/server/store/utils.py,sha256=JumEOMs1h1uGlnHnUGeguee-srGzT7_Y2NVEYt01QuY,92
 acp_sdk/shared/__init__.py,sha256=vZuhdQ6lrWVdyYPmIyXpPvs5eMnzemhNek3gfYPA3Bc,138
 acp_sdk/shared/resources.py,sha256=3oPvevYv2B1YaHyoMH85B7fcHVyhgZDUaDCBVEflTlA,1592
-acp_sdk-0.13.0.dist-info/METADATA,sha256=xEVm9sg7kLKMgLmGOcnOoir8KFQUWovrY8YGq8JUx-I,1713
-acp_sdk-0.13.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.13.0.dist-info/RECORD,,
+acp_sdk-1.0.0.dist-info/METADATA,sha256=Tez7d-Zl2X_j5oXqVCil6kUZdaIO_BWOep_LPbaPF6g,1712
+acp_sdk-1.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-1.0.0.dist-info/RECORD,,