arize-phoenix 7.12.0__py3-none-any.whl → 7.12.1__py3-none-any.whl

{arize_phoenix-7.12.0.dist-info → arize_phoenix-7.12.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 7.12.0
+Version: 7.12.1
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-7.12.0.dist-info → arize_phoenix-7.12.1.dist-info}/RECORD

@@ -6,7 +6,7 @@ phoenix/exceptions.py,sha256=n2L2KKuecrdflB9MsCdAYCiSEvGJptIsfRkXMoJle7A,169
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=kpW1WL0kiB8XJsO6XycvZVJ-lBkNoenhQ7atCvBoSe8,5365
 phoenix/settings.py,sha256=ht-0oN-sMV6SPXrk7Tu1EZlngpAYkGNLYPhO8DyrdQI,661
-phoenix/version.py,sha256=epZfTrsMWfpY2no1bS7o-MeQ23JncNk7yI5RH-5Pqlg,23
+phoenix/version.py,sha256=2szdaPGvYYHEsBMMeewskoN2Yu2URh06p_5-YZo0P9M,23
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=qBFraOtmwCCnWJltKNP18DDG0mULXigytlFsa6YOz6k,4837
@@ -16,7 +16,7 @@ phoenix/db/README.md,sha256=Mjrq0tG68uKQ81mN1Vmu3C795EEpZndmnE6CJXsPCXE,1909
 phoenix/db/__init__.py,sha256=pDjEFXukHmJBM-1D8RjmXkvLsz85YWNxMQczt81ec3A,118
 phoenix/db/alembic.ini,sha256=GIS6HpHaKaJbbuahZg1Rc1D2_QqyCkV9r58wdARGf6w,3262
 phoenix/db/bulk_inserter.py,sha256=faNjuwLqqsw4ky8sa4D0h9u5TvEDTOjrccUW89L008E,12725
-phoenix/db/engines.py,sha256=YqJJlziCzvNg3y5hLz4CkRjcgMxug2BXmoyFV2THcdk,6914
+phoenix/db/engines.py,sha256=esKGo_cqQLKKXNlfK7iQ7BtkOYY2xNsLdjvW2DddGjU,7321
 phoenix/db/enums.py,sha256=tt7iovXLhVTLZ3_LbHNGgcI44SnNjXfkKtLAZG57T54,428
 phoenix/db/facilitator.py,sha256=sAYqzBXYSVBKPTQVYrd7ZmtqMAr1zP9dVJwjfNGW7hc,4207
 phoenix/db/helpers.py,sha256=daKbpY2QhTPo9a_T1xNHKI4WzWHkMmmrGIws7Hw-RZ4,4884
@@ -337,9 +337,9 @@ phoenix/utilities/project.py,sha256=auVpARXkDb-JgeX5f2aStyFIkeKvGwN9l7qrFeJMVxI,
 phoenix/utilities/re.py,sha256=6YyUWIkv0zc2SigsxfOWIHzdpjKA_TZo2iqKq7zJKvw,2081
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/utilities/template_formatters.py,sha256=gh9PJD6WEGw7TEYXfSst1UR4pWWwmjxMLrDVQ_CkpkQ,2779
-arize_phoenix-7.12.0.dist-info/METADATA,sha256=PJlLicxrJuM_QJI6MgWWu0L29jZxMZEmq1GVREn6Jck,23390
-arize_phoenix-7.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-7.12.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-7.12.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-7.12.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-7.12.0.dist-info/RECORD,,
+arize_phoenix-7.12.1.dist-info/METADATA,sha256=EdZdYv0yVK-kvYDe-h3UPPFs9OHt55AhD8FWkTsDtdQ,23390
+arize_phoenix-7.12.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-7.12.1.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-7.12.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-7.12.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-7.12.1.dist-info/RECORD,,

phoenix/db/engines.py

@@ -4,7 +4,7 @@ from collections.abc import Callable
 from datetime import datetime
 from enum import Enum
 from sqlite3 import Connection
-from typing import Any
+from typing import Any, Mapping
 
 import aiosqlite
 import numpy as np
@@ -163,23 +163,15 @@ def aio_postgresql_engine(
     log_to_stdout: bool = False,
     log_migrations_to_stdout: bool = True,
 ) -> AsyncEngine:
-    url_query = dict(url.query)
-    sslmode = url_query.pop("sslmode", None) or url_query.pop("ssl", None)
     engine = create_async_engine(
-        url=url.set(
-            # https://github.com/MagicStack/asyncpg/issues/737
-            query={**url_query, "ssl": sslmode} if sslmode else url_query,
-        ),
+        url=url.set(query=_asyncpg_url_query(url.query)),
         echo=log_to_stdout,
         json_serializer=_dumps,
     )
     if not migrate:
         return engine
     sync_engine = sqlalchemy.create_engine(
-        url=url.set(
-            drivername="postgresql+psycopg",
-            query={**url_query, "sslmode": sslmode} if sslmode else url_query,
-        ),
+        url=url.set(drivername="postgresql+psycopg", query=_psycopg_url_query(url.query)),
         echo=log_migrations_to_stdout,
         json_serializer=_dumps,
     )
@@ -189,6 +181,24 @@ def aio_postgresql_engine(
     return engine
 
 
+def _asyncpg_url_query(query: Mapping[str, Any]) -> dict[str, Any]:
+    ans = dict(query)
+    if sslmode := (ans.pop("sslmode", None) or ans.pop("ssl", None)):
+        # https://github.com/MagicStack/asyncpg/issues/737
+        ans["ssl"] = sslmode
+    return ans
+
+
+def _psycopg_url_query(query: Mapping[str, Any]) -> dict[str, Any]:
+    ans = dict(query)
+    if sslmode := (ans.pop("sslmode", None) or ans.pop("ssl", None)):
+        ans["sslmode"] = sslmode
+    # prepared_statement_cache_size is only used by asyncpg, see:
+    # https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#prepared-statement-cache
+    ans.pop("prepared_statement_cache_size", None)
+    return ans
+
+
 def _dumps(obj: Any) -> str:
     return json.dumps(obj, cls=_Encoder)
 

phoenix/version.py

@@ -1 +1 @@
-__version__ = "7.12.0"
+__version__ = "7.12.1"