fix(sdk): enforce agent registration integration type

elydora/async_client.py

@@ -11,6 +11,7 @@ import aiohttp
 
 from .crypto import compute_chain_hash, compute_payload_hash, sign_eor
 from .errors import ElydoraError
+from .integration_types import require_integration_type
 from .types import (
     AuditQueryResponse,
     AuthLoginResponse,
@@ -27,6 +28,7 @@ from .types import (
     GetOperationResponse,
     HealthResponse,
     IssueApiTokenResponse,
+    IntegrationType,
     ListAdminEventsResponse,
     ListMembersResponse,
     ListWebhooksResponse,
@@ -248,6 +250,7 @@ class AsyncElydoraClient:
 
     async def register_agent(self, request: RegisterAgentRequest) -> RegisterAgentResponse:
         """Register a new agent with the organization."""
+        require_integration_type(request.get("integration_type"))
         return await self._request("POST", "/v1/agents/register", json_body=request)
 
     async def get_agent(self, agent_id: str) -> GetAgentResponse:
@@ -266,8 +269,9 @@ class AsyncElydoraClient:
         """Unfreeze an agent."""
         return await self._request("POST", f"/v1/agents/{agent_id}/unfreeze", json_body={"reason": reason})
 
-    async def update_agent(self, agent_id: str, integration_type: str) -> UpdateAgentResponse:
+    async def update_agent(self, agent_id: str, integration_type: IntegrationType) -> UpdateAgentResponse:
         """Update an agent's integration type."""
+        require_integration_type(integration_type)
         return await self._request("PATCH", f"/v1/agents/{agent_id}", json_body={"integration_type": integration_type})
 
     async def delete_agent(self, agent_id: str) -> DeleteAgentResponse: