arize-phoenix 11.3.0__tar.gz → 11.4.0__tar.gz

{arize_phoenix-11.3.0 → arize_phoenix-11.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 11.3.0
+Version: 11.4.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://arize.com/docs/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -43,6 +43,7 @@ Requires-Dist: protobuf<6.0,>=4.25.8
 Requires-Dist: psutil
 Requires-Dist: pyarrow
 Requires-Dist: pydantic>=2.1.0
+Requires-Dist: python-dateutil
 Requires-Dist: python-multipart
 Requires-Dist: scikit-learn
 Requires-Dist: scipy
@@ -108,6 +109,7 @@ Requires-Dist: tabulate; extra == 'dev'
 Requires-Dist: tox-uv==1.11.3; extra == 'dev'
 Requires-Dist: tox==4.18.1; extra == 'dev'
 Requires-Dist: types-jsonschema; extra == 'dev'
+Requires-Dist: types-python-dateutil; extra == 'dev'
 Requires-Dist: types-tabulate; extra == 'dev'
 Requires-Dist: uv==0.7.12; extra == 'dev'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'dev'

{arize_phoenix-11.3.0 → arize_phoenix-11.4.0}/pyproject.toml

@@ -59,6 +59,7 @@ dependencies = [
   "authlib",
   "arize-phoenix-client",
   "email-validator",
+  "python-dateutil",
 ]
 dynamic = ["version"]
 
@@ -103,6 +104,7 @@ dev = [
   "uvloop; platform_system != 'Windows'",
   "grpc-interceptor[testing]",
   "types-jsonschema",
+  "types-python-dateutil",
 ]
 embeddings = [
   "fast-hdbscan>=0.2.0",

{arize_phoenix-11.3.0 → arize_phoenix-11.4.0}/src/phoenix/db/facilitator.py

@@ -406,17 +406,6 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
     with open(_COST_MODEL_MANIFEST) as f:
         manifest = json.load(f)
 
-    # Define all supported token types with their prompt/non-prompt classification
-    # This determines how tokens are categorized for billing purposes
-    token_types: list[_TokenTypeKey] = [
-        _TokenTypeKey("input", True),  # Standard input tokens
-        _TokenTypeKey("cache_write", True),  # Tokens written to cache
-        _TokenTypeKey("cache_read", True),  # Tokens read from cache
-        _TokenTypeKey("output", False),  # Generated output tokens
-        _TokenTypeKey("audio", True),  # Audio input tokens
-        _TokenTypeKey("audio", False),  # Audio output tokens
-    ]
-
     async with db() as session:
         # Fetch all existing built-in models with their token prices eagerly loaded
         # Using .unique() to deduplicate models when multiple token prices are joined
@@ -435,9 +424,8 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
         seen_names: set[str] = set()
         seen_patterns: set[tuple[re.Pattern[str], str]] = set()
 
-        # Process each model in the manifest
-        for model_data in manifest:
-            name = str(model_data.get("model") or "").strip()
+        for model_data in manifest["models"]:
+            name = str(model_data.get("name") or "").strip()
             if not name:
                 logger.warning("Skipping model with empty name in manifest")
                 continue
@@ -445,7 +433,7 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
                 logger.warning(f"Skipping model '{name}' with duplicate name in manifest")
                 continue
             seen_names.add(name)
-            regex = str(model_data.get("regex") or "").strip()
+            regex = str(model_data.get("name_pattern") or "").strip()
             try:
                 pattern = re.compile(regex)
             except re.error as e:
@@ -460,7 +448,7 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
             seen_patterns.add((pattern, provider))
             # Remove model from built_in_models dict (for cleanup tracking)
             # or create new model if not found
-            model = built_in_models.pop(model_data["model"], None)
+            model = built_in_models.pop(model_data["name"], None)
             if model is None:
                 # Create new built-in model from manifest data
                 model = models.GenerativeModel(
@@ -477,24 +465,27 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
 
             # Create lookup table for existing token prices by (token_type, is_prompt)
             # Using pop() during iteration allows us to track which prices are no longer needed
-            token_prices = {
+            existing_token_prices = {
                 _TokenTypeKey(token_price.token_type, token_price.is_prompt): token_price
                 for token_price in model.token_prices
             }
 
             # Synchronize token prices for all supported token types
-            for token_type, is_prompt in token_types:
+            for manifest_token_price in model_data["token_prices"]:
                 # Skip if this token type has no rate in the manifest
-                if not (base_rate := model_data.get(token_type)):
+                if not (base_rate := manifest_token_price.get("base_rate")):
                     continue
 
-                key = _TokenTypeKey(token_type, is_prompt)
+                key = _TokenTypeKey(
+                    manifest_token_price["token_type"],
+                    manifest_token_price["is_prompt"],
+                )
                 # Remove from tracking dict and get existing price (if any)
-                if not (token_price := token_prices.pop(key, None)):
+                if not (token_price := existing_token_prices.pop(key, None)):
                     # Create new token price if it doesn't exist
                     token_price = models.TokenPrice(
-                        token_type=token_type,
-                        is_prompt=is_prompt,
+                        token_type=manifest_token_price["token_type"],
+                        is_prompt=manifest_token_price["is_prompt"],
                         base_rate=base_rate,
                     )
                     model.token_prices.append(token_price)
@@ -504,7 +495,7 @@ async def _ensure_model_costs(db: DbSessionFactory) -> None:
 
             # Remove any token prices that are no longer in the manifest
             # These are prices that weren't popped from the token_prices dict above
-            for token_price in token_prices.values():
+            for token_price in existing_token_prices.values():
                 model.token_prices.remove(token_price)
 
     # Clean up built-in models that are no longer in the manifest

{arize_phoenix-11.3.0 → arize_phoenix-11.4.0}/src/phoenix/server/api/routers/v1/experiment_evaluations.py

@@ -1,6 +1,7 @@
 from datetime import datetime
 from typing import Any, Literal, Optional
 
+from dateutil.parser import isoparse
 from fastapi import APIRouter, HTTPException
 from pydantic import Field
 from starlette.requests import Request
@@ -95,8 +96,8 @@ async def upsert_experiment_evaluation(
             explanation=explanation,
             error=error,
             metadata_=metadata,  # `metadata_` must match database
-            start_time=datetime.fromisoformat(start_time),
-            end_time=datetime.fromisoformat(end_time),
+            start_time=isoparse(start_time),
+            end_time=isoparse(end_time),
             trace_id=payload.get("trace_id"),
         )
         dialect = SupportedSQLDialect(session.bind.dialect.name)