feat: add accessors for error payload

Author: guillaq

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "workflowai"
-version = "0.6.0.dev13"
+version = "0.6.0.dev14"
 description = ""
 authors = ["Guillaume Aquilina <guillaume@workflowai.com>"]
 readme = "README.md"

workflowai/core/domain/errors.py

@@ -169,5 +169,21 @@ def retry_after_delay_seconds(self) -> Optional[float]:
 
         return None
 
+    @property
+    def code(self) -> Optional[ErrorCode]:
+        return self.error.code
+
+    @property
+    def status_code(self) -> Optional[int]:
+        return self.error.status_code
+
+    @property
+    def message(self) -> str:
+        return self.error.message
+
+    @property
+    def details(self) -> Optional[dict[str, Any]]:
+        return self.error.details
+
 
 class InvalidGenerationError(WorkflowAIError): ...

workflowai/core/domain/errors_test.py

@@ -5,6 +5,7 @@
 from freezegun import freeze_time
 
 from workflowai.core.domain.errors import (
+    BaseError,
     WorkflowAIError,
     _retry_after_to_delay_seconds,  # pyright: ignore [reportPrivateUsage]
 )
@@ -33,3 +34,52 @@ def test_workflow_ai_error_404():
 )
 def test_retry_after_to_delay_seconds(retry_after: Optional[str], expected: Optional[float]):
     assert _retry_after_to_delay_seconds(retry_after) == expected
+
+
+def test_workflow_ai_error_code():
+    error = WorkflowAIError(
+        response=Mock(),
+        error=BaseError(
+            message="test",
+            status_code=404,
+            code="object_not_found",
+        ),
+    )
+    assert error.code == "object_not_found"
+
+
+def test_workflow_ai_error_status_code():
+    error = WorkflowAIError(
+        response=Mock(),
+        error=BaseError(
+            message="test",
+            status_code=404,
+            code="object_not_found",
+        ),
+    )
+    assert error.status_code == 404
+
+
+def test_workflow_ai_error_message():
+    error = WorkflowAIError(
+        response=Mock(),
+        error=BaseError(
+            message="test",
+            status_code=404,
+            code="object_not_found",
+        ),
+    )
+    assert error.message == "test"
+
+
+def test_workflow_ai_error_details():
+    error = WorkflowAIError(
+        response=Mock(),
+        error=BaseError(
+            message="test",
+            status_code=404,
+            code="object_not_found",
+            details={"test": "test"},
+        ),
+    )
+    assert error.details == {"test": "test"}