arize-phoenix 3.22.0__tar.gz → 3.24.0__tar.gz

{arize_phoenix-3.22.0 → arize_phoenix-3.24.0}/PKG-INFO

@@ -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 → arize_phoenix-3.24.0}/pyproject.toml

@@ -43,7 +43,7 @@ dependencies = [
   "tqdm",
   "requests",
   "opentelemetry-sdk",
-  "opentelemetry-proto",
+  "opentelemetry-proto>=1.12.0",  # needed to avoid this issue: https://github.com/Arize-ai/phoenix/issues/2695
   "opentelemetry-exporter-otlp",
   "openinference-semantic-conventions>=0.1.5",
   "openinference-instrumentation",

{arize_phoenix-3.22.0 → arize_phoenix-3.24.0}/src/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

{arize_phoenix-3.22.0 → arize_phoenix-3.24.0}/src/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()

{arize_phoenix-3.22.0 → arize_phoenix-3.24.0}/src/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)
             )

{arize_phoenix-3.22.0 → arize_phoenix-3.24.0}/src/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)])],
                     )
                 ],
             )

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

{arize_phoenix-3.22.0 → arize_phoenix-3.24.0}/src/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)

arize_phoenix-3.24.0/src/phoenix/version.py

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

arize_phoenix-3.22.0/src/phoenix/version.py

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