arize-phoenix 3.11.0__py3-none-any.whl → 3.12.0__py3-none-any.whl

phoenix/trace/__init__.py

@@ -1,6 +1,7 @@
 import contextlib
 from typing import Iterator
 
+from .projects import using_project
 from .span_evaluations import DocumentEvaluations, Evaluations, SpanEvaluations, TraceEvaluations
 from .trace_dataset import TraceDataset
 
@@ -10,6 +11,7 @@ __all__ = [
     "SpanEvaluations",
     "DocumentEvaluations",
     "TraceEvaluations",
+    "using_project",
 ]
 
 

phoenix/trace/projects.py

@@ -0,0 +1,56 @@
+import types
+from typing import Any, Callable, Optional, Type
+
+from openinference.semconv.resource import ResourceAttributes
+from opentelemetry.sdk import trace
+from opentelemetry.sdk.resources import Resource
+from wrapt import wrap_function_wrapper
+
+
+def project_override_wrapper(project_name: str) -> Callable[..., None]:
+    def wrapper(
+        wrapped: Any,
+        instance: "trace.ReadableSpan",
+        args: Any,
+        kwargs: Any,
+    ) -> None:
+        wrapped(*args, **kwargs)
+        instance._resource = Resource(
+            {
+                **instance._resource.attributes,
+                ResourceAttributes.PROJECT_NAME: project_name,
+            },
+            instance._resource.schema_url,
+        )
+
+    return wrapper
+
+
+class using_project:
+    """
+    A context manager that switches the project for all spans created within the context.
+
+    This is useful for managing projects in notebook environments, however this should not be used
+    in production environments or complicated OpenTelemetry setups, as dynamically modifying the
+    span resource can lead to unexpected behavior.
+    """
+
+    def __init__(self, project_name: str) -> None:
+        self.project_name = project_name
+
+    def __enter__(self) -> None:
+        self.unwrapped_init: Optional[Callable[..., None]] = trace.ReadableSpan.__init__
+        wrap_function_wrapper(
+            module="opentelemetry.sdk.trace",
+            name="ReadableSpan.__init__",
+            wrapper=project_override_wrapper(self.project_name),
+        )
+
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_value: Optional[BaseException],
+        traceback: Optional[types.TracebackType],
+    ) -> None:
+        setattr(trace.ReadableSpan, "__init__", self.unwrapped_init)
+        self.unwrapped_init = None

phoenix/version.py

@@ -1 +1 @@
-__version__ = "3.11.0"
+__version__ = "3.12.0"