From 2c2054903241c44983baeed1929b7a6cddccb071 Mon Sep 17 00:00:00 2001
From: a2a-cloud <noreply@a2acloud.io>
Date: Mon, 13 Jul 2026 11:13:07 +0000
Subject: [PATCH] a2a-source-edit: write agent.py

---
 agent.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/agent.py b/agent.py
index 5962b82..0f298a0 100644
--- a/agent.py
+++ b/agent.py
@@ -42,12 +42,13 @@ APPROVAL_PHRASE = "I_APPROVE_THE_LISTED_STEPS"
 OUTPUT_ROOT = "outputs/integrations"
 SECRET_WORDS = ("secret", "token", "password", "passwd", "apikey", "api_key", "authorization", "bearer")
 
-_PUBLIC_REQUEST_SCHEMA: dict[str, Any] = {
-    "type": "object",
-    "required": ["request"],
-    "additionalProperties": False,
-    "properties": {"request": {"type": "object", "additionalProperties": True}},
-}
+def _tool_input_schema(request_model: type[BaseModel]) -> dict[str, Any]:
+    return {
+        "type": "object",
+        "required": ["request"],
+        "additionalProperties": False,
+        "properties": {"request": request_model.model_json_schema()},
+    }
 
 
 class StrictModel(BaseModel):