arize-phoenix 10.14.0__tar.gz → 11.0.0__tar.gz

{arize_phoenix-10.14.0 → arize_phoenix-11.0.0}/.gitignore

@@ -85,4 +85,6 @@ packages/phoenix-client/docs/build/
 packages/phoenix-evals/docs/build/
 
 # Deployment
-helm/charts/*.tgz
+helm/charts/*.tgz
+# mypy
+.mypy_cache

{arize_phoenix-10.14.0 → arize_phoenix-11.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 10.14.0
+Version: 11.0.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://arize.com/docs/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -33,7 +33,7 @@ Requires-Dist: httpx
 Requires-Dist: jinja2
 Requires-Dist: numpy!=2.0.0
 Requires-Dist: openinference-instrumentation>=0.1.32
-Requires-Dist: openinference-semantic-conventions>=0.1.17
+Requires-Dist: openinference-semantic-conventions>=0.1.20
 Requires-Dist: opentelemetry-exporter-otlp
 Requires-Dist: opentelemetry-proto>=1.12.0
 Requires-Dist: opentelemetry-sdk
@@ -58,6 +58,7 @@ Provides-Extra: container
 Requires-Dist: aiohttp; extra == 'container'
 Requires-Dist: anthropic>=0.49.0; extra == 'container'
 Requires-Dist: azure-identity; extra == 'container'
+Requires-Dist: boto3; extra == 'container'
 Requires-Dist: fast-hdbscan>=0.2.0; extra == 'container'
 Requires-Dist: google-generativeai; extra == 'container'
 Requires-Dist: numba>=0.60.0; extra == 'container'

{arize_phoenix-10.14.0 → arize_phoenix-11.0.0}/pyproject.toml

@@ -43,7 +43,7 @@ dependencies = [
   "opentelemetry-proto>=1.12.0",  # needed to avoid this issue: https://github.com/Arize-ai/phoenix/issues/2695
   "opentelemetry-exporter-otlp",
   "opentelemetry-semantic-conventions",
-  "openinference-semantic-conventions>=0.1.17",
+  "openinference-semantic-conventions>=0.1.20",
   "openinference-instrumentation>=0.1.32",
   "sqlalchemy[asyncio]>=2.0.4, <3",
   "alembic>=1.3.0, <2",
@@ -142,6 +142,7 @@ container = [
   "fast-hdbscan>=0.2.0",
   "numba>=0.60.0",  # https://github.com/astral-sh/uv/issues/6281
   "umap-learn",
+  "boto3",
 ]
 test = [
 ]

{arize_phoenix-10.14.0 → arize_phoenix-11.0.0}/src/phoenix/config.py

@@ -135,8 +135,11 @@ ENV_PHOENIX_DANGEROUSLY_DISABLE_MIGRATIONS = "PHOENIX_DANGEROUSLY_DISABLE_MIGRAT
 """
 Whether or not to disable migrations. Defaults to None / False.
 
-This should only be used by developers working on the Phoenix server that need to be
-switching between branches without having to run migrations.
+This should only be used by developers working on the Phoenix server that need
+to be switching between branches without having to run migrations.
+
+This can also be useful if a migration fails and you want to put the applicaiton
+in a running state.
 """
 
 # Phoenix server OpenTelemetry instrumentation environment variables

{arize_phoenix-10.14.0 → arize_phoenix-11.0.0}/src/phoenix/datetime_utils.py

@@ -28,7 +28,7 @@ def normalize_datetime(
     """
     if not isinstance(dt, datetime):
         return None
-    if dt.tzinfo is None or dt.tzinfo.utcoffset(dt) is None:
+    if not is_timezone_aware(dt):
         dt = dt.replace(tzinfo=tz if tz else _LOCAL_TIMEZONE)
     return dt.astimezone(timezone.utc)
 
@@ -106,3 +106,10 @@ def right_open_time_range(
         floor_to_minute(min_time) if min_time else None,
         floor_to_minute(max_time + timedelta(minutes=1)) if max_time else None,
     )
+
+
+def is_timezone_aware(dt: datetime) -> bool:
+    """
+    Returns True if the datetime is timezone-aware, False otherwise.
+    """
+    return dt.tzinfo is not None and dt.tzinfo.utcoffset(dt) is not None

{arize_phoenix-10.14.0 → arize_phoenix-11.0.0}/src/phoenix/db/bulk_inserter.py

@@ -8,6 +8,7 @@ from itertools import islice
 from time import perf_counter
 from typing import Any, Optional, cast
 
+from openinference.semconv.trace import SpanAttributes
 from typing_extensions import TypeAlias
 
 import phoenix.trace.v1 as pb
@@ -17,11 +18,19 @@ from phoenix.db.insertion.evaluation import (
     InsertEvaluationError,
     insert_evaluation,
 )
-from phoenix.db.insertion.helpers import DataManipulation, DataManipulationEvent
+from phoenix.db.insertion.helpers import (
+    DataManipulation,
+    DataManipulationEvent,
+    should_calculate_span_cost,
+)
 from phoenix.db.insertion.span import SpanInsertionEvent, insert_span
 from phoenix.db.insertion.span_annotation import SpanAnnotationQueueInserter
 from phoenix.db.insertion.trace_annotation import TraceAnnotationQueueInserter
 from phoenix.db.insertion.types import Insertables, Precursors
+from phoenix.server.daemons.span_cost_calculator import (
+    SpanCostCalculator,
+    SpanCostCalculatorQueueItem,
+)
 from phoenix.server.dml_event import DmlEvent, SpanInsertEvent
 from phoenix.server.types import CanPutItem, DbSessionFactory
 from phoenix.trace.schemas import Span
@@ -42,6 +51,7 @@ class BulkInserter:
         db: DbSessionFactory,
         *,
         event_queue: CanPutItem[DmlEvent],
+        span_cost_calculator: SpanCostCalculator,
         initial_batch_of_spans: Optional[Iterable[tuple[Span, str]]] = None,
         initial_batch_of_evaluations: Optional[Iterable[pb.Evaluation]] = None,
         sleep: float = 0.1,
@@ -78,6 +88,7 @@ class BulkInserter:
         self._retry_delay_sec = retry_delay_sec
         self._retry_allowance = retry_allowance
         self._queue_inserters = _QueueInserters(db, self._retry_delay_sec, self._retry_allowance)
+        self._span_cost_calculator = span_cost_calculator
 
     async def __aenter__(
         self,
@@ -175,6 +186,7 @@ class BulkInserter:
 
     async def _insert_spans(self, spans: list[tuple[Span, str]]) -> None:
         project_ids = set()
+        span_cost_calculator_queue: list[SpanCostCalculatorQueueItem] = []
         for i in range(0, len(spans), self._max_ops_per_transaction):
             try:
                 start = perf_counter()
@@ -198,6 +210,16 @@ class BulkInserter:
                             )
                         if result is not None:
                             project_ids.add(result.project_rowid)
+                            if should_calculate_span_cost(span.attributes):
+                                span_cost_calculator_queue.append(
+                                    SpanCostCalculatorQueueItem(
+                                        span_rowid=result.span_rowid,
+                                        trace_rowid=result.trace_rowid,
+                                        attributes=span.attributes,
+                                        span_start_time=span.start_time,
+                                    )
+                                )
+
                 if self._enable_prometheus:
                     from phoenix.server.prometheus import BULK_LOADER_INSERTION_TIME
 
@@ -209,6 +231,8 @@ class BulkInserter:
                     BULK_LOADER_EXCEPTIONS.inc()
                 logger.exception("Failed to insert spans")
         self._event_queue.put(SpanInsertEvent(tuple(project_ids)))
+        for item in span_cost_calculator_queue:
+            self._span_cost_calculator.put_nowait(item)
 
     async def _insert_evaluations(self, evaluations: list[pb.Evaluation]) -> None:
         for i in range(0, len(evaluations), self._max_ops_per_transaction):
@@ -292,3 +316,18 @@ class _QueueInserters:
     @_enqueue.register(Insertables.DocumentAnnotation)
     async def _(self, item: Precursors.DocumentAnnotation) -> None:
         await self._document_annotations.enqueue(item)
+
+
+LLM_MODEL_NAME = SpanAttributes.LLM_MODEL_NAME
+LLM_PROVIDER = SpanAttributes.LLM_PROVIDER
+LLM_TOKEN_COUNT_COMPLETION = SpanAttributes.LLM_TOKEN_COUNT_COMPLETION
+LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO = SpanAttributes.LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO
+LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING = (
+    SpanAttributes.LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING
+)
+LLM_TOKEN_COUNT_PROMPT = SpanAttributes.LLM_TOKEN_COUNT_PROMPT
+LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO = SpanAttributes.LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO
+LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ = SpanAttributes.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ
+LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE = (
+    SpanAttributes.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE
+)