arize-phoenix 8.31.0__py3-none-any.whl → 8.32.1__py3-none-any.whl

{arize_phoenix-8.31.0.dist-info → arize_phoenix-8.32.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.31.0
+Version: 8.32.1
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -214,6 +214,7 @@ Phoenix is built on top of OpenTelemetry and is vendor, language, and framework
 | [AWS Bedrock](https://docs.arize.com/phoenix/tracing/integrations-tracing/bedrock) | `openinference-instrumentation-bedrock` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-bedrock.svg)](https://pypi.python.org/pypi/openinference-instrumentation-bedrock) |
 | [LangChain](https://docs.arize.com/phoenix/tracing/integrations-tracing/langchain) | `openinference-instrumentation-langchain` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-langchain.svg)](https://pypi.python.org/pypi/openinference-instrumentation-langchain) |
 | [MistralAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/mistralai) | `openinference-instrumentation-mistralai` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-mistralai.svg)](https://pypi.python.org/pypi/openinference-instrumentation-mistralai) |
+| [Google GenAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/google-gen-ai) | `openinference-instrumentation-google-genai` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-google-genai.svg)](https://pypi.python.org/pypi/openinference-instrumentation-google-genai) |
 | [Guardrails](https://docs.arize.com/phoenix/tracing/integrations-tracing/guardrails) | `openinference-instrumentation-guardrails` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-guardrails.svg)](https://pypi.python.org/pypi/openinference-instrumentation-guardrails) |
 | [VertexAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/vertexai) | `openinference-instrumentation-vertexai` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-vertexai.svg)](https://pypi.python.org/pypi/openinference-instrumentation-vertexai) |
 | [CrewAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/crewai) | `openinference-instrumentation-crewai` | [![PyPI Version](https://img.shields.io/pypi/v/openinference-instrumentation-crewai.svg)](https://pypi.python.org/pypi/openinference-instrumentation-crewai) |
@@ -248,6 +249,7 @@ Join our community to connect with thousands of AI builders.
 - 🐞 Report bugs with [GitHub Issues](https://github.com/Arize-ai/phoenix/issues).
 - 𝕏 Follow us on [𝕏](https://twitter.com/ArizePhoenix).
 - 🗺️ Check out our [roadmap](https://github.com/orgs/Arize-ai/projects/45) to see where we're heading next.
+- 🧑‍🏫 Deep dive into everything [Agents](http://arize.com/ai-agents/) and [LLM Evaluations](https://arize.com/llm-evaluation) on Arize's Learning Hubs.
 
 ## Breaking Changes
 

{arize_phoenix-8.31.0.dist-info → arize_phoenix-8.32.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=ngkyKGVatX3cO2WJdo2hKdaVKP-xJCMvqthvga6kJss,5196
 phoenix/settings.py,sha256=x87BX7hWGQQZbrW_vrYqFR_izCGfO9gFc--JXUG4Tdk,754
-phoenix/version.py,sha256=Sa-CqA7DAHrfdGC8Wbr_rWQJ6kTLqJ_S7kdFFAVwwmg,23
+phoenix/version.py,sha256=UaNVMdObIxyLeFhiJhW0aDRn8ABTA1EjarJjwY2CmEY,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,12 +16,13 @@ phoenix/db/README.md,sha256=7BFJuecWJjFSxEYmoSv4CvoFHHszX4tFy3HJExOxS-4,8661
 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=gxcP2aNy_JyKHv1MO4d2UM47GTMy1jDcN-FQETZ3iNA,7348
+phoenix/db/engines.py,sha256=tB_8iWMDz0folryVvw29sbBUxJOB2XZ-Xx0Uexj3uns,6889
 phoenix/db/enums.py,sha256=tt7iovXLhVTLZ3_LbHNGgcI44SnNjXfkKtLAZG57T54,428
 phoenix/db/facilitator.py,sha256=6CeUMNTkNXrdviSQspY9ktFpVX-JHXcMJz6L0PuHxrI,7481
 phoenix/db/helpers.py,sha256=rbbHcl-STzcEpcXCYx6jbKzko7r3ggrWHHsXjZ48HsM,5352
 phoenix/db/migrate.py,sha256=oUrXH8yEbcpL4eh09aSCuUiSrhFli0eT5D_j4ZmYChY,2797
 phoenix/db/models.py,sha256=TbHgtT7WWdkQK-OtLsUqp3MwP23HGV1IaSAWTqCf5ac,45707
+phoenix/db/pg_config.py,sha256=h6mB7qF7t4Zk6VGvAiyefHGVu74o-yJynaWzeE39k9Y,6001
 phoenix/db/insertion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/db/insertion/constants.py,sha256=8wifm7X-1XvroZ__R2Gc96NsgLhTDn0zXl4lehlXtcA,70
 phoenix/db/insertion/dataset.py,sha256=I9OC1ouVx7m6BH_c8hvcxW1dWGRAtpvXee29yBTuFkg,7136
@@ -220,7 +221,7 @@ phoenix/server/api/routers/v1/experiments.py,sha256=V9_sxqLTE1MKGFu9H3FEdGKr70lY
 phoenix/server/api/routers/v1/models.py,sha256=r0nM2kFJ3mxDqgc5vFr1cjNuyOPs3RIKE_DS2VMdF48,1749
 phoenix/server/api/routers/v1/projects.py,sha256=qv4RffYVGCJQuJ3FzaTVkueY8qKY0-GUbG9eSdm7oRY,14181
 phoenix/server/api/routers/v1/prompts.py,sha256=aBOUBwLDzZDIzJQkxJcR8ZKnakNJOLMwzsLKINSs1mA,26545
-phoenix/server/api/routers/v1/spans.py,sha256=uoU_bwIgz86fuvPjP5sX8goDyuCcnsTig-x3f17p60U,9625
+phoenix/server/api/routers/v1/spans.py,sha256=kvWRbMy5LjEPOw1Hd3PyncSspRYNi6ISe3Wr3qRpHa0,9751
 phoenix/server/api/routers/v1/traces.py,sha256=hSv35QIB4mwFgp53rOpz3zWIiSwbZzQnjafD790QuJU,7908
 phoenix/server/api/routers/v1/utils.py,sha256=SoRl0Dc8By15ZckhNcXg2QRrqYjMvgTjVcqrZ6MwVmo,3065
 phoenix/server/api/types/Annotation.py,sha256=r1JDvkWoFKeLRysos3l1exqKfs8zoklVv2tNSG9kFzQ,849
@@ -319,10 +320,10 @@ phoenix/server/static/apple-touch-icon-76x76.png,sha256=CT_xT12I0u2i0WU8JzBZBuOQ
 phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZVooBJ_SGk2ZkDs,3801
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
-phoenix/server/static/.vite/manifest.json,sha256=5ikTlLYIjdjOVgckDRctdD8qzVkeLjZbq7Ej0yMPFqU,2165
-phoenix/server/static/assets/components-C0W7rSGA.js,sha256=AMKAUIAq-84OfX-YRErFUSsj7De-FBf4XGHXMRsLz4I,456411
-phoenix/server/static/assets/index-BmLpy9qA.js,sha256=b1J1-ZmyqBD4P6UPPtEs_nXqVsgBoA-qi1ewdJgu0oE,60460
-phoenix/server/static/assets/pages-wYat6c3P.js,sha256=x8xm3-E1ceUhC6MznDcQkbU63ljqLwo-nj34dTfRqRk,864937
+phoenix/server/static/.vite/manifest.json,sha256=lkhQOaQx8cjiJQeVrrDfp2_pSvKAS8o-DDrzjdLXY7Y,2165
+phoenix/server/static/assets/components-x-gKFJ8C.js,sha256=X4qsvxt0Au4bURJd_ntsb_5zDbCCf6ExCyWjTc3QANc,458101
+phoenix/server/static/assets/index-B0CbpsxD.js,sha256=0Y1a5THRDD7MIVBniyp-LQRcFniB0L6GuLI7LIorVNo,60460
+phoenix/server/static/assets/pages-BU4VdyeH.js,sha256=7rSqUn1XhMhJJdw8u6lEcvwWGpVJI0ciX1y3Z5euwF0,866121
 phoenix/server/static/assets/vendor-BfhM_F1u.js,sha256=S90L1KRZOw_NX6C9FENfLs6bSuEzf7zVDAqjZAZJZgE,2514280
 phoenix/server/static/assets/vendor-Cg6lcjUC.css,sha256=nZrkr0u6NNElFGvpWHk9GTHeGoibCXCli1bE7mXZGZg,1816
 phoenix/server/static/assets/vendor-arizeai-CxXYQNUl.js,sha256=V0umBNhH2psKhzwNKvLylYrqVfHu4I6NDSqejmR2OIU,193248
@@ -333,7 +334,7 @@ phoenix/server/static/assets/vendor-three-C5WAXd5r.js,sha256=ELkg06u70N7h8oFmvqd
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=e8_jdi7Eo19SK7DI_gglkTW094D17E0VAegoMmmmvIc,4330
 phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/session/client.py,sha256=51UCIZ_k4Wd4b8f4Yi888MOie6nhF8E_5in41eBTUVU,35234
+phoenix/session/client.py,sha256=uw5WlCuFcN_eEj7Ko2bhJVcaihEIp7Evy50KnL6Sq-k,35602
 phoenix/session/data_extractor.py,sha256=Y0RzYFaNy9fQj8PEIeQ76TBZ90_E1FW7bXu3K5x0EZY,2782
 phoenix/session/evaluation.py,sha256=Q3fOMNELvqkk-b6a6PKc8pDJdsNQ0ZbTpseUSA2NKqs,5300
 phoenix/session/session.py,sha256=twE8tld8knShmTpZp6R80CUt2jau3RmfdfZAQCLSTKU,27621
@@ -342,7 +343,7 @@ phoenix/trace/attributes.py,sha256=hyEKYZWPCP4NRmW7VmiC2voa3TH7FYKUBR9DYiVfXlw,1
 phoenix/trace/errors.py,sha256=wB1z8qdPckngdfU-TORToekvg3344oNFAA83_hC2yFY,180
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
 phoenix/trace/exporter.py,sha256=bUXh8fjJIbHurrnt4bAm-cCWqUN5FqNsIc8DZzzklkQ,4695
-phoenix/trace/fixtures.py,sha256=k5gduf21C5NbZx95O_WB7mqGnrRiig0ti2VVYQiB9vo,19537
+phoenix/trace/fixtures.py,sha256=1c7fsyvmxC53Fib9T_Qxp_Ly3OZdDbkLQ0XpFzikEjk,20298
 phoenix/trace/otel.py,sha256=RJSbAuzS4KBS0t-fntXQaaYwv7FmIXRMrw65DI67z8k,10622
 phoenix/trace/projects.py,sha256=9dKv1aiKL4IYMFsg2xnC6EOIRO0YHtkR5o9ALHbMK9g,2178
 phoenix/trace/schemas.py,sha256=Su6e567Bei9oo6PsWO2srTcPAj9C2bMgbGtx64Sgqeg,6332
@@ -355,7 +356,7 @@ phoenix/trace/dsl/README.md,sha256=ihmP9zGUC5V-TDbzKla76LuyDqPDQIBUH2BORwxNI68,2
 phoenix/trace/dsl/__init__.py,sha256=WIQIjJg362XD3s50OsPJJ0xbDsGp41bSv7vDllLrPuA,144
 phoenix/trace/dsl/filter.py,sha256=yCmQKvIh7SA5Q-bQ-MKMlYY34lZnG39MLEmL2A93MTw,30472
 phoenix/trace/dsl/helpers.py,sha256=pJESAz7sDNE7HTVCFzxIx5sEsyUSh7RIN7yqmndAaF8,4171
-phoenix/trace/dsl/query.py,sha256=ePqfY6ssyHX3E1QQYsxMz_mr-1Uiq6GKQHEb_-3wrdg,31753
+phoenix/trace/dsl/query.py,sha256=qi5k83Ch7mouSzd8w8dkMHVJ9C45tYXZLtb4lMwToZM,36350
 phoenix/trace/v1/__init__.py,sha256=-IbAD0ruESMjvQLvGAg9CTfjBUATFDx1OXseDPis6-0,88
 phoenix/trace/v1/evaluation_pb2.py,sha256=8sXvv2BW_vqD30MOMbmkeE2zpmm7ncik21kl3e-HzeQ,2254
 phoenix/trace/v1/evaluation_pb2.pyi,sha256=cCbbx06gwQmaH14s3J1X25TtaARh-k1abbxQdQCXGm8,4500
@@ -369,9 +370,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-8.31.0.dist-info/METADATA,sha256=gg_zdpnAgEqbF0nuN1D4eG-nsk8is9bAVoccNYraMdY,24478
-arize_phoenix-8.31.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-8.31.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-8.31.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-8.31.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-8.31.0.dist-info/RECORD,,
+arize_phoenix-8.32.1.dist-info/METADATA,sha256=Tmih_769p0j9crLeyhY-u6bNxpEXH8MlLXKuMKxC0bg,24950
+arize_phoenix-8.32.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-8.32.1.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-8.32.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-8.32.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-8.32.1.dist-info/RECORD,,

phoenix/db/engines.py

@@ -1,10 +1,13 @@
+from __future__ import annotations
+
 import asyncio
 import json
+import logging
 from collections.abc import Callable
 from datetime import datetime
 from enum import Enum
 from sqlite3 import Connection
-from typing import Any, Mapping
+from typing import Any
 
 import aiosqlite
 import numpy as np
@@ -18,10 +21,13 @@ from phoenix.config import LoggingMode, get_env_database_schema
 from phoenix.db.helpers import SupportedSQLDialect
 from phoenix.db.migrate import migrate_in_thread
 from phoenix.db.models import init_models
+from phoenix.db.pg_config import get_pg_config
 from phoenix.settings import Settings
 
 sqlean.extensions.enable("text", "stats")
 
+logger = logging.getLogger(__name__)
+
 
 def set_sqlite_pragma(connection: Connection, _: Any) -> None:
     cursor = connection.cursor()
@@ -163,15 +169,20 @@ def aio_postgresql_engine(
     log_to_stdout: bool = False,
     log_migrations_to_stdout: bool = True,
 ) -> AsyncEngine:
+    asyncpg_url, asyncpg_args = get_pg_config(url, "asyncpg")
     engine = create_async_engine(
-        url=url.set(query=_asyncpg_url_query(url.query)),
+        url=asyncpg_url,
+        connect_args=asyncpg_args,
         echo=log_to_stdout,
         json_serializer=_dumps,
     )
     if not migrate:
         return engine
+
+    psycopg_url, psycopg_args = get_pg_config(url, "psycopg")
     sync_engine = sqlalchemy.create_engine(
-        url=url.set(drivername="postgresql+psycopg", query=_psycopg_url_query(url.query)),
+        url=psycopg_url,
+        connect_args=psycopg_args,
         echo=log_migrations_to_stdout,
         json_serializer=_dumps,
     )
@@ -181,24 +192,6 @@ 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/db/pg_config.py

@@ -0,0 +1,197 @@
+from __future__ import annotations
+
+import ssl
+from typing import Any, Container, Final, Literal, Mapping, TypedDict, get_type_hints
+
+from sqlalchemy import URL
+from typing_extensions import assert_never
+
+
+def get_pg_config(
+    url: URL,
+    driver: Literal["psycopg", "asyncpg"],
+) -> tuple[URL, dict[str, Any]]:
+    """Convert SQLAlchemy URL to driver-specific configuration.
+
+    Args:
+        url: SQLAlchemy URL
+        driver: "psycopg" or "asyncpg"
+
+    Returns:
+        Tuple of (base_url, connect_args):
+        - base_url: URL with driver prefix and non-SSL parameters
+        - connect_args: SSL configuration for the driver
+    """
+    # Create new URL with appropriate driver
+    query = url.query
+    ssl_args = _get_ssl_args(query)
+
+    # Create base URL without SSL parameters
+    base_url = url.set(
+        drivername=f"postgresql+{driver}",
+        query={k: v for k, v in query.items() if k not in _SSL_KEYS},
+    )
+
+    # Get appropriate SSL configuration based on driver
+    if driver == "psycopg":
+        connect_args = dict(ssl_args)
+        # Remove asyncpg-specific parameters from base URL
+        base_url = base_url.set(query=_remove_asyncpg_only_params(base_url.query))
+    elif driver == "asyncpg":
+        # Only create SSL context if we have SSL parameters and sslmode is not disable
+        if ssl_args and ssl_args.get("sslmode") != "disable":
+            connect_args = {"ssl": _get_ssl_context(ssl_args)}
+        else:
+            connect_args = {}
+    else:
+        assert_never(driver)
+    return base_url, connect_args
+
+
+class _SSLArgs(TypedDict, total=False):
+    """SSL parameters for PostgreSQL connections.
+
+    All fields are optional. Only includes parameters that can be converted to ssl.SSLContext.
+
+    Attributes:
+        sslmode: SSL mode (disable, allow, prefer, require, verify-ca, verify-full)
+        sslrootcert: Path to root CA certificate
+        sslcert: Path to client certificate
+        sslkey: Path to client private key
+        sslpassword: Password for private key
+        sslcrl: Path to CRL file
+        sslcrldir: Path to CRL directory
+        sslsni: Enable SNI (0 or 1)
+    """
+
+    sslmode: str
+    sslrootcert: str
+    sslcert: str
+    sslkey: str
+    sslpassword: str
+    sslcrl: str
+    sslcrldir: str
+    sslsni: str
+
+
+_SSL_KEYS: Final[Container[str]] = get_type_hints(_SSLArgs).keys()
+
+
+def _get_ssl_args(
+    query_params: Mapping[str, str | tuple[str, ...]],
+) -> _SSLArgs:
+    """Extract SSL parameters from a SQLAlchemy URL query.
+
+    Args:
+        query_params: SQLAlchemy URL query parameters
+
+    Returns:
+        Dictionary of SSL parameters
+    """
+    result: _SSLArgs = {}
+
+    def get_str(key: str) -> str | None:
+        if value := query_params.get(key):
+            if not isinstance(value, str):
+                raise ValueError(f"Invalid value type for {key}: {type(value)}")
+            return value
+        return None
+
+    if sslmode := get_str("sslmode"):
+        result["sslmode"] = sslmode
+
+    if ca_cert := get_str("sslrootcert"):
+        result["sslrootcert"] = ca_cert
+
+    if cert := get_str("sslcert"):
+        result["sslcert"] = cert
+
+    if key := get_str("sslkey"):
+        result["sslkey"] = key
+
+    if password := get_str("sslpassword"):
+        result["sslpassword"] = password
+
+    if crl := get_str("sslcrl"):
+        result["sslcrl"] = crl
+
+    if crl_dir := get_str("sslcrldir"):
+        result["sslcrldir"] = crl_dir
+
+    if sslsni := get_str("sslsni"):
+        result["sslsni"] = sslsni
+
+    return result
+
+
+def _get_ssl_context(ssl_args: _SSLArgs) -> ssl.SSLContext:
+    """Convert PostgreSQL SSL parameters to an SSL context.
+
+    Args:
+        ssl_args: PostgreSQL SSL parameters from _SSLArgs TypedDict
+
+    Returns:
+        Configured SSL context with:
+        - Root CA certificate for server verification (if provided)
+        - Client certificate and key for mutual TLS (if provided)
+        - SSL mode appropriate verification settings
+        - Certificate revocation list checking (if provided)
+    """
+    # Create SSL context
+    ssl_context = ssl.create_default_context()
+
+    # Load CA certificate if provided
+    if ca_cert := ssl_args.get("sslrootcert"):
+        ssl_context.load_verify_locations(cafile=ca_cert)
+
+    # Load client certificates if provided
+    if (cert := ssl_args.get("sslcert")) and (key := ssl_args.get("sslkey")):
+        ssl_context.load_cert_chain(
+            certfile=cert,
+            keyfile=key,
+            password=ssl_args.get("sslpassword"),
+        )
+
+    # Load CRL if provided
+    if crl := ssl_args.get("sslcrl"):
+        ssl_context.load_verify_locations(cafile=crl)
+    if crl_dir := ssl_args.get("sslcrldir"):
+        ssl_context.load_verify_locations(capath=crl_dir)
+
+    # Set verification mode based on sslmode
+    sslmode = ssl_args.get("sslmode", "prefer")
+    if sslmode == "verify-full":
+        # Full verification: certificate and hostname
+        ssl_context.check_hostname = True
+        ssl_context.verify_mode = ssl.CERT_REQUIRED
+    elif sslmode == "verify-ca":
+        # Certificate verification only
+        ssl_context.check_hostname = False
+        ssl_context.verify_mode = ssl.CERT_REQUIRED
+    else:  # require, prefer, allow, disable
+        # No verification, just encryption
+        ssl_context.check_hostname = False
+        ssl_context.verify_mode = ssl.CERT_NONE
+
+    return ssl_context
+
+
+def _remove_asyncpg_only_params(
+    query: Mapping[str, str | tuple[str, ...]],
+) -> dict[str, str | tuple[str, ...]]:
+    """Remove asyncpg-specific parameters from a SQLAlchemy URL query.
+
+    Args:
+        query: SQLAlchemy URL query parameters
+
+    Returns:
+        Dictionary of query parameters with asyncpg-specific parameters removed
+    """
+    return {k: v for k, v in query.items() if k not in _ASYNCPG_ONLY_KEYS}
+
+
+# Asyncpg-specific parameter keys
+_ASYNCPG_ONLY_KEYS: Final[tuple[str, ...]] = (
+    "prepared_statement_cache_size",
+    # Add other asyncpg-specific parameters here if needed
+)

phoenix/server/api/routers/v1/spans.py

@@ -47,6 +47,7 @@ class QuerySpansRequestBody(V1RoutesBaseModel):
     end_time: Optional[datetime] = None
     limit: int = DEFAULT_SPAN_LIMIT
     root_spans_only: Optional[bool] = None
+    orphan_span_as_root_span: bool = True
     project_name: Optional[str] = Field(
         default=None,
         description=(
@@ -116,6 +117,7 @@ async def query_spans_handler(
                     ),
                     limit=request_body.limit,
                     root_spans_only=request_body.root_spans_only,
+                    orphan_span_as_root_span=request_body.orphan_span_as_root_span,
                 )
             )
     if not results:

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

@@ -1,22 +1,22 @@
 {
-  "_components-C0W7rSGA.js": {
-    "file": "assets/components-C0W7rSGA.js",
+  "_components-x-gKFJ8C.js": {
+    "file": "assets/components-x-gKFJ8C.js",
     "name": "components",
     "imports": [
       "_vendor-BfhM_F1u.js",
-      "_pages-wYat6c3P.js",
+      "_pages-BU4VdyeH.js",
       "_vendor-arizeai-CxXYQNUl.js",
       "_vendor-codemirror-B0NIFPOL.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-wYat6c3P.js": {
-    "file": "assets/pages-wYat6c3P.js",
+  "_pages-BU4VdyeH.js": {
+    "file": "assets/pages-BU4VdyeH.js",
     "name": "pages",
     "imports": [
       "_vendor-BfhM_F1u.js",
       "_vendor-arizeai-CxXYQNUl.js",
-      "_components-C0W7rSGA.js",
+      "_components-x-gKFJ8C.js",
       "_vendor-codemirror-B0NIFPOL.js",
       "_vendor-recharts-CrrDFWK1.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-BmLpy9qA.js",
+    "file": "assets/index-B0CbpsxD.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-BfhM_F1u.js",
       "_vendor-arizeai-CxXYQNUl.js",
-      "_pages-wYat6c3P.js",
-      "_components-C0W7rSGA.js",
+      "_pages-BU4VdyeH.js",
+      "_components-x-gKFJ8C.js",
       "_vendor-three-C5WAXd5r.js",
       "_vendor-codemirror-B0NIFPOL.js",
       "_vendor-shiki-C5bJ-RPf.js",