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

{arize_phoenix-8.2.2 → arize_phoenix-8.4.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.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 8.2.2
+Version: 8.4.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'
@@ -162,6 +162,7 @@ Phoenix is an open-source AI observability platform designed for experimentation
 - [**_Datasets_**](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets) - Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
 - [**_Experiments_**](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets#experiments) - Track and evaluate changes to prompts, LLMs, and retrieval.
 - [**_Playground_**](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)- Optimize prompts, compare models, adjust parameters, and replay traced LLM calls.
+- [**_Prompt Mangement_**](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts/prompt-management)- Manage and test prompt changes systematically using version control, tagging, and experimentation.
 
 Phoenix is vendor and language agnostic with out-of-the-box support for popular frameworks (🦙[LlamaIndex](https://docs.arize.com/phoenix/tracing/integrations-tracing/llamaindex), 🦜⛓[LangChain](https://docs.arize.com/phoenix/tracing/integrations-tracing/langchain), [Haystack](https://docs.arize.com/phoenix/tracing/integrations-tracing/haystack), 🧩[DSPy](https://docs.arize.com/phoenix/tracing/integrations-tracing/dspy)) and LLM providers ([OpenAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/openai), [Bedrock](https://docs.arize.com/phoenix/tracing/integrations-tracing/bedrock), [MistralAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/mistralai), [VertexAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/vertexai), [LiteLLM](https://docs.arize.com/phoenix/tracing/integrations-tracing/litellm), and more). For details on auto-instrumentation, check out the [OpenInference](https://github.com/Arize-ai/openinference) project.
 

{arize_phoenix-8.2.2 → arize_phoenix-8.4.0}/README.md

@@ -34,6 +34,7 @@ Phoenix is an open-source AI observability platform designed for experimentation
 - [**_Datasets_**](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets) - Create versioned datasets of examples for experimentation, evaluation, and fine-tuning.
 - [**_Experiments_**](https://docs.arize.com/phoenix/datasets-and-experiments/overview-datasets#experiments) - Track and evaluate changes to prompts, LLMs, and retrieval.
 - [**_Playground_**](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts)- Optimize prompts, compare models, adjust parameters, and replay traced LLM calls.
+- [**_Prompt Mangement_**](https://docs.arize.com/phoenix/prompt-engineering/overview-prompts/prompt-management)- Manage and test prompt changes systematically using version control, tagging, and experimentation.
 
 Phoenix is vendor and language agnostic with out-of-the-box support for popular frameworks (🦙[LlamaIndex](https://docs.arize.com/phoenix/tracing/integrations-tracing/llamaindex), 🦜⛓[LangChain](https://docs.arize.com/phoenix/tracing/integrations-tracing/langchain), [Haystack](https://docs.arize.com/phoenix/tracing/integrations-tracing/haystack), 🧩[DSPy](https://docs.arize.com/phoenix/tracing/integrations-tracing/dspy)) and LLM providers ([OpenAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/openai), [Bedrock](https://docs.arize.com/phoenix/tracing/integrations-tracing/bedrock), [MistralAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/mistralai), [VertexAI](https://docs.arize.com/phoenix/tracing/integrations-tracing/vertexai), [LiteLLM](https://docs.arize.com/phoenix/tracing/integrations-tracing/litellm), and more). For details on auto-instrumentation, check out the [OpenInference](https://github.com/Arize-ai/openinference) project.
 

{arize_phoenix-8.2.2 → arize_phoenix-8.4.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.4.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.4.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.4.0}/src/phoenix/server/api/context.py

@@ -29,6 +29,7 @@ from phoenix.server.api.dataloaders import (
     ExperimentSequenceNumberDataLoader,
     LatencyMsQuantileDataLoader,
     MinStartOrMaxEndTimeDataLoader,
+    NumChildSpansDataLoader,
     ProjectByNameDataLoader,
     PromptVersionSequenceNumberDataLoader,
     RecordCountDataLoader,
@@ -41,8 +42,8 @@ from phoenix.server.api.dataloaders import (
     SpanByIdDataLoader,
     SpanDatasetExamplesDataLoader,
     SpanDescendantsDataLoader,
-    SpanFieldsDataLoader,
     SpanProjectsDataLoader,
+    TableFieldsDataLoader,
     TokenCountDataLoader,
     TraceByTraceIdsDataLoader,
     TraceRootSpansDataLoader,
@@ -76,6 +77,8 @@ class DataLoaders:
     experiment_sequence_number: ExperimentSequenceNumberDataLoader
     latency_ms_quantile: LatencyMsQuantileDataLoader
     min_start_or_max_end_times: MinStartOrMaxEndTimeDataLoader
+    num_child_spans: NumChildSpansDataLoader
+    project_fields: TableFieldsDataLoader
     prompt_version_sequence_number: PromptVersionSequenceNumberDataLoader
     record_counts: RecordCountDataLoader
     session_first_inputs: SessionIODataLoader
@@ -88,10 +91,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.4.0}/src/phoenix/server/api/dataloaders/__init__.py

@@ -17,6 +17,7 @@ from .experiment_run_counts import ExperimentRunCountsDataLoader
 from .experiment_sequence_number import ExperimentSequenceNumberDataLoader
 from .latency_ms_quantile import LatencyMsQuantileCache, LatencyMsQuantileDataLoader
 from .min_start_or_max_end_times import MinStartOrMaxEndTimeCache, MinStartOrMaxEndTimeDataLoader
+from .num_child_spans import NumChildSpansDataLoader
 from .project_by_name import ProjectByNameDataLoader
 from .prompt_version_sequence_number import PromptVersionSequenceNumberDataLoader
 from .record_counts import RecordCountCache, RecordCountDataLoader
@@ -29,8 +30,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
@@ -53,6 +54,7 @@ __all__ = [
     "ExperimentSequenceNumberDataLoader",
     "LatencyMsQuantileDataLoader",
     "MinStartOrMaxEndTimeDataLoader",
+    "NumChildSpansDataLoader",
     "PromptVersionSequenceNumberDataLoader",
     "RecordCountDataLoader",
     "SessionIODataLoader",
@@ -60,11 +62,11 @@ __all__ = [
     "SessionNumTracesWithErrorDataLoader",
     "SessionTokenUsagesDataLoader",
     "SessionTraceLatencyMsQuantileDataLoader",
-    "SpanFieldsDataLoader",
     "SpanByIdDataLoader",
     "SpanDatasetExamplesDataLoader",
     "SpanDescendantsDataLoader",
     "SpanProjectsDataLoader",
+    "TableFieldsDataLoader",
     "TokenCountDataLoader",
     "TraceByTraceIdsDataLoader",
     "TraceRootSpansDataLoader",

arize_phoenix-8.4.0/src/phoenix/server/api/dataloaders/num_child_spans.py

@@ -0,0 +1,35 @@
+from typing import Iterable
+
+from sqlalchemy import func, select
+from strawberry.dataloader import DataLoader
+from typing_extensions import TypeAlias
+
+from phoenix.db import models
+from phoenix.server.types import DbSessionFactory
+
+SpanRowId: TypeAlias = int
+
+Key: TypeAlias = SpanRowId
+Result: TypeAlias = int
+
+
+class NumChildSpansDataLoader(DataLoader[Key, Result]):
+    def __init__(self, db: DbSessionFactory) -> None:
+        super().__init__(load_fn=self._load_fn)
+        self._db = db
+
+    async def _load_fn(self, keys: Iterable[Key]) -> list[Result]:
+        span_rowids = list(set(keys))
+        result: dict[Key, Result] = {}
+        children = select(models.Span).alias("children")
+        stmt = (
+            select(models.Span.id, func.count())
+            .where(models.Span.id.in_(span_rowids))
+            .join(children, children.c.parent_id == models.Span.span_id)
+            .group_by(models.Span.id)
+        )
+        async with self._db() as session:
+            data = await session.stream(stmt)
+            async for span_rowid, num_child_spans in data:
+                result[span_rowid] = num_child_spans
+        return [result.get(span_rowid, 0) for span_rowid in keys]

arize_phoenix-8.2.2/src/phoenix/server/api/dataloaders/span_fields.py → arize_phoenix-8.4.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.4.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.4.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