Update agents.py

Author: bbatha

src/gradient/resources/agents/agents.py

@@ -680,7 +680,7 @@ def wait_until_ready(
                 if status in ("STATUS_FAILED", "STATUS_UNDEPLOYMENT_FAILED", "STATUS_DELETED"):
                     raise AgentDeploymentError(
                         f"Agent deployment failed with status: {status}",
-                        status=status or "UNKNOWN",
+                        status=status,
                     )
 
             # Check timeout
@@ -1264,7 +1264,7 @@ async def wait_until_ready(
                 if status in ("STATUS_FAILED", "STATUS_UNDEPLOYMENT_FAILED", "STATUS_DELETED"):
                     raise AgentDeploymentError(
                         f"Agent deployment failed with status: {status}",
-                        status=status or "UNKNOWN",
+                        status=status,
                     )
 
             # Check timeout