refactor: use canonical SGP error categories

Require the released tracing SDK types so Agentex no longer maintains a duplicate ownership taxonomy that can drift.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: jshaikScale

adk/pyproject.toml

@@ -53,7 +53,7 @@ dependencies = [
     "pydantic-ai-slim>=1.0,<2",
     "langgraph-checkpoint>=2.0.0",
     "scale-gp>=0.1.0a59",
-    "scale-gp-beta>=0.2.0",
+    "scale-gp-beta>=0.5.0",
     "mcp>=1.4.1",
     # Observability
     "ddtrace>=3.13.0",

src/agentex/lib/core/tracing/__init__.py

@@ -2,6 +2,7 @@
 from agentex.lib.core.tracing.trace import Trace, AsyncTrace
 from agentex.lib.core.tracing.tracer import Tracer, AsyncTracer
 from agentex.lib.core.tracing.span_error import (
+    ErrorCategory,
     PlatformError,
     ApplicationError,
     CategorizedError,
@@ -21,6 +22,7 @@
     "CategorizedError",
     "ApplicationError",
     "PlatformError",
+    "ErrorCategory",
     "AsyncSpanQueue",
     "get_default_span_queue",
     "shutdown_default_span_queue",

src/agentex/lib/core/tracing/span_error.py

@@ -1,6 +1,13 @@
 from __future__ import annotations
 
-from typing import Any, Literal, cast
+from typing import Any, cast
+
+from scale_gp_beta.lib.tracing import (
+    PlatformError as PlatformError,
+    ApplicationError as ApplicationError,
+    CategorizedError,
+)
+from scale_gp_beta.lib.tracing.types import ErrorCategory
 
 from agentex.types.span import Span
 
@@ -13,36 +20,10 @@
 # SGP and agentex-native span stores.
 SPAN_ERROR_KEY = "__error__"
 
-ErrorCategory = Literal["application", "platform", "unknown"]
 ERROR_CATEGORY_UNKNOWN: ErrorCategory = "unknown"
 _ERROR_CATEGORIES = frozenset({"application", "platform", "unknown"})
 
 
-class CategorizedError(Exception):
-    """Base class for failures with known operational ownership.
-
-    Use ``ApplicationError`` for failures owned by agent or caller code, such
-    as business logic, user input, tools, or application configuration. Use
-    ``PlatformError`` only at a known Agentex/SGP-owned boundary, such as
-    managed runtime, tracing, persistence, or platform networking. Leave
-    unclassified failures as ordinary exceptions so they remain ``unknown``.
-    """
-
-    error_category: ErrorCategory = ERROR_CATEGORY_UNKNOWN
-
-
-class ApplicationError(CategorizedError):
-    """Failure owned by the agent application or its caller."""
-
-    error_category: ErrorCategory = "application"
-
-
-class PlatformError(CategorizedError):
-    """Failure owned by Agentex/SGP or a platform-managed dependency."""
-
-    error_category: ErrorCategory = "platform"
-
-
 def _normalize_error_category(value: object) -> ErrorCategory | None:
     if isinstance(value, str):
         normalized = value.strip().lower()

tests/lib/core/tracing/test_span_error.py

@@ -6,13 +6,19 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
+from scale_gp_beta.lib.tracing import (
+    PlatformError as SGPPlatformError,
+    ApplicationError as SGPApplicationError,
+    CategorizedError as SGPCategorizedError,
+)
 
 from agentex.types.span import Span
 from agentex.lib.core.tracing.trace import Trace, AsyncTrace
 from agentex.lib.core.tracing.span_error import (
     SPAN_ERROR_KEY,
     PlatformError,
     ApplicationError,
+    CategorizedError,
     get_span_error,
     set_span_error,
 )
@@ -36,6 +42,11 @@ def _make_span(data=None) -> Span:
 
 
 class TestSpanErrorHelpers:
+    def test_uses_canonical_sgp_error_types(self):
+        assert CategorizedError is SGPCategorizedError
+        assert ApplicationError is SGPApplicationError
+        assert PlatformError is SGPPlatformError
+
     def test_set_then_get_on_none_data(self):
         span = _make_span(data=None)
         set_span_error(span, ValueError("boom"))