acp-sdk 0.12.0__py3-none-any.whl → 0.12.1__py3-none-any.whl

acp_sdk/models/models.py

@@ -9,7 +9,7 @@ from pydantic import AnyUrl, BaseModel, ConfigDict, Field
 
 from acp_sdk.models.errors import ACPError, Error
 from acp_sdk.models.platform import PlatformUIAnnotation
-from acp_sdk.models.types import AgentName, ResourceId, ResourceUrl, RunId, SessionId
+from acp_sdk.models.types import AgentName, ResourceUrl, RunId, SessionId
 from acp_sdk.shared import ResourceLoader, ResourceStore
 
 
@@ -357,6 +357,6 @@ class Session(BaseModel):
         if not store:
             raise ValueError("Store must be specified")
 
-        id = ResourceId()
+        id = uuid.uuid4()
         await store.store(id, data)
         return await store.url(id)

acp_sdk/server/app.py

@@ -51,6 +51,7 @@ from acp_sdk.server.errors import (
     validation_exception_handler,
 )
 from acp_sdk.server.executor import CancelData, Executor, RunData
+from acp_sdk.server.resources import ServerResourceLoader
 from acp_sdk.server.store import MemoryStore, Store
 from acp_sdk.server.utils import stream_sse, wait_util_stop
 from acp_sdk.shared import ResourceLoader, ResourceStore
@@ -161,14 +162,30 @@ def create_app(
         if request.session_id and request.session and request.session_id != request.session.id:
             raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Session ID mismatch")
 
+        def create_resource_url_forwarded(id: ResourceId) -> ResourceUrl:
+            if not forward_resources:
+                raise RuntimeError("Resource forwarding disabled")
+            return ResourceUrl(url=str(req.url_for("get_resource", resource_id=id)))
+
+        async def create_resource_url(id: ResourceId) -> ResourceUrl:
+            if forward_resources:
+                return create_resource_url_forwarded(id)
+            else:
+                return await resource_store.url(id)
+
+        server_resource_loader = ServerResourceLoader(
+            loader=resource_loader,
+            store=resource_store,
+            create_resource_url=create_resource_url_forwarded if forward_resources else None,
+        )
+
         session = request.session or (
-            (
-                await session_store.get(request.session_id)
-                or Session(id=request.session_id, loader=resource_loader, store=resource_store)
-            )
+            (await session_store.get(request.session_id) or Session(id=request.session_id))
             if request.session_id
-            else Session(loader=resource_loader, store=resource_store)
+            else Session()
         )
+        session.loader = server_resource_loader
+        session.store = resource_store
 
         nonlocal executor
         run_data = RunData(
@@ -183,12 +200,6 @@ def create_app(
         headers = {Headers.RUN_ID: str(run_data.run.run_id)}
         ready = asyncio.Event()
 
-        async def create_resource_url(id: ResourceId) -> ResourceUrl:
-            if forward_resources:
-                return ResourceUrl(url=str(req.url_for("get_resource", resource_id=id)))
-            else:
-                return await resource_store.url(id)
-
         Executor(
             agent=agent,
             run_data=run_data,
@@ -200,7 +211,7 @@ def create_app(
             executor=executor,
             request=req,
             resource_store=resource_store,
-            resource_loader=resource_loader,
+            resource_loader=server_resource_loader,
             create_resource_url=create_resource_url,
         ).execute(request.input, wait=ready)
 

acp_sdk/server/resources.py

@@ -0,0 +1,38 @@
+import re
+import uuid
+from typing import Callable
+
+import cachetools
+
+from acp_sdk.models import ResourceUrl
+from acp_sdk.models.types import ResourceId
+from acp_sdk.shared.resources import ResourceLoader, ResourceStore
+
+
+class ServerResourceLoader(ResourceLoader):
+    def __init__(
+        self,
+        *,
+        loader: ResourceLoader,
+        store: ResourceStore,
+        create_resource_url: Callable[[ResourceId], ResourceUrl] | None,
+    ) -> None:
+        self._loader = loader
+        self._store = store
+
+        placeholder_id = uuid.uuid4()
+        self._url_pattern = (
+            re.escape(str(create_resource_url(placeholder_id))).replace(re.escape(str(placeholder_id)), r"([^/?&#]+)")
+            if create_resource_url
+            else None
+        )
+
+    @cachetools.func.lfu_cache
+    async def load(self, url: ResourceUrl) -> bytes:
+        if self._url_pattern:
+            match = re.match(self._url_pattern, str(url))
+            if match:
+                id = ResourceId(match.group(1))
+                result = await self._store.load(id)
+                return (await result.bytes_async()).to_bytes()
+        return await self._loader.load(url)

{acp_sdk-0.12.0.dist-info → acp_sdk-0.12.1.dist-info}/METADATA

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

{acp_sdk-0.12.0.dist-info → acp_sdk-0.12.1.dist-info}/RECORD

@@ -8,17 +8,18 @@ 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=OBpCsRq7vSkPN67MrT40LvfzitwjXK_darSFdj1G_zE,10821
+acp_sdk/models/models.py,sha256=-ORIe32MOWVVh22njH596lKeQQQCRsXzKLvOk1dQJB8,10809
 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=KKjZcVxJxQvxAEP_cQna4f9ed1O5mN9Ldo1X_tXfFfI,11107
+acp_sdk/server/app.py,sha256=dugceYPELl6N1_OQQ6JP5VCfVSm0rcanFtjDpscOy6o,11586
 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=HJOub1ynHiJTQrfKq_bq0r9aqgL1kQeboZH_ArZcuh0,14651
 acp_sdk/server/telemetry.py,sha256=lbB2ppijUcqbHUOn0e-15LGcVvT_qrMguq8qBokICac,2016
 acp_sdk/server/types.py,sha256=gLb5wCkMYhmu2laj_ymK-TPfN9LSjRgKOP1H_893UzA,304
@@ -31,6 +32,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.12.0.dist-info/METADATA,sha256=AFY7mxImu6UwA0LZlc0ItpdmhLzw8Tw59AJXBjeWq2Q,1713
-acp_sdk-0.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.12.0.dist-info/RECORD,,
+acp_sdk-0.12.1.dist-info/METADATA,sha256=MRAqSfZU7kiGP1mmHqXMcPVxfDVhsQBUbPJVOhaIxYw,1713
+acp_sdk-0.12.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.12.1.dist-info/RECORD,,