feat(experimental): postgres support

src/phoenix/db/engines.py

@@ -24,12 +24,15 @@ def set_sqlite_pragma(connection: Connection, _: Any) -> None:
     cursor.close()
 
 
+def get_db_url(driver: str = "sqlite+aiosqlite", database: Union[str, Path] = ":memory:") -> URL:
+    return URL.create(driver, database=str(database))
+
+
 def aiosqlite_engine(
     database: Union[str, Path] = ":memory:",
     echo: bool = False,
 ) -> AsyncEngine:
-    driver_name = "sqlite+aiosqlite"
-    url = URL.create(driver_name, database=str(database))
+    url = get_db_url(driver="sqlite+aiosqlite", database=database)
     engine = create_async_engine(url=url, echo=echo, json_serializer=_dumps)
     event.listen(engine.sync_engine, "connect", set_sqlite_pragma)
     if str(database) == ":memory:":

src/phoenix/server/main.py

@@ -26,7 +26,7 @@
 from phoenix.core.traces import Traces
 from phoenix.datasets.dataset import EMPTY_DATASET, Dataset
 from phoenix.datasets.fixtures import FIXTURES, get_datasets
-from phoenix.db.engines import aiosqlite_engine
+from phoenix.db.engines import aiosqlite_engine, get_db_url
 from phoenix.pointcloud.umap_parameters import (
     DEFAULT_MIN_DIST,
     DEFAULT_N_NEIGHBORS,
@@ -70,7 +70,7 @@
 |  🚀 Phoenix Server 🚀
 |  Phoenix UI: http://{host}:{port}
 |  Log traces: /v1/traces over HTTP
-|  Storage location: {working_dir}
+|  Storage: {storage}
 """
 
 
@@ -268,6 +268,7 @@ def _send_spans(spans: Iterable[Span], url: str) -> None:
         start_prometheus()
 
     working_dir = get_working_dir().resolve()
+    sql_url = get_db_url(database=working_dir / "phoenix.db")
     engine = aiosqlite_engine(working_dir / "phoenix.db")
     app = create_app(
         engine=engine,
@@ -290,7 +291,7 @@ def _send_spans(spans: Iterable[Span], url: str) -> None:
         "version": phoenix_version,
         "host": host,
         "port": port,
-        "working_dir": working_dir,
+        "storage": sql_url,
     }
     print(_WELCOME_MESSAGE.format(**config))