bobseares-2/customer-integration-engineer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

a2a-source-edit: write agent.py

Browse Source
This commit is contained in:
a2a-cloud
2026-07-13 03:26:50 +00:00
parent 16ed73e0f2
commit 7abca3998a
1 changed files with 59 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
agent.py
View File

@@ -42,6 +42,65 @@ APPROVAL_PHRASE = "I_APPROVE_THE_LISTED_STEPS"
OUTPUT_ROOT = "outputs/integrations" OUTPUT_ROOT = "outputs/integrations"
SECRET_WORDS = ("secret", "token", "password", "passwd", "apikey", "api_key", "authorization", "bearer") SECRET_WORDS = ("secret", "token", "password", "passwd", "apikey", "api_key", "authorization", "bearer")
_BOUNDED_OBJECT_SCHEMA: dict[str, Any] = {
"type": "object",
"maxProperties": 200,
"additionalProperties": True,
}
_BOUNDED_OBJECT_ARRAY_SCHEMA: dict[str, Any] = {
"type": "array",
"maxItems": 50,
"items": _BOUNDED_OBJECT_SCHEMA,
}
_INTEGRATION_REQUIREMENTS_SCHEMA: dict[str, Any] = {
"type": "object",
"required": ["integration_id", "title", "systems", "success_criteria"],
"additionalProperties": True,
"properties": {
"integration_id": {"type": "string", "pattern": "^[a-z][a-z0-9-]{2,63}$"},
"title": {"type": "string", "minLength": 1, "maxLength": 160},
"systems": {"type": "array", "minItems": 1, "maxItems": 12, "items": _BOUNDED_OBJECT_SCHEMA},
"auth_methods": {"type": "array", "maxItems": 12, "items": _BOUNDED_OBJECT_SCHEMA},
"data_contracts": {"type": "array", "maxItems": 20, "items": _BOUNDED_OBJECT_SCHEMA},
"rate_limits": {"type": "array", "maxItems": 20, "items": _BOUNDED_OBJECT_SCHEMA},
"environments": {"type": "array", "maxItems": 12, "items": _BOUNDED_OBJECT_SCHEMA},
"success_criteria": {"type": "array", "minItems": 1, "maxItems": 20, "items": _BOUNDED_OBJECT_SCHEMA},
"external_host_allowlist": {"type": "array", "maxItems": 50, "items": {"type": "string", "maxLength": 253}},
"repository_write_allowlist": {"type": "array", "maxItems": 20, "items": {"type": "string", "maxLength": 201}},
"dry_run": {"type": "boolean", "default": True},
},
}
_DESIGN_INPUT_SCHEMA: dict[str, Any] = {
"type": "object",
"required": ["request"],
"additionalProperties": False,
"properties": {"request": {"type": "object", "required": ["requirements"], "additionalProperties": False, "properties": {"requirements": _INTEGRATION_REQUIREMENTS_SCHEMA, "create_artifacts": {"type": "boolean", "default": True}, "dry_run": {"type": "boolean", "default": True}}}},
}
_GENERATE_INPUT_SCHEMA: dict[str, Any] = {
"type": "object",
"required": ["request"],
"additionalProperties": False,
"properties": {"request": {"type": "object", "required": ["requirements"], "additionalProperties": False, "properties": {"requirements": _INTEGRATION_REQUIREMENTS_SCHEMA, "connector_language": {"type": "string", "enum": ["python", "typescript"], "default": "python"}, "include_tests": {"type": "boolean", "default": True}, "dry_run": {"type": "boolean", "default": True}, "external_host_allowlist": {"type": "array", "maxItems": 50, "items": {"type": "string", "maxLength": 253}}, "repository_write_allowlist": {"type": "array", "maxItems": 20, "items": {"type": "string", "maxLength": 201}}}}},
}
_VALIDATE_INPUT_SCHEMA: dict[str, Any] = {
"type": "object",
"required": ["request"],
"additionalProperties": False,
"properties": {"request": {"type": "object", "required": ["integration_id", "contracts"], "additionalProperties": False, "properties": {"integration_id": {"type": "string", "pattern": "^[a-z][a-z0-9-]{2,63}$"}, "contracts": {"type": "array", "minItems": 1, "maxItems": 20, "items": _BOUNDED_OBJECT_SCHEMA}, "fail_on_drift": {"type": "boolean", "default": True}, "dry_run": {"type": "boolean", "default": True}}}},
}
_ACCEPTANCE_INPUT_SCHEMA: dict[str, Any] = {
"type": "object",
"required": ["request"],
"additionalProperties": False,
"properties": {"request": {"type": "object", "required": ["integration_id"], "additionalProperties": False, "properties": {"integration_id": {"type": "string", "pattern": "^[a-z][a-z0-9-]{2,63}$"}, "endpoint_tests": _BOUNDED_OBJECT_ARRAY_SCHEMA, "contract_fixtures": _BOUNDED_OBJECT_ARRAY_SCHEMA, "external_host_allowlist": {"type": "array", "maxItems": 50, "items": {"type": "string", "maxLength": 253}}, "allow_network": {"type": "boolean", "default": False}, "max_retries": {"type": "integer", "minimum": 0, "maximum": 3, "default": 1}, "timeout_seconds": {"type": "number", "minimum": 1.0, "maximum": 30.0, "default": 8.0}, "dry_run": {"type": "boolean", "default": True}}}},
}
_HANDOFF_INPUT_SCHEMA: dict[str, Any] = {
"type": "object",
"required": ["request"],
"additionalProperties": False,
"properties": {"request": {"type": "object", "required": ["integration_id", "title", "rollout_steps", "rollback_steps", "monitoring_checks"], "additionalProperties": False, "properties": {"integration_id": {"type": "string", "pattern": "^[a-z][a-z0-9-]{2,63}$"}, "title": {"type": "string", "minLength": 1, "maxLength": 160}, "rollout_steps": {"type": "array", "minItems": 1, "maxItems": 30, "items": {"type": "string", "maxLength": 8000}}, "rollback_steps": {"type": "array", "minItems": 1, "maxItems": 30, "items": {"type": "string", "maxLength": 8000}}, "monitoring_checks": {"type": "array", "minItems": 1, "maxItems": 30, "items": {"type": "string", "maxLength": 8000}}, "operator_notes": _BOUNDED_OBJECT_ARRAY_SCHEMA, "delivery_actions": {"type": "array", "maxItems": 20, "items": {"type": "string", "maxLength": 8000}}, "plan_digest": {"type": ["string", "null"], "maxLength": 128}, "approval_phrase": {"type": ["string", "null"], "maxLength": 64}, "approve_delivery": {"type": "boolean", "default": False}, "dry_run": {"type": "boolean", "default": True}}}},
}
class StrictModel(BaseModel): class StrictModel(BaseModel):
model_config = ConfigDict(extra="forbid", str_strip_whitespace=True) model_config = ConfigDict(extra="forbid", str_strip_whitespace=True)