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

{arize_phoenix-3.11.1.dist-info → arize_phoenix-3.12.0.dist-info}/METADATA

@@ -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.dist-info → arize_phoenix-3.12.0.dist-info}/RECORD

@@ -4,7 +4,7 @@ phoenix/datetime_utils.py,sha256=D955QLrkgrrSdUM6NyqbCeAu2SMsjhR5rHVQEsVUdng,277
 phoenix/exceptions.py,sha256=X5k9ipUDfwSCwZB-H5zFJLas86Gf9tAx0W4l5TZxp5k,108
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=f6AeyKTuOpy9RCcTCjVH3gx5nYZhbTMFOuv1WSUOB5o,4992
-phoenix/version.py,sha256=UuKR3QjWk9sJtn2huDFsPdSgtx1CgiEl2B4BUtVRd58,23
+phoenix/version.py,sha256=Mf-gTb-tQiknxbfFkhQrqGBahlm5T1zJn8PDBgM9NiI,23
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=C-kDATyJEgP-oqYVKOiQM76Ljs66F6VZdT93_b8kTGk,4725
@@ -55,7 +55,7 @@ phoenix/pointcloud/pointcloud.py,sha256=4zAIkKs2xOUbchpj4XDAV-iPMXrfAJ15TG6rlIYG
 phoenix/pointcloud/projectors.py,sha256=zO_RrtDYSv2rqVOfIP2_9Cv11Dc8EmcZR94xhFcBYPU,1057
 phoenix/pointcloud/umap_parameters.py,sha256=lJsEOrbSuSiqI7g4Yt6xj7kgYxEqoep4ZHWLr6VWBqw,1760
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=MCXt9y0kDrOieAvcIr4Ex76l6lZMs_ffjKnPSKU30Mg,6719
+phoenix/server/app.py,sha256=KMHQsy8Sypp8Ua6UrWvqByLt2FHXoKirnkakf-90JbU,6774
 phoenix/server/main.py,sha256=gGaQ0jqbsiGG1Enlr7UwL5Woc1k4kJyZhTa6OplwxgM,9090
 phoenix/server/thread_server.py,sha256=dP6cm6Cf08jNhDA1TRlVZpziu1YgtPDmaeIJMm725eI,2154
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -136,12 +136,13 @@ phoenix/session/client.py,sha256=9qlybwGGdPHql7n6Mye8feuh5_5AMq34xSajvcPn2yg,585
 phoenix/session/data_extractor.py,sha256=0Kf-2mKY_YbYoD2fZkAYpKdFgsXrC3OKQ5d2iZsGgAI,1947
 phoenix/session/evaluation.py,sha256=YCv1XkWHi7vM_W5V7rorrrAxadv78wuMPeCVJvf5-oE,5444
 phoenix/session/session.py,sha256=1RV6FBNqNDDrPFYU2qLffmI4RCRKy22KCLZFcH9oqJA,24556
-phoenix/trace/__init__.py,sha256=qMipNV47ahKsOOz8uQpoI4VnNRnjc5uh_0cJpL1ABvY,706
+phoenix/trace/__init__.py,sha256=DOuudYVyWlhUnIy-E_aqvJ6ulmk_q7ghuctuXQCGCJA,763
 phoenix/trace/errors.py,sha256=wB1z8qdPckngdfU-TORToekvg3344oNFAA83_hC2yFY,180
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
 phoenix/trace/exporter.py,sha256=vh2RO1CpP143HxIX94KV0qks8p1x66RE3Tgf8kcBCCg,4519
 phoenix/trace/fixtures.py,sha256=HOjuYOB_xtR7JUeLz7WpEroiGj4E5_SxVLSjBYUy8RQ,7055
 phoenix/trace/otel.py,sha256=mpuj_eOdBUPAcslpxk8XZflb9kkzpMJ8X9acJS8ACPA,15322
+phoenix/trace/projects.py,sha256=uIicJWWBAVwiRY-8EH2YlQmCUzsdspTxsz-ceK35m8E,1826
 phoenix/trace/schemas.py,sha256=6PVPnQIIDsWijwCwU19TGKG3xhFCYFY9K_5IqR5dWF4,5904
 phoenix/trace/span_evaluations.py,sha256=T67grfU71iANMM0g6dp2OQ_ZjtkON2viZz95_U9FLQA,12954
 phoenix/trace/span_json_decoder.py,sha256=IAFakPRqSMYxTPKYFMiXYxm7U-FipdN8_xbvapDS0Qc,3131
@@ -165,8 +166,8 @@ phoenix/trace/v1/evaluation_pb2.pyi,sha256=cCbbx06gwQmaH14s3J1X25TtaARh-k1abbxQd
 phoenix/utilities/__init__.py,sha256=8w1Ivw0KO9YKWrhcdnO73cSVqP9VHAp0pSfsi_oDiuQ,672
 phoenix/utilities/error_handling.py,sha256=7b5rpGFj9EWZ8yrZK1IHvxB89suWk3lggDayUQcvZds,1946
 phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,222
-arize_phoenix-3.11.1.dist-info/METADATA,sha256=CxX1LnZOSawfGoh1H2Q222Rw7PWanWDpxUUzbAKWMAo,29080
-arize_phoenix-3.11.1.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-arize_phoenix-3.11.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-3.11.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-3.11.1.dist-info/RECORD,,
+arize_phoenix-3.12.0.dist-info/METADATA,sha256=o2x8-y_lowmRbrLRRyau-RpB5fgtc4-l9EQ_gFhZe9Q,29080
+arize_phoenix-3.12.0.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+arize_phoenix-3.12.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-3.12.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-3.12.0.dist-info/RECORD,,

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
 
 

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.1"
+__version__ = "3.12.0"