fix(codegen): expose protocol method metadata

Author: PsiACE

scripts/gen_meta.py

@@ -54,6 +54,7 @@ def render_meta(*, protocol_version: int = 1) -> str:
     data = json.loads(meta_json.read_text("utf-8"))
     agent_methods = data.get("agentMethods", {})
     client_methods = data.get("clientMethods", {})
+    protocol_methods = data.get("protocolMethods")
     version = data.get("version", 1)
     header_lines = [f"# Generated from {meta_json.relative_to(ROOT)}. Do not edit by hand."]
     if version_file.exists():
@@ -66,6 +67,7 @@ def render_meta(*, protocol_version: int = 1) -> str:
         + "\n"
         + f"AGENT_METHODS = {json.dumps(agent_methods, indent=4)}\n"
         + f"CLIENT_METHODS = {json.dumps(client_methods, indent=4)}\n"
+        + (f"PROTOCOL_METHODS = {json.dumps(protocol_methods, indent=4)}\n" if protocol_methods is not None else "")
         + f"PROTOCOL_VERSION = {int(version)}\n"
     )
     result = subprocess.run(  # noqa: S603

src/acp/experimental/v2/__init__.py

@@ -1,5 +1,5 @@
 """Experimental ACP protocol v2 bindings."""
 
-from .meta import AGENT_METHODS, CLIENT_METHODS, PROTOCOL_VERSION
+from .meta import AGENT_METHODS, CLIENT_METHODS, PROTOCOL_METHODS, PROTOCOL_VERSION
 
-__all__ = ["AGENT_METHODS", "CLIENT_METHODS", "PROTOCOL_VERSION"]
+__all__ = ["AGENT_METHODS", "CLIENT_METHODS", "PROTOCOL_METHODS", "PROTOCOL_VERSION"]

src/acp/experimental/v2/meta.py

@@ -37,4 +37,5 @@
     "elicitation_create": "elicitation/create",
     "elicitation_complete": "elicitation/complete",
 }
+PROTOCOL_METHODS = {"cancel_request": "$/cancel_request"}
 PROTOCOL_VERSION = 2

src/acp/meta.py

@@ -46,4 +46,5 @@
     "elicitation_create": "elicitation/create",
     "elicitation_complete": "elicitation/complete",
 }
+PROTOCOL_METHODS = {"cancel_request": "$/cancel_request"}
 PROTOCOL_VERSION = 1

tests/test_v2_schema.py

@@ -1,7 +1,7 @@
 import pytest
 from pydantic import ValidationError
 
-from acp.experimental.v2 import PROTOCOL_VERSION
+from acp.experimental.v2 import PROTOCOL_METHODS, PROTOCOL_VERSION
 from acp.experimental.v2.schema import (
     AgentMessageChunk,
     OtherSessionUpdate,
@@ -17,6 +17,7 @@ def test_v2_models_fill_protocol_discriminators() -> None:
     )
 
     assert PROTOCOL_VERSION == 2
+    assert PROTOCOL_METHODS == {"cancel_request": "$/cancel_request"}
     assert update.model_dump(by_alias=True, exclude_none=True) == {
         "sessionUpdate": "agent_message_chunk",
         "messageId": "message-1",