arize-phoenix 11.5.0__py3-none-any.whl → 11.6.1__py3-none-any.whl

phoenix/server/daemons/db_disk_usage_monitor.py

@@ -11,9 +11,11 @@ from sqlalchemy import text
 from typing_extensions import assert_never
 
 from phoenix.config import (
+    ENV_PHOENIX_SQL_DATABASE_SCHEMA,
     get_env_database_allocated_storage_capacity_gibibytes,
     get_env_database_usage_email_warning_threshold_percentage,
     get_env_database_usage_insertion_blocking_threshold_percentage,
+    getenv,
 )
 from phoenix.db import models
 from phoenix.db.helpers import SupportedSQLDialect
@@ -92,14 +94,18 @@ class DbDiskUsageMonitor(DaemonTask):
                 page_size = await session.scalar(text("PRAGMA page_size;"))
             current_usage_bytes = (page_count - freelist_count) * page_size
         elif self._db.dialect is SupportedSQLDialect.POSTGRESQL:
+            nspname = getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA) or "public"
+            stmt = text("""\
+                SELECT sum(pg_total_relation_size(c.oid))
+                FROM pg_class as c
+                INNER JOIN pg_namespace as n ON n.oid = c.relnamespace
+                WHERE c.relkind = 'r'
+                AND n.nspname = :nspname;
+            """).bindparams(nspname=nspname)
             async with self._db() as session:
-                current_usage_bytes = await session.scalar(
-                    text("SELECT pg_database_size(current_database());")
-                )
+                current_usage_bytes = await session.scalar(stmt)
         else:
             assert_never(self._db.dialect)
-        if not isinstance(current_usage_bytes, (int, float)):
-            raise TypeError(f"Expected int or float, got {type(current_usage_bytes)}")
         return float(current_usage_bytes)
 
     async def _check_thresholds(self, current_usage_gibibytes: float) -> None:

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

@@ -1,22 +1,22 @@
 {
-  "_components-Bwf6zNbg.js": {
-    "file": "assets/components-Bwf6zNbg.js",
+  "_components-BQP0XPOQ.js": {
+    "file": "assets/components-BQP0XPOQ.js",
     "name": "components",
     "imports": [
       "_vendor-DRWIRkSJ.js",
-      "_pages-BCR8hW_l.js",
+      "_pages-DACgXBiR.js",
       "_vendor-arizeai-DUhQaeau.js",
       "_vendor-codemirror-D_6Q6Auv.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-BCR8hW_l.js": {
-    "file": "assets/pages-BCR8hW_l.js",
+  "_pages-DACgXBiR.js": {
+    "file": "assets/pages-DACgXBiR.js",
     "name": "pages",
     "imports": [
       "_vendor-DRWIRkSJ.js",
       "_vendor-arizeai-DUhQaeau.js",
-      "_components-Bwf6zNbg.js",
+      "_components-BQP0XPOQ.js",
       "_vendor-codemirror-D_6Q6Auv.js",
       "_vendor-recharts-BNBwj7vz.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-Bfg9uQ43.js",
+    "file": "assets/index-CIOCExre.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-DRWIRkSJ.js",
       "_vendor-arizeai-DUhQaeau.js",
-      "_pages-BCR8hW_l.js",
-      "_components-Bwf6zNbg.js",
+      "_pages-DACgXBiR.js",
+      "_components-BQP0XPOQ.js",
       "_vendor-three-C5WAXd5r.js",
       "_vendor-codemirror-D_6Q6Auv.js",
       "_vendor-shiki-k1qj_XjP.js",