diff --git a/agent.py b/agent.py
index 2ce36e3..505ddbd 100644
--- a/agent.py
+++ b/agent.py
@@ -397,6 +397,13 @@ class CustomerIntegrationEngineerConfig(StrictModel):
     default_dry_run: bool = True
 
 
+_DESIGN_OUTPUT_SCHEMA = DesignIntegrationResult.model_json_schema()
+_GENERATE_OUTPUT_SCHEMA = GenerateConnectorResult.model_json_schema()
+_VALIDATE_OUTPUT_SCHEMA = ValidateContractResult.model_json_schema()
+_ACCEPTANCE_OUTPUT_SCHEMA = RunAcceptanceResult.model_json_schema()
+_HANDOFF_OUTPUT_SCHEMA = PrepareHandoffResult.model_json_schema()
+
+
 class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, NoAuth]):
     name = "customer-integration-engineer"
     description = (