Send Agent DSL from agent-builder deploys

tests/test_template_init.py

@@ -272,6 +272,89 @@ class TemplateInitTests(unittest.TestCase):
 
         self.assertIn("manifest_json must be a JSON object", result["error"])
 
+    def test_cp_deploy_tarball_posts_agent_dsl(self) -> None:
+        workspace = _FakeWorkspace()
+        prefix = "agents/demo-agent/"
+        files = {
+            "agent.py": "\n".join(
+                [
+                    "from a2a_pack import A2AAgent, NoAuth, RunContext, skill",
+                    "",
+                    "class DemoAgent(A2AAgent[None, NoAuth]):",
+                    '    name = "demo-agent"',
+                    '    description = "Demo agent"',
+                    '    version = "0.1.0"',
+                    "    auth_model = NoAuth",
+                    "",
+                    '    @skill(description="Echo text")',
+                    "    async def echo(self, ctx: RunContext[NoAuth], text: str) -> dict:",
+                    "        return {'ok': True, 'text': text}",
+                    "",
+                ]
+            ),
+            "a2a.yaml": "\n".join(
+                [
+                    "name: demo-agent",
+                    "version: 0.1.0",
+                    "entrypoint: agent:DemoAgent",
+                    "expose:",
+                    "  public: false",
+                    "",
+                ]
+            ),
+            "requirements.txt": "",
+        }
+        for rel, content in files.items():
+            workspace.write_bytes(prefix + rel, content.encode("utf-8"))
+        tools = build_tools(
+            ToolContext(
+                bucket="bucket",
+                settings=_settings(deploy_wait_timeout_s=0),
+                workspace=workspace,
+                cp_jwt="jwt-user",
+            )
+        )
+        deploy = _tool_by_name(tools, "cp_deploy_tarball")
+
+        async def _not_live(*args: object, **kwargs: object):
+            return False, {}
+
+        async def _no_latest(*args: object, **kwargs: object):
+            return None
+
+        _FakeAsyncClient.posts = []
+        with (
+            patch("agent_builder.tools.httpx.AsyncClient", _FakeAsyncClient),
+            patch("agent_builder.tools._wait_for_live_card", _not_live),
+            patch("agent_builder.tools._latest_cp_deployment", _no_latest),
+        ):
+            result = json.loads(
+                asyncio.run(
+                    deploy.ainvoke(
+                        {
+                            "name": "demo-agent",
+                            "version": "0.1.0",
+                            "public": False,
+                        }
+                    )
+                )
+            )
+
+        self.assertFalse(result["ok"])
+        self.assertEqual(len(_FakeAsyncClient.posts), 1)
+        post = _FakeAsyncClient.posts[0]
+        self.assertEqual(post["url"], "http://cp.test/v1/agents/from-tarball")
+        self.assertEqual(post["headers"]["authorization"], "bearer jwt-user")
+        data = post["data"]
+        self.assertEqual(data["name"], "demo-agent")
+        self.assertEqual(data["entrypoint"], "agent:DemoAgent")
+        self.assertEqual(data["public"], "false")
+        dsl = json.loads(data["agent_dsl"])
+        self.assertEqual(dsl["name"], "demo-agent")
+        self.assertEqual(dsl["version"], "0.1.0")
+        self.assertEqual(dsl["entrypoint"]["module"], "agent")
+        self.assertEqual(dsl["entrypoint"]["class_name"], "DemoAgent")
+
     def test_cp_compose_meta_agent_posts_manifest_and_records_state(self) -> None:
         workspace = _FakeWorkspace()
         tools = build_tools(
@@ -497,14 +580,23 @@ class _FakeAsyncClient:
         url: str,
         *,
         headers: dict[str, str],
-        json: dict[str, object],
+        json: dict[str, object] | None = None,
+        data: dict[str, object] | None = None,
+        files: dict[str, object] | None = None,
     ) -> _FakeResponse:
-        self.posts.append({"url": url, "headers": headers, "json": json})
+        self.posts.append({
+            "url": url,
+            "headers": headers,
+            "json": json,
+            "data": data,
+            "files": files,
+        })
+        payload = json or data or {}
         return _FakeResponse(
             201,
             {
-                "name": json["name"],
-                "version": json["version"],
+                "name": payload["name"],
+                "version": payload["version"],
                 "status": "building",
                 "expected_url": "https://report-meta.a2acloud.io",
                 "deployment_id": "dpl_123",