From 541fb6928426956117ef59adbfea18004a8e9609 Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Sat, 18 Jul 2026 09:24:52 +0000 Subject: [PATCH] a2a-source-edit: write agent.py --- agent.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/agent.py b/agent.py index 86f6523..76f9cde 100644 --- a/agent.py +++ b/agent.py @@ -47,6 +47,10 @@ class LaunchCheckStudioV1Config(BaseModel): max_response_bytes: int = Field(default=MAX_RESPONSE_BYTES, ge=32_000, le=1_000_000) +class LaunchCheckStudioV1State(BaseModel): + storage: str = "managed-postgres" + + @dataclass(frozen=True) class FetchResult: final_url: str @@ -112,6 +116,7 @@ class LaunchCheckStudioV1(A2AAgent[LaunchCheckStudioV1Config, PlatformUserAuth]) auth_model = PlatformUserAuth state = State.DURABLE + state_model = LaunchCheckStudioV1State resources = Resources(cpu="500m", memory="512Mi", max_runtime_seconds=120) egress = EgressPolicy(deny_internet_by_default=False) tools_used = ("httpx", "psycopg", "postgres") @@ -276,7 +281,7 @@ async def _validate_public_url(raw_url: str) -> dict[str, Any]: netloc = host if parsed.port: netloc = f"{host}:{parsed.port}" - normalized = urlunparse((parsed.scheme.lower(), netloc, parsed.path or "/", "", parsed.query, "")) + normalized = urlunparse((parsed.scheme.lower(), netloc, parsed.path or "", "", parsed.query, "")) return {"ok": True, "url": normalized}