edgeone 1.5.9 → 1.6.1

package/edgeone-dist/pages/observability-python/tracer.py

@@ -0,0 +1,165 @@
+"""User-facing tracer API for context.tracer."""
+from __future__ import annotations
+
+import asyncio
+from typing import Any, Awaitable, Callable, TypeVar
+
+from opentelemetry import context as otel_context
+from opentelemetry import trace as otel_trace
+from opentelemetry.trace import Span, StatusCode
+
+T = TypeVar("T")
+
+# LangGraph 通过抛出这些异常实现 graph 暂停 / 控制流跳转，不属于业务错误。
+# 命中后 user-tracer / exporter 都会把 span 状态从 ERROR 归一回 UNSET，
+# 避免在 APM/langfuse 上误报。GeneratorExit / CancelledError 是 interrupt 触发
+# 后流被拆除 / 取消时的 teardown 信号，同样不是业务错误，一并纳入。
+_INTERRUPT_EXCEPTION_NAMES = frozenset({
+    "GraphInterrupt",
+    "NodeInterrupt",
+    "ParentCommand",
+    "GeneratorExit",
+    "CancelledError",
+})
+
+
+def _is_interrupt_exception(err: BaseException) -> bool:
+    """Match LangGraph control-flow exceptions by class hierarchy name.
+
+    Walk the MRO so subclasses of GraphInterrupt etc. are also recognized,
+    while keeping it strict enough to avoid catching arbitrary user errors
+    that merely contain the substring "interrupt".
+    """
+    for cls in type(err).__mro__:
+        if cls.__name__ in _INTERRUPT_EXCEPTION_NAMES:
+            return True
+    return False
+
+
+class TracerSpan:
+    """Wrapper around an OTel Span with a simplified API."""
+
+    def __init__(self, span: Span) -> None:
+        self._span = span
+
+    @property
+    def span_id(self) -> str:
+        return format(self._span.get_span_context().span_id, "016x")
+
+    def set_attribute(self, key: str, value: Any) -> None:
+        self._span.set_attribute(key, value)
+
+    def set_attributes(self, attributes: dict[str, Any]) -> None:
+        self._span.set_attributes(attributes)
+
+    def event(self, name: str, attributes: dict[str, Any] | None = None) -> None:
+        self._span.add_event(name, attributes=attributes)
+
+    def end(self) -> None:
+        self._span.end()
+
+
+class AgentTracer:
+    """User-facing tracer, set on ctx.tracer."""
+
+    def __init__(self, tracer_provider=None) -> None:
+        provider = tracer_provider or otel_trace.get_tracer_provider()
+        self._tracer = provider.get_tracer("edgeone-agent-user", "0.1.0")
+
+    def start_span(
+        self, name: str, attributes: dict[str, Any] | None = None
+    ) -> TracerSpan:
+        parent_ctx = otel_context.get_current()
+        span = self._tracer.start_span(name, attributes=attributes or {}, context=parent_ctx)
+        return TracerSpan(span)
+
+    async def span(
+        self,
+        name: str,
+        fn: Callable[[TracerSpan], Awaitable[T] | T],
+        attributes: dict[str, Any] | None = None,
+    ) -> T:
+        parent_ctx = otel_context.get_current()
+        span = self._tracer.start_span(name, attributes=attributes or {}, context=parent_ctx)
+        span_ctx = otel_trace.set_span_in_context(span, parent_ctx)
+        token = otel_context.attach(span_ctx)
+        try:
+            result = fn(TracerSpan(span))
+            if asyncio.iscoroutine(result):
+                result = await result
+            span.end()
+            return result  # type: ignore[return-value]
+        except BaseException as err:
+            # LangGraph 的 interrupt() / Command 跳转通过抛异常实现，属于正常控制流，
+            # 不应记为 ERROR；但仍打标 agent.interrupt=true 方便平台筛选。
+            if _is_interrupt_exception(err):
+                try:
+                    span.set_attribute("agent.interrupt", True)
+                    span.set_attribute("agent.interrupt.type", type(err).__name__)
+                except Exception:
+                    pass
+            else:
+                span.record_exception(err)
+                span.set_status(StatusCode.ERROR, str(err))
+            span.end()
+            raise
+        finally:
+            otel_context.detach(token)
+
+    def event(self, name: str, attributes: dict[str, Any] | None = None) -> None:
+        active_span = otel_trace.get_current_span()
+        if active_span:
+            active_span.add_event(name, attributes=attributes)
+
+    def set_attributes(self, attributes: dict[str, Any]) -> None:
+        active_span = otel_trace.get_current_span()
+        if active_span:
+            active_span.set_attributes(attributes)
+
+    def record_exception(
+        self, error: Exception, attributes: dict[str, Any] | None = None
+    ) -> None:
+        active_span = otel_trace.get_current_span()
+        if active_span:
+            active_span.record_exception(error)
+            active_span.set_status(StatusCode.ERROR, str(error))
+            if attributes:
+                active_span.set_attributes(attributes)
+
+
+class NoOpTracerSpan:
+    @property
+    def span_id(self) -> str:
+        return "0" * 16
+
+    def set_attribute(self, key: str, value: Any) -> None:
+        pass
+
+    def set_attributes(self, attributes: dict[str, Any]) -> None:
+        pass
+
+    def event(self, name: str, attributes: dict[str, Any] | None = None) -> None:
+        pass
+
+    def end(self) -> None:
+        pass
+
+
+class NoOpTracer:
+    def start_span(self, name: str, attributes: dict[str, Any] | None = None) -> NoOpTracerSpan:
+        return NoOpTracerSpan()
+
+    async def span(self, name: str, fn: Callable, attributes: dict[str, Any] | None = None):
+        result = fn(NoOpTracerSpan())
+        if asyncio.iscoroutine(result):
+            return await result
+        return result
+
+    def event(self, name: str, attributes: dict[str, Any] | None = None) -> None:
+        pass
+
+    def set_attributes(self, attributes: dict[str, Any]) -> None:
+        pass
+
+    def record_exception(self, error: Exception, attributes: dict[str, Any] | None = None) -> None:
+        pass

package/edgeone-dist/pages/templates/agent-python/__init__.py

@@ -0,0 +1,11 @@
+# src/pages/builder/templates/agent-python/__init__.py
+from .memory import (  # noqa: F401
+    ConversationMemory,
+    Message,
+    ConversationMeta,
+    MemoryError,
+    MemoryValidationError,
+    MemoryNotFoundError,
+    MemoryQuotaExceededError,
+    MemoryStorageError,
+)