arize-phoenix 11.7.0__py3-none-any.whl → 11.8.0__py3-none-any.whl

phoenix/server/authorization.py

@@ -25,6 +25,7 @@ Usage:
 from fastapi import HTTPException, Request
 from fastapi import status as fastapi_status
 
+from phoenix.config import get_env_support_email
 from phoenix.server.bearer_auth import PhoenixUser
 
 
@@ -54,9 +55,33 @@ def require_admin(request: Request) -> None:
 
 
 def is_not_locked(request: Request) -> None:
+    """
+    FastAPI dependency to ensure database operations are not locked due to insufficient storage.
+
+    This dependency checks if data insertion and update operations are disabled due to
+    storage capacity limits. When storage thresholds are exceeded, it raises an HTTP 507
+    error with actionable guidance for users.
+
+    Usage:
+        Add as a dependency to any route that modifies data:
+
+            @router.post("/create-data", dependencies=[Depends(is_not_locked)])
+            async def create_data(...):
+                ...
+
+    Raises:
+        HTTPException: HTTP 507 Insufficient Storage when database operations are locked.
+            The error includes guidance on resolving storage issues and support contact
+            information if configured.
+    """
     if request.app.state.db.should_not_insert_or_update:
+        detail = (
+            "Database operations are disabled due to insufficient storage. "
+            "Please delete old data or increase storage."
+        )
+        if support_email := get_env_support_email():
+            detail += f" Need help? Contact us at {support_email}"
         raise HTTPException(
             status_code=fastapi_status.HTTP_507_INSUFFICIENT_STORAGE,
-            detail="Operations that insert or update database "
-            "records are currently not allowed.",
+            detail=detail,
         )

phoenix/server/cost_tracking/cost_details_calculator.py

@@ -1,5 +1,5 @@
 from itertools import chain
-from typing import Any, Iterable, Mapping
+from typing import Any, Iterable, Mapping, Optional
 
 from typing_extensions import TypeAlias
 
@@ -60,7 +60,7 @@ class SpanCostDetailsCalculator:
             for p in prices
             if p.is_prompt
         }
-        if "input" not in self._prompt:
+        if self._prompt and "input" not in self._prompt:
             raise ValueError("Token prices for prompt must include an 'input' token type")
 
         # Create calculators for completion token types (is_prompt=False)
@@ -69,7 +69,7 @@ class SpanCostDetailsCalculator:
             for p in prices
             if not p.is_prompt
         }
-        if "output" not in self._completion:
+        if self._completion and "output" not in self._completion:
             raise ValueError("Token prices for completion must include an 'output' token type")
 
     def calculate_details(
@@ -112,6 +112,9 @@ class SpanCostDetailsCalculator:
         """
         prompt_details: dict[_TokenType, models.SpanCostDetail] = {}
         completion_details: dict[_TokenType, models.SpanCostDetail] = {}
+        calculator: Optional[TokenCostCalculator]
+        cost: Optional[float]
+        cost_per_token: Optional[float]
 
         # Phase 1: Process detailed token counts from span attributes
         for is_prompt, prefix, calculators, results in (
@@ -128,18 +131,19 @@ class SpanCostDetailsCalculator:
                     tokens = max(0, int(token_count))
 
                     # Calculate cost using specific calculator or fallback to default
+                    calculator = None
+                    calculator_key = "input" if is_prompt else "output"
                     if token_type in calculators:
                         # Use specific calculator for this token type
                         calculator = calculators[token_type]
-                    else:
-                        # Fallback to default calculator: "input" for prompts,
-                        # "output" for completions
-                        key = "input" if is_prompt else "output"
-                        calculator = calculators[key]
-                    cost = calculator.calculate_cost(attributes, tokens)
+                    elif calculator_key in calculators:
+                        calculator = calculators[calculator_key]
 
-                    # Calculate cost per token (avoid division by zero)
-                    cost_per_token = cost / tokens if tokens else None
+                    cost = None
+                    cost_per_token = None
+                    if calculator:
+                        cost = calculator.calculate_cost(attributes, tokens)
+                        cost_per_token = cost / tokens if tokens else None
 
                     detail = models.SpanCostDetail(
                         token_type=token_type,
@@ -171,11 +175,13 @@ class SpanCostDetailsCalculator:
             if tokens <= 0:
                 continue
 
-            # Calculate cost using guaranteed default calculator (input/output are required)
-            cost = calculators[token_type].calculate_cost(attributes, tokens)
-
-            # Calculate cost per token (avoid division by zero)
-            cost_per_token = cost / tokens if cost and tokens else None
+            # Calculate cost using calculator if available
+            cost = None
+            cost_per_token = None
+            if token_type in calculators:
+                calculator = calculators[token_type]
+                cost = calculator.calculate_cost(attributes, tokens)
+                cost_per_token = cost / tokens if tokens else None
 
             detail = models.SpanCostDetail(
                 token_type=token_type,

phoenix/server/daemons/span_cost_calculator.py

@@ -5,7 +5,6 @@ from asyncio import sleep
 from datetime import datetime
 from typing import Any, Mapping, NamedTuple, Optional
 
-from sqlalchemy import inspect
 from typing_extensions import TypeAlias
 
 from phoenix.db import models
@@ -84,18 +83,13 @@ class SpanCostCalculator(DaemonTask):
             start_time=start_time,
             attributes=attributes,
         )
-        if not cost_model:
-            return None
-        if not isinstance(inspect(cost_model).attrs.token_prices.loaded_value, list):
-            return None
-
-        calculator = SpanCostDetailsCalculator(cost_model.token_prices)
+        calculator = SpanCostDetailsCalculator(cost_model.token_prices if cost_model else [])
         details = calculator.calculate_details(attributes)
         if not details:
             return None
 
         cost = models.SpanCost(
-            model_id=cost_model.id,
+            model_id=cost_model.id if cost_model else None,
             span_start_time=start_time,
         )
         for detail in details:

phoenix/server/email/sender.py

@@ -8,7 +8,7 @@ from anyio import to_thread
 from jinja2 import Environment, FileSystemLoader, select_autoescape
 from typing_extensions import TypeAlias
 
-from phoenix.config import get_env_root_url
+from phoenix.config import get_env_root_url, get_env_support_email
 
 EMAIL_TEMPLATE_FOLDER = Path(__file__).parent / "templates"
 
@@ -96,6 +96,7 @@ class SimpleEmailSender:
             current_usage_gibibytes=current_usage_gibibytes,
             allocated_storage_gibibytes=allocated_storage_gibibytes,
             notification_threshold_percentage=notification_threshold_percentage,
+            support_email=get_env_support_email(),
         )
 
         msg = EmailMessage()

phoenix/server/email/templates/db_disk_usage_notification.html

@@ -12,5 +12,8 @@
 <p><strong>Usage Percentage:</strong> {{ ((current_usage_gibibytes / allocated_storage_gibibytes) * 100)|round(1) }}%</p>
 <p><strong>Notification Threshold:</strong> {{ notification_threshold_percentage }}%</p>
 <p>Please consider removing old data or increasing your storage allocation to prevent interruption.</p>
+{% if support_email %}
+<p>If you need assistance, please contact support at <a id="support-email" href="mailto:{{ support_email }}">{{ support_email }}</a>.</p>
+{% endif %}
 </body>
 </html>

phoenix/server/static/.vite/manifest.json

@@ -1,28 +1,28 @@
 {
-  "_components-J3qjrjBf.js": {
-    "file": "assets/components-J3qjrjBf.js",
+  "_components-5M9nebi4.js": {
+    "file": "assets/components-5M9nebi4.js",
     "name": "components",
     "imports": [
-      "_vendor-BnPh9i9e.js",
-      "_pages-CW1UdBht.js",
-      "_vendor-arizeai-Cr9o_Iu_.js",
-      "_vendor-codemirror-k3zCIjlN.js",
+      "_vendor-Bl7CyFDw.js",
+      "_pages-DF8rqxJ4.js",
+      "_vendor-arizeai-B_viEUUA.js",
+      "_vendor-codemirror-vlcH1_iR.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-CW1UdBht.js": {
-    "file": "assets/pages-CW1UdBht.js",
+  "_pages-DF8rqxJ4.js": {
+    "file": "assets/pages-DF8rqxJ4.js",
     "name": "pages",
     "imports": [
-      "_vendor-BnPh9i9e.js",
-      "_vendor-arizeai-Cr9o_Iu_.js",
-      "_components-J3qjrjBf.js",
-      "_vendor-codemirror-k3zCIjlN.js",
-      "_vendor-recharts-BdblEuGB.js"
+      "_vendor-Bl7CyFDw.js",
+      "_vendor-arizeai-B_viEUUA.js",
+      "_components-5M9nebi4.js",
+      "_vendor-codemirror-vlcH1_iR.js",
+      "_vendor-recharts-C9cQu72o.js"
     ]
   },
-  "_vendor-BnPh9i9e.js": {
-    "file": "assets/vendor-BnPh9i9e.js",
+  "_vendor-Bl7CyFDw.js": {
+    "file": "assets/vendor-Bl7CyFDw.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-C5WAXd5r.js"
@@ -35,33 +35,33 @@
     "file": "assets/vendor-WIZid84E.css",
     "src": "_vendor-WIZid84E.css"
   },
-  "_vendor-arizeai-Cr9o_Iu_.js": {
-    "file": "assets/vendor-arizeai-Cr9o_Iu_.js",
+  "_vendor-arizeai-B_viEUUA.js": {
+    "file": "assets/vendor-arizeai-B_viEUUA.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-BnPh9i9e.js"
+      "_vendor-Bl7CyFDw.js"
     ]
   },
-  "_vendor-codemirror-k3zCIjlN.js": {
-    "file": "assets/vendor-codemirror-k3zCIjlN.js",
+  "_vendor-codemirror-vlcH1_iR.js": {
+    "file": "assets/vendor-codemirror-vlcH1_iR.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-BnPh9i9e.js",
-      "_vendor-shiki-DPtuv2M4.js"
+      "_vendor-Bl7CyFDw.js",
+      "_vendor-shiki-BsknB7bv.js"
     ]
   },
-  "_vendor-recharts-BdblEuGB.js": {
-    "file": "assets/vendor-recharts-BdblEuGB.js",
+  "_vendor-recharts-C9cQu72o.js": {
+    "file": "assets/vendor-recharts-C9cQu72o.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-BnPh9i9e.js"
+      "_vendor-Bl7CyFDw.js"
     ]
   },
-  "_vendor-shiki-DPtuv2M4.js": {
-    "file": "assets/vendor-shiki-DPtuv2M4.js",
+  "_vendor-shiki-BsknB7bv.js": {
+    "file": "assets/vendor-shiki-BsknB7bv.js",
     "name": "vendor-shiki",
     "imports": [
-      "_vendor-BnPh9i9e.js"
+      "_vendor-Bl7CyFDw.js"
     ]
   },
   "_vendor-three-C5WAXd5r.js": {
@@ -69,19 +69,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-CEObsQf_.js",
+    "file": "assets/index-OU2WTnGN.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-BnPh9i9e.js",
-      "_vendor-arizeai-Cr9o_Iu_.js",
-      "_pages-CW1UdBht.js",
-      "_components-J3qjrjBf.js",
+      "_vendor-Bl7CyFDw.js",
+      "_vendor-arizeai-B_viEUUA.js",
+      "_pages-DF8rqxJ4.js",
+      "_components-5M9nebi4.js",
       "_vendor-three-C5WAXd5r.js",
-      "_vendor-codemirror-k3zCIjlN.js",
-      "_vendor-shiki-DPtuv2M4.js",
-      "_vendor-recharts-BdblEuGB.js"
+      "_vendor-codemirror-vlcH1_iR.js",
+      "_vendor-shiki-BsknB7bv.js",
+      "_vendor-recharts-C9cQu72o.js"
     ]
   }
 }