arize-phoenix 8.2.2__tar.gz → 8.3.0__tar.gz

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/.gitignore

@@ -13,6 +13,9 @@ sdist
 # Server static files
 /src/phoenix/server/static/*
 
+# Docgen
+/js/**/docs
+
 # Data files
 *.csv
 !tests/**/fixtures/*.csv

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.2.2
+Version: 8.3.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -55,7 +55,7 @@ Requires-Dist: uvicorn
 Requires-Dist: websockets
 Requires-Dist: wrapt>=1.17.2
 Provides-Extra: container
-Requires-Dist: anthropic>=0.45.2; extra == 'container'
+Requires-Dist: anthropic>=0.47.0; extra == 'container'
 Requires-Dist: fast-hdbscan>=0.2.0; extra == 'container'
 Requires-Dist: google-generativeai; extra == 'container'
 Requires-Dist: numba>=0.60.0; extra == 'container'
@@ -73,7 +73,7 @@ Requires-Dist: strawberry-graphql[opentelemetry]==0.253.1; extra == 'container'
 Requires-Dist: umap-learn; extra == 'container'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'container'
 Provides-Extra: dev
-Requires-Dist: anthropic; extra == 'dev'
+Requires-Dist: anthropic>=0.47.0; extra == 'dev'
 Requires-Dist: arize[autoembeddings,llm-evaluation]; extra == 'dev'
 Requires-Dist: asgi-lifespan; extra == 'dev'
 Requires-Dist: asyncpg; extra == 'dev'

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/pyproject.toml

@@ -95,7 +95,7 @@ dev = [
   "langchain>=0.0.334",
   "litellm>=1.0.3,<1.57.5", # windows compatibility broken on 1.57.5 (https://github.com/BerriAI/litellm/issues/7677)
   "google-cloud-aiplatform>=1.3",
-  "anthropic",
+  "anthropic>=0.47.0",
   "prometheus_client",
   "asgi-lifespan",
   "Faker>=30.1.0",
@@ -123,7 +123,7 @@ pg = [
   "psycopg[binary,pool]",
 ]
 container = [
-  "anthropic>=0.45.2",
+  "anthropic>=0.47.0",
   "google-generativeai",
   "prometheus-client",
   "openai>=1.0.0",

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/config.py

@@ -54,6 +54,33 @@ Phoenix supports two types of database URLs:
 Note that if you plan on using SQLite, it's advised to to use a persistent volume
 and simply point the PHOENIX_WORKING_DIR to that volume.
 """
+ENV_PHOENIX_POSTGRES_HOST = "PHOENIX_POSTGRES_HOST"
+"""
+As an alternative to setting PHOENIX_SQL_DATABASE_URL, you can set the following
+environment variables to connect to a PostgreSQL database:
+- PHOENIX_POSTGRES_HOST
+- PHOENIX_POSTGRES_PORT
+- PHOENIX_POSTGRES_USER
+- PHOENIX_POSTGRES_PASSWORD
+- PHOENIX_POSTGRES_DB
+"""
+ENV_PHOENIX_POSTGRES_PORT = "PHOENIX_POSTGRES_PORT"
+"""
+Used with PHOENIX_POSTGRES_HOST to specify the port to use for the PostgreSQL database.
+"""
+ENV_PHOENIX_POSTGRES_USER = "PHOENIX_POSTGRES_USER"
+"""
+Used with PHOENIX_POSTGRES_HOST to specify the user to use for the PostgreSQL database (required).
+"""
+ENV_PHOENIX_POSTGRES_PASSWORD = "PHOENIX_POSTGRES_PASSWORD"
+"""
+Used with PHOENIX_POSTGRES_HOST to specify the password to use for the PostgreSQL database
+(required).
+"""
+ENV_PHOENIX_POSTGRES_DB = "PHOENIX_POSTGRES_DB"
+"""
+Used with PHOENIX_POSTGRES_HOST to specify the database to use for the PostgreSQL database.
+"""
 ENV_PHOENIX_SQL_DATABASE_SCHEMA = "PHOENIX_SQL_DATABASE_SCHEMA"
 """
 The schema to use for the PostgresSQL database. (This is ignored for SQLite.)
@@ -635,11 +662,11 @@ def get_env_database_connection_str() -> str:
 
 
 def get_env_postgres_connection_str() -> Optional[str]:
-    pg_user = os.getenv("PHOENIX_POSTGRES_USER")
-    pg_password = os.getenv("PHOENIX_POSTGRES_PASSWORD")
-    pg_host = os.getenv("PHOENIX_POSTGRES_HOST")
-    pg_port = os.getenv("PHOENIX_POSTGRES_PORT")
-    pg_db = os.getenv("PHOENIX_POSTGRES_DB")
+    pg_user = os.getenv(ENV_PHOENIX_POSTGRES_USER)
+    pg_password = os.getenv(ENV_PHOENIX_POSTGRES_PASSWORD)
+    pg_host = os.getenv(ENV_PHOENIX_POSTGRES_HOST)
+    pg_port = os.getenv(ENV_PHOENIX_POSTGRES_PORT)
+    pg_db = os.getenv(ENV_PHOENIX_POSTGRES_DB)
 
     if pg_host and ":" in pg_host:
         pg_host, parsed_port = pg_host.split(":")

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/db/models.py

@@ -271,6 +271,7 @@ class ExperimentRunOutput(TypedDict, total=False):
 
 
 class Base(DeclarativeBase):
+    id: Mapped[int] = mapped_column(Integer, primary_key=True)
     # Enforce best practices for naming constraints
     # https://alembic.sqlalchemy.org/en/latest/naming.html#integration-of-naming-conventions-into-operations-autogenerate
     metadata = MetaData(
@@ -292,7 +293,6 @@ class Base(DeclarativeBase):
 
 class Project(Base):
     __tablename__ = "projects"
-    id: Mapped[int] = mapped_column(primary_key=True)
     name: Mapped[str]
     description: Mapped[Optional[str]]
     gradient_start_color: Mapped[str] = mapped_column(
@@ -325,7 +325,6 @@ class Project(Base):
 
 class ProjectSession(Base):
     __tablename__ = "project_sessions"
-    id: Mapped[int] = mapped_column(primary_key=True)
     session_id: Mapped[str] = mapped_column(String, nullable=False, unique=True)
     project_id: Mapped[int] = mapped_column(
         ForeignKey("projects.id", ondelete="CASCADE"),
@@ -343,7 +342,6 @@ class ProjectSession(Base):
 
 class Trace(Base):
     __tablename__ = "traces"
-    id: Mapped[int] = mapped_column(primary_key=True)
     project_rowid: Mapped[int] = mapped_column(
         ForeignKey("projects.id", ondelete="CASCADE"),
         nullable=False,
@@ -393,7 +391,6 @@ class Trace(Base):
 
 class Span(Base):
     __tablename__ = "spans"
-    id: Mapped[int] = mapped_column(primary_key=True)
     trace_rowid: Mapped[int] = mapped_column(
         ForeignKey("traces.id", ondelete="CASCADE"),
         index=True,
@@ -660,7 +657,6 @@ async def init_models(engine: AsyncEngine) -> None:
 
 class SpanAnnotation(Base):
     __tablename__ = "span_annotations"
-    id: Mapped[int] = mapped_column(primary_key=True)
     span_rowid: Mapped[int] = mapped_column(
         ForeignKey("spans.id", ondelete="CASCADE"),
         index=True,
@@ -687,7 +683,6 @@ class SpanAnnotation(Base):
 
 class TraceAnnotation(Base):
     __tablename__ = "trace_annotations"
-    id: Mapped[int] = mapped_column(primary_key=True)
     trace_rowid: Mapped[int] = mapped_column(
         ForeignKey("traces.id", ondelete="CASCADE"),
         index=True,
@@ -714,7 +709,6 @@ class TraceAnnotation(Base):
 
 class DocumentAnnotation(Base):
     __tablename__ = "document_annotations"
-    id: Mapped[int] = mapped_column(primary_key=True)
     span_rowid: Mapped[int] = mapped_column(
         ForeignKey("spans.id", ondelete="CASCADE"),
         index=True,
@@ -745,7 +739,6 @@ class DocumentAnnotation(Base):
 
 class Dataset(Base):
     __tablename__ = "datasets"
-    id: Mapped[int] = mapped_column(primary_key=True)
     name: Mapped[str] = mapped_column(unique=True)
     description: Mapped[Optional[str]]
     metadata_: Mapped[dict[str, Any]] = mapped_column("metadata")
@@ -805,7 +798,6 @@ class Dataset(Base):
 
 class DatasetVersion(Base):
     __tablename__ = "dataset_versions"
-    id: Mapped[int] = mapped_column(primary_key=True)
     dataset_id: Mapped[int] = mapped_column(
         ForeignKey("datasets.id", ondelete="CASCADE"),
         index=True,
@@ -817,7 +809,6 @@ class DatasetVersion(Base):
 
 class DatasetExample(Base):
     __tablename__ = "dataset_examples"
-    id: Mapped[int] = mapped_column(primary_key=True)
     dataset_id: Mapped[int] = mapped_column(
         ForeignKey("datasets.id", ondelete="CASCADE"),
         index=True,
@@ -834,7 +825,6 @@ class DatasetExample(Base):
 
 class DatasetExampleRevision(Base):
     __tablename__ = "dataset_example_revisions"
-    id: Mapped[int] = mapped_column(primary_key=True)
     dataset_example_id: Mapped[int] = mapped_column(
         ForeignKey("dataset_examples.id", ondelete="CASCADE"),
         index=True,
@@ -863,7 +853,6 @@ class DatasetExampleRevision(Base):
 
 class Experiment(Base):
     __tablename__ = "experiments"
-    id: Mapped[int] = mapped_column(primary_key=True)
     dataset_id: Mapped[int] = mapped_column(
         ForeignKey("datasets.id", ondelete="CASCADE"),
         index=True,
@@ -885,7 +874,6 @@ class Experiment(Base):
 
 class ExperimentRun(Base):
     __tablename__ = "experiment_runs"
-    id: Mapped[int] = mapped_column(primary_key=True)
     experiment_id: Mapped[int] = mapped_column(
         ForeignKey("experiments.id", ondelete="CASCADE"),
         index=True,
@@ -928,7 +916,6 @@ class ExperimentRun(Base):
 
 class ExperimentRunAnnotation(Base):
     __tablename__ = "experiment_run_annotations"
-    id: Mapped[int] = mapped_column(primary_key=True)
     experiment_run_id: Mapped[int] = mapped_column(
         ForeignKey("experiment_runs.id", ondelete="CASCADE"),
         index=True,
@@ -956,14 +943,12 @@ class ExperimentRunAnnotation(Base):
 
 class UserRole(Base):
     __tablename__ = "user_roles"
-    id: Mapped[int] = mapped_column(primary_key=True)
     name: Mapped[str] = mapped_column(unique=True, index=True)
     users: Mapped[list["User"]] = relationship("User", back_populates="role")
 
 
 class User(Base):
     __tablename__ = "users"
-    id: Mapped[int] = mapped_column(primary_key=True)
     user_role_id: Mapped[int] = mapped_column(
         ForeignKey("user_roles.id", ondelete="CASCADE"),
         index=True,
@@ -1038,7 +1023,6 @@ class User(Base):
 
 class PasswordResetToken(Base):
     __tablename__ = "password_reset_tokens"
-    id: Mapped[int] = mapped_column(primary_key=True)
     user_id: Mapped[int] = mapped_column(
         ForeignKey("users.id", ondelete="CASCADE"),
         unique=True,
@@ -1052,7 +1036,6 @@ class PasswordResetToken(Base):
 
 class RefreshToken(Base):
     __tablename__ = "refresh_tokens"
-    id: Mapped[int] = mapped_column(primary_key=True)
     user_id: Mapped[int] = mapped_column(
         ForeignKey("users.id", ondelete="CASCADE"),
         index=True,
@@ -1065,7 +1048,6 @@ class RefreshToken(Base):
 
 class AccessToken(Base):
     __tablename__ = "access_tokens"
-    id: Mapped[int] = mapped_column(primary_key=True)
     user_id: Mapped[int] = mapped_column(
         ForeignKey("users.id", ondelete="CASCADE"),
         index=True,
@@ -1083,7 +1065,6 @@ class AccessToken(Base):
 
 class ApiKey(Base):
     __tablename__ = "api_keys"
-    id: Mapped[int] = mapped_column(primary_key=True)
     user_id: Mapped[int] = mapped_column(
         ForeignKey("users.id", ondelete="CASCADE"),
         index=True,
@@ -1099,7 +1080,6 @@ class ApiKey(Base):
 class PromptLabel(Base):
     __tablename__ = "prompt_labels"
 
-    id: Mapped[int] = mapped_column(primary_key=True)
     name: Mapped[str] = mapped_column(String, unique=True, index=True, nullable=False)
     description: Mapped[Optional[str]]
     color: Mapped[str] = mapped_column(String, nullable=True)
@@ -1115,7 +1095,6 @@ class PromptLabel(Base):
 class Prompt(Base):
     __tablename__ = "prompts"
 
-    id: Mapped[int] = mapped_column(primary_key=True)
     source_prompt_id: Mapped[Optional[int]] = mapped_column(
         ForeignKey("prompts.id", ondelete="SET NULL"),
         index=True,
@@ -1154,7 +1133,6 @@ class Prompt(Base):
 class PromptPromptLabel(Base):
     __tablename__ = "prompts_prompt_labels"
 
-    id: Mapped[int] = mapped_column(primary_key=True)
     prompt_label_id: Mapped[int] = mapped_column(
         ForeignKey("prompt_labels.id", ondelete="CASCADE"),
         index=True,
@@ -1177,7 +1155,6 @@ class PromptPromptLabel(Base):
 class PromptVersion(Base):
     __tablename__ = "prompt_versions"
 
-    id: Mapped[int] = mapped_column(primary_key=True)
     prompt_id: Mapped[int] = mapped_column(
         ForeignKey("prompts.id", ondelete="CASCADE"),
         index=True,
@@ -1227,7 +1204,6 @@ class PromptVersion(Base):
 class PromptVersionTag(Base):
     __tablename__ = "prompt_version_tags"
 
-    id: Mapped[int] = mapped_column(primary_key=True)
     name: Mapped[Identifier] = mapped_column(_Identifier, nullable=False)
     description: Mapped[Optional[str]] = mapped_column(String, nullable=True)
     prompt_id: Mapped[int] = mapped_column(

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/context.py

@@ -41,8 +41,8 @@ from phoenix.server.api.dataloaders import (
     SpanByIdDataLoader,
     SpanDatasetExamplesDataLoader,
     SpanDescendantsDataLoader,
-    SpanFieldsDataLoader,
     SpanProjectsDataLoader,
+    TableFieldsDataLoader,
     TokenCountDataLoader,
     TraceByTraceIdsDataLoader,
     TraceRootSpansDataLoader,
@@ -76,6 +76,7 @@ class DataLoaders:
     experiment_sequence_number: ExperimentSequenceNumberDataLoader
     latency_ms_quantile: LatencyMsQuantileDataLoader
     min_start_or_max_end_times: MinStartOrMaxEndTimeDataLoader
+    project_fields: TableFieldsDataLoader
     prompt_version_sequence_number: PromptVersionSequenceNumberDataLoader
     record_counts: RecordCountDataLoader
     session_first_inputs: SessionIODataLoader
@@ -88,10 +89,11 @@ class DataLoaders:
     span_by_id: SpanByIdDataLoader
     span_dataset_examples: SpanDatasetExamplesDataLoader
     span_descendants: SpanDescendantsDataLoader
-    span_fields: SpanFieldsDataLoader
+    span_fields: TableFieldsDataLoader
     span_projects: SpanProjectsDataLoader
     token_counts: TokenCountDataLoader
     trace_by_trace_ids: TraceByTraceIdsDataLoader
+    trace_fields: TableFieldsDataLoader
     trace_root_spans: TraceRootSpansDataLoader
     project_by_name: ProjectByNameDataLoader
     users: UsersDataLoader

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/dataloaders/__init__.py

@@ -29,8 +29,8 @@ from .span_annotations import SpanAnnotationsDataLoader
 from .span_by_id import SpanByIdDataLoader
 from .span_dataset_examples import SpanDatasetExamplesDataLoader
 from .span_descendants import SpanDescendantsDataLoader
-from .span_fields import SpanFieldsDataLoader
 from .span_projects import SpanProjectsDataLoader
+from .table_fields import TableFieldsDataLoader
 from .token_counts import TokenCountCache, TokenCountDataLoader
 from .trace_by_trace_ids import TraceByTraceIdsDataLoader
 from .trace_root_spans import TraceRootSpansDataLoader
@@ -60,11 +60,11 @@ __all__ = [
     "SessionNumTracesWithErrorDataLoader",
     "SessionTokenUsagesDataLoader",
     "SessionTraceLatencyMsQuantileDataLoader",
-    "SpanFieldsDataLoader",
     "SpanByIdDataLoader",
     "SpanDatasetExamplesDataLoader",
     "SpanDescendantsDataLoader",
     "SpanProjectsDataLoader",
+    "TableFieldsDataLoader",
     "TokenCountDataLoader",
     "TraceByTraceIdsDataLoader",
     "TraceRootSpansDataLoader",

arize_phoenix-8.2.2/src/phoenix/server/api/dataloaders/span_fields.py → arize_phoenix-8.3.0/src/phoenix/server/api/dataloaders/table_fields.py

@@ -8,35 +8,36 @@ from typing_extensions import TypeAlias
 from phoenix.db import models
 from phoenix.server.types import DbSessionFactory
 
-SpanRowId: TypeAlias = int
+RowId: TypeAlias = int
 
-Key: TypeAlias = tuple[SpanRowId, QueryableAttribute[Any]]
+Key: TypeAlias = tuple[RowId, QueryableAttribute[Any]]
 Result: TypeAlias = Any
 
-
 _ResultColumnPosition: TypeAlias = int
 _AttrStrIdentifier: TypeAlias = str
 
 
-class SpanFieldsDataLoader(DataLoader[Key, Result]):
-    def __init__(self, db: DbSessionFactory) -> None:
+class TableFieldsDataLoader(DataLoader[Key, Result]):
+    def __init__(self, db: DbSessionFactory, table: type[models.Base]) -> None:
         super().__init__(load_fn=self._load_fn)
         self._db = db
+        self._table = table
 
     async def _load_fn(self, keys: Iterable[Key]) -> list[Union[Result, ValueError]]:
-        result: dict[tuple[SpanRowId, _AttrStrIdentifier], Result] = {}
-        stmt, attr_strs = _get_stmt(keys)
+        result: dict[tuple[RowId, _AttrStrIdentifier], Result] = {}
+        stmt, attr_strs = _get_stmt(keys, self._table)
         async with self._db() as session:
             data = await session.stream(stmt)
             async for row in data:
-                span_rowid: SpanRowId = row[0]  # models.Span's primary key
+                rowid: RowId = row[0]  # models.Span's primary key
                 for i, value in enumerate(row[1:]):
-                    result[span_rowid, attr_strs[i]] = value
-        return [result.get((span_rowid, str(attr))) for span_rowid, attr in keys]
+                    result[rowid, attr_strs[i]] = value
+        return [result.get((rowid, str(attr))) for rowid, attr in keys]
 
 
 def _get_stmt(
-    keys: Iterable[Key],
+    keys: Iterable[tuple[RowId, QueryableAttribute[Any]]],
+    table: type[models.Base],
 ) -> tuple[
     Select[Any],
     dict[_ResultColumnPosition, _AttrStrIdentifier],
@@ -52,7 +53,8 @@ def _get_stmt(
 
     Args:
         keys (list[Key]): A list of tuples, where each tuple contains an integer ID, i.e. the
-            primary key of models.Span, and a QueryableAttribute.
+            primary key of table, and a QueryableAttribute.
+        table (models.Base): The table to query.
 
     Returns:
         tuple: A tuple containing:
@@ -61,16 +63,16 @@ def _get_stmt(
                 at the second column (because the first column is the span's primary key)--in the
                 result to the attribute's string identifier.
     """
-    span_rowids: set[SpanRowId] = set()
+    rowids: set[RowId] = set()
     attrs: dict[_AttrStrIdentifier, QueryableAttribute[Any]] = {}
     joins = set()
-    for span_rowid, attr in keys:
-        span_rowids.add(span_rowid)
+    for rowid, attr in keys:
+        rowids.add(rowid)
         attrs[str(attr)] = attr
-        if (entity := attr.parent.entity) is not models.Span:
+        if (entity := attr.parent.entity) is not table:
             joins.add(entity)
-    stmt = select(models.Span.id).where(models.Span.id.in_(span_rowids))
-    for table in joins:
-        stmt = stmt.join(table)
+    stmt = select(table.id).where(table.id.in_(rowids))
+    for other_table in joins:
+        stmt = stmt.join(other_table)
     identifiers, columns = zip(*attrs.items())
     return stmt.add_columns(*columns), dict(enumerate(identifiers))

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/helpers/playground_clients.py

@@ -806,6 +806,10 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
                     raise NotImplementedError
                 elif isinstance(event, anthropic_streaming._types.CitationEvent):
                     raise NotImplementedError
+                elif isinstance(event, anthropic_streaming._types.ThinkingEvent):
+                    raise NotImplementedError
+                elif isinstance(event, anthropic_streaming._types.SignatureEvent):
+                    raise NotImplementedError
                 else:
                     assert_never(event)
 

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/helpers/prompts/models.py

@@ -348,6 +348,7 @@ class AnthropicToolDefinition(PromptModel):
 class PromptOpenAIInvocationParametersContent(PromptModel):
     temperature: float = UNDEFINED
     max_tokens: int = UNDEFINED
+    max_completion_tokens: int = UNDEFINED
     frequency_penalty: float = UNDEFINED
     presence_penalty: float = UNDEFINED
     top_p: float = UNDEFINED

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/queries.py

@@ -66,7 +66,7 @@ from phoenix.server.api.types.PromptVersion import PromptVersion, to_gql_prompt_
 from phoenix.server.api.types.SortDir import SortDir
 from phoenix.server.api.types.Span import Span
 from phoenix.server.api.types.SystemApiKey import SystemApiKey
-from phoenix.server.api.types.Trace import to_gql_trace
+from phoenix.server.api.types.Trace import Trace
 from phoenix.server.api.types.User import User, to_gql_user
 from phoenix.server.api.types.UserApiKey import UserApiKey, to_gql_api_key
 from phoenix.server.api.types.UserRole import UserRole
@@ -236,10 +236,8 @@ class Query:
             projects = await session.stream_scalars(stmt)
             data = [
                 Project(
-                    id_attr=project.id,
-                    name=project.name,
-                    gradient_start_color=project.gradient_start_color,
-                    gradient_end_color=project.gradient_end_color,
+                    project_rowid=project.id,
+                    db_project=project,
                 )
                 async for project in projects
             ]
@@ -448,21 +446,14 @@ class Query:
             embedding_dimension = info.context.model.embedding_dimensions[node_id]
             return to_gql_embedding_dimension(node_id, embedding_dimension)
         elif type_name == "Project":
-            project_stmt = select(
-                models.Project.id,
-                models.Project.name,
-                models.Project.gradient_start_color,
-                models.Project.gradient_end_color,
-            ).where(models.Project.id == node_id)
+            project_stmt = select(models.Project).filter_by(id=node_id)
             async with info.context.db() as session:
-                project = (await session.execute(project_stmt)).first()
+                project = await session.scalar(project_stmt)
             if project is None:
                 raise NotFound(f"Unknown project: {id}")
             return Project(
-                id_attr=project.id,
-                name=project.name,
-                gradient_start_color=project.gradient_start_color,
-                gradient_end_color=project.gradient_end_color,
+                project_rowid=project.id,
+                db_project=project,
             )
         elif type_name == "Trace":
             trace_stmt = select(models.Trace).filter_by(id=node_id)
@@ -470,7 +461,7 @@ class Query:
                 trace = await session.scalar(trace_stmt)
             if trace is None:
                 raise NotFound(f"Unknown trace: {id}")
-            return to_gql_trace(trace)
+            return Trace(trace_rowid=trace.id, db_trace=trace)
         elif type_name == Span.__name__:
             span_stmt = (
                 select(models.Span)

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/types/Experiment.py

@@ -122,10 +122,8 @@ class Experiment(Node):
             return None
 
         return Project(
-            id_attr=db_project.id,
-            name=db_project.name,
-            gradient_start_color=db_project.gradient_start_color,
-            gradient_end_color=db_project.gradient_end_color,
+            project_rowid=db_project.id,
+            db_project=db_project,
         )
 
     @strawberry.field

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/types/ExperimentRun.py

@@ -20,7 +20,7 @@ from phoenix.server.api.types.pagination import (
     CursorString,
     connection_from_list,
 )
-from phoenix.server.api.types.Trace import Trace, to_gql_trace
+from phoenix.server.api.types.Trace import Trace
 
 if TYPE_CHECKING:
     from phoenix.server.api.types.DatasetExample import DatasetExample
@@ -64,7 +64,7 @@ class ExperimentRun(Node):
         dataloader = info.context.data_loaders.trace_by_trace_ids
         if (trace := await dataloader.load(self.trace_id)) is None:
             return None
-        return to_gql_trace(trace)
+        return Trace(trace_rowid=trace.id, db_trace=trace)
 
     @strawberry.field
     async def example(

{arize_phoenix-8.2.2 → arize_phoenix-8.3.0}/src/phoenix/server/api/types/ExperimentRunAnnotation.py

@@ -8,7 +8,7 @@ from strawberry.scalars import JSON
 
 from phoenix.db import models
 from phoenix.server.api.types.AnnotatorKind import ExperimentRunAnnotatorKind
-from phoenix.server.api.types.Trace import Trace, to_gql_trace
+from phoenix.server.api.types.Trace import Trace
 
 
 @strawberry.type
@@ -32,7 +32,7 @@ class ExperimentRunAnnotation(Node):
         dataloader = info.context.data_loaders.trace_by_trace_ids
         if (trace := await dataloader.load(self.trace_id)) is None:
             return None
-        return to_gql_trace(trace)
+        return Trace(trace_rowid=trace.id, db_trace=trace)
 
 
 def to_gql_experiment_run_annotation(