arize-phoenix 3.11.1__tar.gz → 3.12.0__tar.gz

{arize_phoenix-3.11.1 → arize_phoenix-3.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 3.11.1
+Version: 3.12.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-3.11.1 → arize_phoenix-3.12.0}/src/phoenix/server/app.py

@@ -87,6 +87,7 @@ class HeadersMiddleware(BaseHTTPMiddleware):
     ) -> Response:
         response = await call_next(request)
         response.headers["x-colab-notebook-cache-control"] = "no-cache"
+        response.headers["Cache-Control"] = "no-store"
         return response
 
 

{arize_phoenix-3.11.1 → arize_phoenix-3.12.0}/src/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",
 ]
 
 

arize_phoenix-3.12.0/src/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

arize_phoenix-3.12.0/src/phoenix/version.py

@@ -0,0 +1 @@
+__version__ = "3.12.0"

arize_phoenix-3.11.1/src/phoenix/version.py

@@ -1 +0,0 @@
-__version__ = "3.11.1"