arize-phoenix 3.22.0__py3-none-any.whl → 3.24.0__py3-none-any.whl

{arize_phoenix-3.22.0.dist-info → arize_phoenix-3.24.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 3.22.0
+Version: 3.24.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -26,7 +26,7 @@ Requires-Dist: openinference-instrumentation-llama-index>=1.2.0
 Requires-Dist: openinference-instrumentation-openai>=0.1.4
 Requires-Dist: openinference-semantic-conventions>=0.1.5
 Requires-Dist: opentelemetry-exporter-otlp
-Requires-Dist: opentelemetry-proto
+Requires-Dist: opentelemetry-proto>=1.12.0
 Requires-Dist: opentelemetry-sdk
 Requires-Dist: pandas
 Requires-Dist: protobuf<6.0,>=3.20

{arize_phoenix-3.22.0.dist-info → arize_phoenix-3.24.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=x3T39dw3H8eGDej-Y0jaguyUerrE7bTqWl_cTqwt5aM,23
+phoenix/version.py,sha256=gjTZjq4DQfML3iAhAud_iXc-_X0aEBW4kipWwG8SoVs,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=SBO8BZg_CeQuH5LrSgzlQfqMModzirkQePdhnbLw7dE,4756
@@ -30,7 +30,7 @@ phoenix/experimental/evals/models/anthropic.py,sha256=BZmLvepkSMj_opCWsZoL34a3yA
 phoenix/experimental/evals/models/base.py,sha256=qOX36f4H1PA7YW3fckUsoh4apSz3bzURVtqlXw6Xcqo,6274
 phoenix/experimental/evals/models/bedrock.py,sha256=nVOXRZr-iDwHEINozpO2bqZR2KEeDHNyj6jgQPONQYs,8565
 phoenix/experimental/evals/models/litellm.py,sha256=YHZvwBHo4Y0_b589gjtwPN8N7sYgyIm9Hy5nnBiUtzA,4788
-phoenix/experimental/evals/models/openai.py,sha256=v2qkdFZc-u0ZAfxnV4v5UMWkMc9P3k4Gx1XaTnVFTz4,17922
+phoenix/experimental/evals/models/openai.py,sha256=UOM0GZLZ_kVWHvP1Z3mttt6DPnn876GmvK5GYtFTztQ,18164
 phoenix/experimental/evals/models/rate_limiters.py,sha256=L8xf7mc2d6axmWNjg2wwWIHWeTZ-tiI8m1GoBNFfJd0,10156
 phoenix/experimental/evals/models/vertex.py,sha256=3kj0tdwyCcgs39x1XnMgFoSbZwXvvBAPL7AwHYe-qIE,6236
 phoenix/experimental/evals/models/vertexai.py,sha256=nFN5isv1GPNsvUA4OxSnQd1hTMi-3BcxfDz1y7QcoA0,6189
@@ -60,7 +60,7 @@ phoenix/pointcloud/projectors.py,sha256=zO_RrtDYSv2rqVOfIP2_9Cv11Dc8EmcZR94xhFcB
 phoenix/pointcloud/umap_parameters.py,sha256=lJsEOrbSuSiqI7g4Yt6xj7kgYxEqoep4ZHWLr6VWBqw,1760
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/app.py,sha256=eUw5tygCYmsQ5TAJTM_ZwRKkxDD3-bMJSH-H9YpF0eA,7246
-phoenix/server/main.py,sha256=Su68-BSyN3wQm8od2TvT6iMOL84IvZdwtDCzsbMBmTk,9752
+phoenix/server/main.py,sha256=JoVQo4i2lcEQrXs59f5RiSI7csn-mQCs9JaqI7YTWVw,9798
 phoenix/server/prometheus.py,sha256=YQXwXZt3kxXN5JqGKDMH_jI9r4IL23xn6c8Bdrvg15A,2368
 phoenix/server/thread_server.py,sha256=dP6cm6Cf08jNhDA1TRlVZpziu1YgtPDmaeIJMm725eI,2154
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -81,7 +81,7 @@ phoenix/server/api/input_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 phoenix/server/api/routers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/routers/evaluation_handler.py,sha256=U7poPu9ezEEL5KEWXM-SX9LdvADpZiay1p6h4VAs08A,4058
 phoenix/server/api/routers/span_handler.py,sha256=IM4eiQ7GMU7JVyf0oMbrzEcItVudLy_CeYaWcwHOby4,2387
-phoenix/server/api/routers/trace_handler.py,sha256=0Y9QlePySJZyhKJLWM-m1e837HMZm5QcKR-iXshQdG4,2413
+phoenix/server/api/routers/trace_handler.py,sha256=2AQkYrio9r3DCmLMOL2vq3oB5ncnCm8XwEMLLUu4Eko,2433
 phoenix/server/api/routers/utils.py,sha256=M41BoH-fl37izhRuN2aX7lWm7jOC20A_3uClv9TVUUY,583
 phoenix/server/api/types/Cluster.py,sha256=R08ZKrLl1KK8colxHU57N5XIOTMUwg5ZI50ofPoxxSM,5618
 phoenix/server/api/types/DataQualityMetric.py,sha256=zRKsNvHBu-NdcsunuLhqFpZhi6ks-HMqA1PJD27jTak,590
@@ -138,7 +138,7 @@ phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=lO2wGA5XsftPg03rw_VcyaYf_4vegtlWbIT5ms4fA_c,1982
 phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/session/client.py,sha256=cYLRxV6YfCcWgZdJaStACVIhaN9j5p8S72tCKrT0RZ4,9612
+phoenix/session/client.py,sha256=zaAH7lTlRoWAgwoiCoL8giwG44FliDnEhaflHY5LoUM,10127
 phoenix/session/data_extractor.py,sha256=0Kf-2mKY_YbYoD2fZkAYpKdFgsXrC3OKQ5d2iZsGgAI,1947
 phoenix/session/evaluation.py,sha256=mR5HeBDmoS9oCb-X-2lpLCyMRoOgBufofbsjyEqCpMg,5390
 phoenix/session/session.py,sha256=P75qklh2HMgCCNiBghN3gjwRJiLIWzg1OshGt-fnzdM,24983
@@ -150,7 +150,7 @@ 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/otel.py,sha256=d8i8VPCJfkmT4f_o40kSNWFE94YddUPif0iQEjTQUCo,15368
 phoenix/trace/projects.py,sha256=2BwlNjFE-uwpqYtCu5YyBiYZk9wRPpM13vh3-Cv7GkA,2157
 phoenix/trace/schemas.py,sha256=6PVPnQIIDsWijwCwU19TGKG3xhFCYFY9K_5IqR5dWF4,5904
 phoenix/trace/span_evaluations.py,sha256=T67grfU71iANMM0g6dp2OQ_ZjtkON2viZz95_U9FLQA,12954
@@ -177,9 +177,9 @@ phoenix/utilities/deprecation.py,sha256=cFuTVvjSYyRlrdxdJewjJVieIEHPk30BukSRGRyd
 phoenix/utilities/error_handling.py,sha256=7b5rpGFj9EWZ8yrZK1IHvxB89suWk3lggDayUQcvZds,1946
 phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,222
 phoenix/utilities/project.py,sha256=qWsvKnG1oKhOFUowXf9qiOL2ia7jaFe_ijFFHEt8GJo,431
-phoenix/utilities/span_store.py,sha256=13UK0rE4wQd70yl___WsDRnH0ru-xErng9_Ml7zfEwE,978
-arize_phoenix-3.22.0.dist-info/METADATA,sha256=6Oog16HhrzCNfA5cS3JhS1OceuPvqiHL76Vzgz6hblE,29322
-arize_phoenix-3.22.0.dist-info/WHEEL,sha256=K0BPUNF1N3kQ9olb8aVEtkObePEjdr2JOLT1N83EVws,87
-arize_phoenix-3.22.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-3.22.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-3.22.0.dist-info/RECORD,,
+phoenix/utilities/span_store.py,sha256=puxYoYH_GmSeCWuGEoviD6QnrAZUgyRNO2BXG_au8_w,998
+arize_phoenix-3.24.0.dist-info/METADATA,sha256=YqPB3LwKvVjI50GDgMHjwElRpcFwMDp-udiLPAKwiZ0,29330
+arize_phoenix-3.24.0.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+arize_phoenix-3.24.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-3.24.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-3.24.0.dist-info/RECORD,,

{arize_phoenix-3.22.0.dist-info → arize_phoenix-3.24.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.24.0
+Generator: hatchling 1.24.2
 Root-Is-Purelib: true
 Tag: py3-none-any

phoenix/experimental/evals/models/openai.py

@@ -8,6 +8,7 @@ from typing import (
     Callable,
     Dict,
     List,
+    Mapping,
     Optional,
     Tuple,
     Union,
@@ -108,6 +109,8 @@ class OpenAIModel(BaseEvalModel):
     azure_deployment: Optional[str] = field(default=None)
     azure_ad_token: Optional[str] = field(default=None)
     azure_ad_token_provider: Optional[Callable[[], str]] = field(default=None)
+    default_headers: Optional[Mapping[str, str]] = field(default=None)
+    """Default headers required by AzureOpenAI"""
 
     # Deprecated fields
     model_name: Optional[str] = field(default=None)
@@ -193,6 +196,7 @@ class OpenAIModel(BaseEvalModel):
                 azure_ad_token_provider=azure_options.azure_ad_token_provider,
                 api_key=self.api_key,
                 organization=self.organization,
+                default_headers=self.default_headers,
             )
             self._async_client = self._openai.AsyncAzureOpenAI(
                 azure_endpoint=azure_options.azure_endpoint,
@@ -202,6 +206,7 @@ class OpenAIModel(BaseEvalModel):
                 azure_ad_token_provider=azure_options.azure_ad_token_provider,
                 api_key=self.api_key,
                 organization=self.organization,
+                default_headers=self.default_headers,
             )
             # return early since we don't need to check the model
             return

phoenix/server/api/routers/trace_handler.py

@@ -15,7 +15,7 @@ from starlette.status import HTTP_415_UNSUPPORTED_MEDIA_TYPE, HTTP_422_UNPROCESS
 
 from phoenix.core.traces import Traces
 from phoenix.storage.span_store import SpanStore
-from phoenix.trace.otel import decode
+from phoenix.trace.otel import decode_otlp_span
 from phoenix.utilities.project import get_project_name
 
 
@@ -55,6 +55,6 @@ class TraceHandler(HTTPEndpoint):
             project_name = get_project_name(resource_spans.resource.attributes)
             for scope_span in resource_spans.scope_spans:
                 for span in scope_span.spans:
-                    self.traces.put(decode(span), project_name=project_name)
+                    self.traces.put(decode_otlp_span(span), project_name=project_name)
                     await asyncio.sleep(0)
         return Response()

phoenix/server/main.py

@@ -35,7 +35,7 @@ from phoenix.trace.fixtures import (
     _get_trace_fixture_by_name,
     get_evals_from_fixture,
 )
-from phoenix.trace.otel import decode, encode
+from phoenix.trace.otel import decode_otlp_span, encode_span_to_otlp
 from phoenix.trace.span_json_decoder import json_string_to_span
 from phoenix.utilities.span_store import get_span_store, load_traces_data_from_store
 
@@ -200,7 +200,7 @@ if __name__ == "__main__":
         fixture_spans = list(
             # Apply `encode` here because legacy jsonl files contains UUIDs as strings.
             # `encode` removes the hyphens in the UUIDs.
-            decode(encode(json_string_to_span(json_span)))
+            decode_otlp_span(encode_span_to_otlp(json_string_to_span(json_span)))
             for json_span in _download_traces_fixture(
                 _get_trace_fixture_by_name(trace_dataset_name)
             )

phoenix/session/client.py

@@ -25,7 +25,7 @@ from phoenix.config import (
 from phoenix.session.data_extractor import TraceDataExtractor
 from phoenix.trace import Evaluations, TraceDataset
 from phoenix.trace.dsl import SpanQuery
-from phoenix.trace.otel import encode
+from phoenix.trace.otel import encode_span_to_otlp
 
 logger = logging.getLogger(__name__)
 
@@ -200,6 +200,19 @@ class Client(TraceDataExtractor):
             ).raise_for_status()
 
     def log_traces(self, trace_dataset: TraceDataset, project_name: Optional[str] = None) -> None:
+        """
+        Logs traces from a TraceDataset to the Phoenix server.
+
+        Args:
+            trace_dataset (TraceDataset): A TraceDataset instance with the traces to log to
+                the Phoenix server.
+            project_name (str, optional): The project name under which to log the evaluations.
+                This can be set using environment variables. If not provided, falls back to the
+                default project.
+
+        Returns:
+            None
+        """
         project_name = project_name or get_env_project_name()
         spans = trace_dataset.to_spans()
         otlp_spans = [
@@ -214,7 +227,7 @@ class Client(TraceDataExtractor):
                                 )
                             ]
                         ),
-                        scope_spans=[ScopeSpans(spans=[encode(span)])],
+                        scope_spans=[ScopeSpans(spans=[encode_span_to_otlp(span)])],
                     )
                 ],
             )

phoenix/trace/otel.py

@@ -55,7 +55,7 @@ TOOL_PARAMETERS = SpanAttributes.TOOL_PARAMETERS
 LLM_PROMPT_TEMPLATE_VARIABLES = SpanAttributes.LLM_PROMPT_TEMPLATE_VARIABLES
 
 
-def decode(otlp_span: otlp.Span) -> Span:
+def decode_otlp_span(otlp_span: otlp.Span) -> Span:
     trace_id = cast(TraceID, _decode_identifier(otlp_span.trace_id))
     span_id = cast(SpanID, _decode_identifier(otlp_span.span_id))
     parent_id = _decode_identifier(otlp_span.parent_span_id)
@@ -307,7 +307,7 @@ def _unflatten(
 _BILLION = 1_000_000_000  # for converting seconds to nanoseconds
 
 
-def encode(span: Span) -> otlp.Span:
+def encode_span_to_otlp(span: Span) -> otlp.Span:
     trace_id: bytes = _encode_identifier(span.context.trace_id)
     span_id: bytes = _encode_identifier(span.context.span_id)
     parent_span_id: bytes = _encode_identifier(span.parent_id)
@@ -451,6 +451,6 @@ def _encode_value(value: AttributeValue) -> AnyValue:
 
 
 __all__ = [
-    "encode",
-    "decode",
+    "encode_span_to_otlp",
+    "decode_otlp_span",
 ]

phoenix/utilities/span_store.py

@@ -3,7 +3,7 @@ from typing import Optional
 from phoenix.config import get_env_span_storage_type, get_storage_dir
 from phoenix.core.traces import Traces
 from phoenix.storage.span_store import SPAN_STORE_FACTORIES, SpanStore
-from phoenix.trace.otel import decode
+from phoenix.trace.otel import decode_otlp_span
 from phoenix.utilities.project import get_project_name
 
 
@@ -20,4 +20,4 @@ def load_traces_data_from_store(traces: Traces, span_store: SpanStore) -> None:
             project_name = get_project_name(resource_spans.resource.attributes)
             for scope_span in resource_spans.scope_spans:
                 for span in scope_span.spans:
-                    traces.put(decode(span), project_name=project_name)
+                    traces.put(decode_otlp_span(span), project_name=project_name)

phoenix/version.py

@@ -1 +1 @@
-__version__ = "3.22.0"
+__version__ = "3.24.0"