arize-phoenix 11.9.0__py3-none-any.whl → 11.10.1__py3-none-any.whl

{arize_phoenix-11.9.0.dist-info → arize_phoenix-11.10.1.dist-info}/METADATA

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

{arize_phoenix-11.9.0.dist-info → arize_phoenix-11.10.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=2kHfT3BNOVd4dAO1bq-syEQbHSG8oX2-7NhOwK2QREk,896
-phoenix/version.py,sha256=SXY1e-dnGDUvRtF27y_ABoyLFsY29JXPVQsa1fJZTDU,23
+phoenix/version.py,sha256=xvvOboOAs4Htuc0_lAWkZ6Ru9ZUbUhsIFFXYCIMHAAE,24
 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
@@ -176,8 +176,8 @@ phoenix/server/api/helpers/__init__.py,sha256=m2-xaSPqUiSs91k62JaRDjFNfl-1byxBfY
 phoenix/server/api/helpers/annotations.py,sha256=9gMXKpMTfWEChoSCnvdWYuyB0hlSnNOp-qUdar9Vono,262
 phoenix/server/api/helpers/dataset_helpers.py,sha256=3bdGBoUzqrtg-sr5p2wpQLOU6dhg_3TKFHNeJj8p0TU,9155
 phoenix/server/api/helpers/experiment_run_filters.py,sha256=DOnVwrmn39eAkk2mwuZP8kIcAnR5jrOgllEwWSjsw94,29893
-phoenix/server/api/helpers/playground_clients.py,sha256=vcLc8Ps8ZaZCUJnsN7retFbUR0LjF8nKTtN_OjBfS3E,71172
-phoenix/server/api/helpers/playground_registry.py,sha256=CPLMziFB2wmr-dfbx7VbzO2f8YIG_k5RftzvGXYGQ1w,2570
+phoenix/server/api/helpers/playground_clients.py,sha256=Fq4DNVIdnCiiVt0bh5mrZ7dJb2oOQcLjTttfq0Wcuv0,73589
+phoenix/server/api/helpers/playground_registry.py,sha256=n0v4-KnvZJxeaEwOla5qBbnOQjSWznKmMhZnh9ziJt0,2584
 phoenix/server/api/helpers/playground_spans.py,sha256=QpXwPl_fFNwm_iA1A77XApUyXMl1aDmonw8aXuNZ_4k,17132
 phoenix/server/api/helpers/prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/helpers/prompts/models.py,sha256=nlPtLZaGcHfWNRR0iNRaBUv8eoKOnoGqRm6zadrTt0I,23547
@@ -320,7 +320,7 @@ phoenix/server/api/types/ModelInterface.py,sha256=Qe7H23wDb_Q2-HmeY2t0R5Jsn4aAfY
 phoenix/server/api/types/NumericRange.py,sha256=afEjgF97Go_OvmjMggbPBt-zGM8IONewAyEiKEHRds0,192
 phoenix/server/api/types/PerformanceMetric.py,sha256=KFkmJDqP43eDUtARQOUqR7NYcxvL6Vh2uisHWU6H3ko,387
 phoenix/server/api/types/PlaygroundModel.py,sha256=IqJFxsAAJMRyaFI9ryI3GQrpFOJ5Llf6kIutEO-tFvM,321
-phoenix/server/api/types/Project.py,sha256=gtKuIrKZXN4oP5tfY403zAnyv6yKOJ4TEG89MXZw1XY,63039
+phoenix/server/api/types/Project.py,sha256=FE93mSrWKv1rcCPECfp1hsCen2012SPp-1pCEVuq5ic,64039
 phoenix/server/api/types/ProjectSession.py,sha256=uwqTsDTfSGz13AvP-cwS_mJR5JZ1lHqu10ungbl7g5s,6245
 phoenix/server/api/types/ProjectTraceRetentionPolicy.py,sha256=tYy2kgalPDyuaYZr0VUHjH0YpXaiF_QOzg5yfaV_c7c,3782
 phoenix/server/api/types/Prompt.py,sha256=ccP4eq1e38xbF0afclGWLOuDpBVpNbJ3AOSRClF8yFQ,4955
@@ -387,19 +387,19 @@ 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=QXaZIXfZ1XJbL6ZbmTvaV7nhgFrHNPHk39Z6-KJ2CvU,2165
-phoenix/server/static/assets/components-IBd-PDxA.js,sha256=kymjnoRdLP2OwD5Vj2ceDzClon_jY_9ehCAB4umjzPw,620385
-phoenix/server/static/assets/index-B8EBC_Z5.js,sha256=GIGSeOxAeTTcPl9CydKaWjtEcTxIAlofTEf8Y7olppo,62006
-phoenix/server/static/assets/pages-6D1duYIe.js,sha256=9IoSD0ulNxPRDTQTkKg8AVylMlOPCOMbVkJPCEHOWjg,1173753
-phoenix/server/static/assets/vendor-BzZ0oklU.js,sha256=hZdQcaVySUPWh4GXRhuL-PhUy2K4sJLs6gYxde6qB_I,2748936
-phoenix/server/static/assets/vendor-WIZid84E.css,sha256=spZD2r7XL5GfLO13ln-IuXfnjAref8l6g_n_AvxxOlI,5517
-phoenix/server/static/assets/vendor-arizeai-CvjUqTrl.js,sha256=Db8aE6L0fWXbql2WfogS6GR0xM-uMYbdxT0zkQlUBAc,151750
-phoenix/server/static/assets/vendor-codemirror-CKK25Gd7.js,sha256=SlvwAdf_42ZJzqkhPA-sl18TtA8CvBc3KWU5oeakfWw,781264
-phoenix/server/static/assets/vendor-recharts-CWtaRhQC.js,sha256=KYjaE5frWsva6vjzSpMLXt_Af2eJ__67hLvazywljcQ,231651
-phoenix/server/static/assets/vendor-shiki-D30GF-p9.js,sha256=zmvCVHLZ-XkVKFmo5ZLIWxvvcjFBpB4bza9ElUVNLxE,8980312
+phoenix/server/static/.vite/manifest.json,sha256=67Jr1Pbvt4QUZP4O98Fe_9tPLJvVHSKqSJeYcfC2oVk,2165
+phoenix/server/static/assets/components-XAeml0-1.js,sha256=aCeHz5UUtnMLqWDeVRMmHJnjMHgwEDjpdRoz_tJ063s,620501
+phoenix/server/static/assets/index-D7EtHUpz.js,sha256=FBp-_XLUtdcJ3wLAdu0peRKAPTkU2Ceu_U2Fu-QRfDg,62601
+phoenix/server/static/assets/pages-CPfaxiKa.js,sha256=EyoqjFXTFOgpww-GDBYnH7KfsLfV84OI06WYG0xmUjo,1190498
+phoenix/server/static/assets/vendor-CqDb5u4o.css,sha256=zIyFiNJKxMaQk8AvtLgt1rR01oO10d1MFndSDKH9Clw,5517
+phoenix/server/static/assets/vendor-DhvamIr8.js,sha256=hZdQcaVySUPWh4GXRhuL-PhUy2K4sJLs6gYxde6qB_I,2748936
+phoenix/server/static/assets/vendor-arizeai-4fVwwnrI.js,sha256=8eBZfGf2fjjNoWDW31mHXdxkqSlWItGj7eHhbnH92Qk,151750
+phoenix/server/static/assets/vendor-codemirror-DRfFHb57.js,sha256=v-q8mq6f-EbjtvgzLTQYF-SNKX4R6LOGJe5ucOMqx1w,781264
+phoenix/server/static/assets/vendor-recharts-w6bSawXG.js,sha256=uTuxE0vslP7_y-F1dOU5K2nXke7qe8JMdE2SnRGCVKM,231651
+phoenix/server/static/assets/vendor-shiki-CplrhwOk.js,sha256=ISu7sYmhh_FDTmnBN-icbipm6fa2InqOeQTo2JFr3LI,8980312
 phoenix/server/static/assets/vendor-three-C5WAXd5r.js,sha256=ELkg06u70N7h8oFmvqdoHyPuUf9VgGEWeT4LKFx4VWo,620975
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/templates/index.html,sha256=rTdJZOlbZmm1dMCrHWikAwcecZDxduPU5zCFd2h2cbs,6819
+phoenix/server/templates/index.html,sha256=3VMDmbxYwo3OoqiQyFojU6JaMLKr5k8rITacYS7HTbs,6922
 phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/session/client.py,sha256=uw5WlCuFcN_eEj7Ko2bhJVcaihEIp7Evy50KnL6Sq-k,35602
 phoenix/session/data_extractor.py,sha256=Y0RzYFaNy9fQj8PEIeQ76TBZ90_E1FW7bXu3K5x0EZY,2782
@@ -437,9 +437,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-11.9.0.dist-info/METADATA,sha256=KNs35sXwK0WNwDhesRYOlxn7itnyOf3XwlKfj5b2G3o,30850
-arize_phoenix-11.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-11.9.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-11.9.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-11.9.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-11.9.0.dist-info/RECORD,,
+arize_phoenix-11.10.1.dist-info/METADATA,sha256=wd_5sRdhTNXTvCYeob4xrZbZiPxZ81ENs-yF8PMEIyE,30851
+arize_phoenix-11.10.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-11.10.1.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-11.10.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-11.10.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-11.10.1.dist-info/RECORD,,

phoenix/server/api/helpers/playground_clients.py

@@ -20,7 +20,7 @@ from openinference.semconv.trace import (
 )
 from strawberry import UNSET
 from strawberry.scalars import JSON as JSONScalarType
-from typing_extensions import TypeAlias, assert_never
+from typing_extensions import TypeAlias, assert_never, override
 
 from phoenix.config import getenv
 from phoenix.evals.models.rate_limiters import (
@@ -437,9 +437,9 @@ class OpenAIBaseStreamingClient(PlaygroundStreamingClient):
         if role is ChatCompletionMessageRole.TOOL:
             if tool_call_id is None:
                 raise ValueError("tool_call_id is required for tool messages")
-        return ChatCompletionToolMessageParam(
-            {"content": content, "role": "tool", "tool_call_id": tool_call_id}
-        )
+            return ChatCompletionToolMessageParam(
+                {"content": content, "role": "tool", "tool_call_id": tool_call_id}
+            )
         assert_never(role)
 
     def to_openai_tool_call_param(
@@ -1140,27 +1140,28 @@ class OpenAIStreamingClient(OpenAIBaseStreamingClient):
         self._attributes[LLM_SYSTEM] = OpenInferenceLLMSystemValues.OPENAI.value
 
 
-@register_llm_client(
-    provider_key=GenerativeProviderKey.OPENAI,
-    model_names=[
-        "o1",
-        "o1-pro",
-        "o1-2024-12-17",
-        "o1-pro-2025-03-19",
-        "o1-mini",
-        "o1-mini-2024-09-12",
-        "o1-preview",
-        "o1-preview-2024-09-12",
-        "o3",
-        "o3-pro",
-        "o3-2025-04-16",
-        "o3-mini",
-        "o3-mini-2025-01-31",
-        "o4-mini",
-        "o4-mini-2025-04-16",
-    ],
-)
-class OpenAIReasoningStreamingClient(OpenAIStreamingClient):
+_OPENAI_REASONING_MODELS = [
+    "o1",
+    "o1-pro",
+    "o1-2024-12-17",
+    "o1-pro-2025-03-19",
+    "o1-mini",
+    "o1-mini-2024-09-12",
+    "o1-preview",
+    "o1-preview-2024-09-12",
+    "o3",
+    "o3-pro",
+    "o3-2025-04-16",
+    "o3-mini",
+    "o3-mini-2025-01-31",
+    "o4-mini",
+    "o4-mini-2025-04-16",
+]
+
+
+class OpenAIReasoningReasoningModelsMixin:
+    """Mixin class for OpenAI-style reasoning model clients (o1, o3 series)."""
+
     @classmethod
     def supported_invocation_parameters(cls) -> list[InvocationParameter]:
         return [
@@ -1191,6 +1192,16 @@ class OpenAIReasoningStreamingClient(OpenAIStreamingClient):
             ),
         ]
 
+
+@register_llm_client(
+    provider_key=GenerativeProviderKey.OPENAI,
+    model_names=_OPENAI_REASONING_MODELS,
+)
+class OpenAIReasoningNonStreamingClient(
+    OpenAIReasoningReasoningModelsMixin,
+    OpenAIStreamingClient,
+):
+    @override
     async def chat_completion_create(
         self,
         messages: list[
@@ -1283,46 +1294,11 @@ class OpenAIReasoningStreamingClient(OpenAIStreamingClient):
         if role is ChatCompletionMessageRole.TOOL:
             if tool_call_id is None:
                 raise ValueError("tool_call_id is required for tool messages")
-        return ChatCompletionToolMessageParam(
-            {"content": content, "role": "tool", "tool_call_id": tool_call_id}
-        )
+            return ChatCompletionToolMessageParam(
+                {"content": content, "role": "tool", "tool_call_id": tool_call_id}
+            )
         assert_never(role)
 
-    @staticmethod
-    def _llm_token_counts(usage: "CompletionUsage") -> Iterator[tuple[str, Any]]:
-        yield LLM_TOKEN_COUNT_PROMPT, usage.prompt_tokens
-        yield LLM_TOKEN_COUNT_COMPLETION, usage.completion_tokens
-        yield LLM_TOKEN_COUNT_TOTAL, usage.total_tokens
-
-        if hasattr(usage, "prompt_tokens_details") and usage.prompt_tokens_details is not None:
-            prompt_details = usage.prompt_tokens_details
-            if (
-                hasattr(prompt_details, "cached_tokens")
-                and prompt_details.cached_tokens is not None
-            ):
-                yield LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ, prompt_details.cached_tokens
-            if hasattr(prompt_details, "audio_tokens") and prompt_details.audio_tokens is not None:
-                yield LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO, prompt_details.audio_tokens
-
-        if (
-            hasattr(usage, "completion_tokens_details")
-            and usage.completion_tokens_details is not None
-        ):
-            completion_details = usage.completion_tokens_details
-            if (
-                hasattr(completion_details, "reasoning_tokens")
-                and completion_details.reasoning_tokens is not None
-            ):
-                yield (
-                    LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING,
-                    completion_details.reasoning_tokens,
-                )
-            if (
-                hasattr(completion_details, "audio_tokens")
-                and completion_details.audio_tokens is not None
-            ):
-                yield LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO, completion_details.audio_tokens
-
 
 @register_llm_client(
     provider_key=GenerativeProviderKey.AZURE_OPENAI,
@@ -1376,6 +1352,113 @@ class AzureOpenAIStreamingClient(OpenAIBaseStreamingClient):
         self._attributes[LLM_SYSTEM] = OpenInferenceLLMSystemValues.OPENAI.value
 
 
+@register_llm_client(
+    provider_key=GenerativeProviderKey.AZURE_OPENAI,
+    model_names=_OPENAI_REASONING_MODELS,
+)
+class AzureOpenAIReasoningNonStreamingClient(
+    OpenAIReasoningReasoningModelsMixin,
+    AzureOpenAIStreamingClient,
+):
+    @override
+    async def chat_completion_create(
+        self,
+        messages: list[
+            tuple[ChatCompletionMessageRole, str, Optional[str], Optional[list[JSONScalarType]]]
+        ],
+        tools: list[JSONScalarType],
+        **invocation_parameters: Any,
+    ) -> AsyncIterator[ChatCompletionChunk]:
+        from openai import NOT_GIVEN
+
+        # Convert standard messages to OpenAI messages
+        openai_messages = []
+        for message in messages:
+            openai_message = self.to_openai_chat_completion_param(*message)
+            if openai_message is not None:
+                openai_messages.append(openai_message)
+
+        throttled_create = self.rate_limiter._alimit(self.client.chat.completions.create)
+        response = await throttled_create(
+            messages=openai_messages,
+            model=self.model_name,
+            stream=False,
+            tools=tools or NOT_GIVEN,
+            **invocation_parameters,
+        )
+
+        if response.usage is not None:
+            self._attributes.update(dict(self._llm_token_counts(response.usage)))
+
+        choice = response.choices[0]
+        if choice.message.content:
+            yield TextChunk(content=choice.message.content)
+
+        if choice.message.tool_calls:
+            for tool_call in choice.message.tool_calls:
+                yield ToolCallChunk(
+                    id=tool_call.id,
+                    function=FunctionCallChunk(
+                        name=tool_call.function.name,
+                        arguments=tool_call.function.arguments,
+                    ),
+                )
+
+    def to_openai_chat_completion_param(
+        self,
+        role: ChatCompletionMessageRole,
+        content: JSONScalarType,
+        tool_call_id: Optional[str] = None,
+        tool_calls: Optional[list[JSONScalarType]] = None,
+    ) -> Optional["ChatCompletionMessageParam"]:
+        from openai.types.chat import (
+            ChatCompletionAssistantMessageParam,
+            ChatCompletionDeveloperMessageParam,
+            ChatCompletionToolMessageParam,
+            ChatCompletionUserMessageParam,
+        )
+
+        if role is ChatCompletionMessageRole.USER:
+            return ChatCompletionUserMessageParam(
+                {
+                    "content": content,
+                    "role": "user",
+                }
+            )
+        if role is ChatCompletionMessageRole.SYSTEM:
+            return ChatCompletionDeveloperMessageParam(
+                {
+                    "content": content,
+                    "role": "developer",
+                }
+            )
+        if role is ChatCompletionMessageRole.AI:
+            if tool_calls is None:
+                return ChatCompletionAssistantMessageParam(
+                    {
+                        "content": content,
+                        "role": "assistant",
+                    }
+                )
+            else:
+                return ChatCompletionAssistantMessageParam(
+                    {
+                        "content": content,
+                        "role": "assistant",
+                        "tool_calls": [
+                            self.to_openai_tool_call_param(tool_call) for tool_call in tool_calls
+                        ],
+                    }
+                )
+        if role is ChatCompletionMessageRole.TOOL:
+            if tool_call_id is None:
+                raise ValueError("tool_call_id is required for tool messages")
+            return ChatCompletionToolMessageParam(
+                {"content": content, "role": "tool", "tool_call_id": tool_call_id}
+            )
+        assert_never(role)
+
+
 @register_llm_client(
     provider_key=GenerativeProviderKey.ANTHROPIC,
     model_names=[

phoenix/server/api/helpers/playground_registry.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Callable, Optional, Union
+from typing import TYPE_CHECKING, Any, Callable, Optional, Sequence, Union
 
 from phoenix.server.api.types.GenerativeProvider import GenerativeProviderKey
 
@@ -59,7 +59,7 @@ PLAYGROUND_CLIENT_REGISTRY: PlaygroundClientRegistry = PlaygroundClientRegistry(
 
 def register_llm_client(
     provider_key: GenerativeProviderKey,
-    model_names: list[ModelName],
+    model_names: Sequence[ModelName],
 ) -> Callable[[type["PlaygroundStreamingClient"]], type["PlaygroundStreamingClient"]]:
     def decorator(cls: type["PlaygroundStreamingClient"]) -> type["PlaygroundStreamingClient"]:
         provider_registry = PLAYGROUND_CLIENT_REGISTRY._registry.setdefault(provider_key, {})

phoenix/server/api/types/Project.py

@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Annotated, Any, ClassVar, Literal, Optional, c
 import strawberry
 from aioitertools.itertools import groupby, islice
 from openinference.semconv.trace import SpanAttributes
-from sqlalchemy import and_, desc, distinct, exists, func, or_, select
+from sqlalchemy import and_, case, desc, distinct, exists, func, or_, select
 from sqlalchemy.dialects import postgresql, sqlite
 from sqlalchemy.sql.elements import ColumnElement
 from sqlalchemy.sql.expression import tuple_
@@ -719,6 +719,7 @@ class Project(Node):
         info: Info[Context, None],
         time_range: TimeRange,
         time_bin_config: Optional[TimeBinConfig] = UNSET,
+        filter_condition: Optional[str] = UNSET,
     ) -> SpanCountTimeSeries:
         if time_range.start is None:
             raise BadRequest("Start time is required")
@@ -742,7 +743,17 @@ class Project(Node):
                 field = "year"
         bucket = date_trunc(dialect, field, models.Span.start_time, utc_offset_minutes)
         stmt = (
-            select(bucket, func.count(models.Span.id))
+            select(
+                bucket,
+                func.count(models.Span.id).label("total_count"),
+                func.sum(case((models.Span.status_code == "OK", 1), else_=0)).label("ok_count"),
+                func.sum(case((models.Span.status_code == "ERROR", 1), else_=0)).label(
+                    "error_count"
+                ),
+                func.sum(case((models.Span.status_code == "UNSET", 1), else_=0)).label(
+                    "unset_count"
+                ),
+            )
             .join_from(models.Span, models.Trace)
             .where(models.Trace.project_rowid == self.project_rowid)
             .group_by(bucket)
@@ -752,21 +763,31 @@ class Project(Node):
             stmt = stmt.where(time_range.start <= models.Span.start_time)
         if time_range.end:
             stmt = stmt.where(models.Span.start_time < time_range.end)
+        if filter_condition:
+            span_filter = SpanFilter(condition=filter_condition)
+            stmt = span_filter(stmt)
 
         data = {}
         async with info.context.db() as session:
-            async for t, v in await session.stream(stmt):
+            async for t, total_count, ok_count, error_count, unset_count in await session.stream(
+                stmt
+            ):
                 timestamp = _as_datetime(t)
-                data[timestamp] = TimeSeriesDataPoint(timestamp=timestamp, value=v)
+                data[timestamp] = SpanCountTimeSeriesDataPoint(
+                    timestamp=timestamp,
+                    ok_count=ok_count,
+                    error_count=error_count,
+                    unset_count=unset_count,
+                    total_count=total_count,
+                )
 
         data_timestamps: list[datetime] = [data_point.timestamp for data_point in data.values()]
         min_time = min([*data_timestamps, time_range.start])
         max_time = max(
             [
                 *data_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         for timestamp in get_timestamp_range(
             start_time=min_time,
@@ -775,7 +796,7 @@ class Project(Node):
             utc_offset_minutes=utc_offset_minutes,
         ):
             if timestamp not in data:
-                data[timestamp] = TimeSeriesDataPoint(timestamp=timestamp)
+                data[timestamp] = SpanCountTimeSeriesDataPoint(timestamp=timestamp)
         return SpanCountTimeSeries(data=sorted(data.values(), key=lambda x: x.timestamp))
 
     @strawberry.field
@@ -828,9 +849,8 @@ class Project(Node):
         max_time = max(
             [
                 *data_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         for timestamp in get_timestamp_range(
             start_time=min_time,
@@ -917,9 +937,8 @@ class Project(Node):
         max_time = max(
             [
                 *data_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         for timestamp in get_timestamp_range(
             start_time=min_time,
@@ -1025,9 +1044,8 @@ class Project(Node):
         max_time = max(
             [
                 *data_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         for timestamp in get_timestamp_range(
             start_time=min_time,
@@ -1111,9 +1129,8 @@ class Project(Node):
         max_time = max(
             [
                 *data_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         for timestamp in get_timestamp_range(
             start_time=min_time,
@@ -1195,9 +1212,8 @@ class Project(Node):
         max_time = max(
             [
                 *data_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         for timestamp in get_timestamp_range(
             start_time=min_time,
@@ -1281,9 +1297,8 @@ class Project(Node):
         max_time = max(
             [
                 *score_timestamps,
-                *([time_range.end] if time_range.end else []),
+                *([time_range.end] if time_range.end else [datetime.now(timezone.utc)]),
             ],
-            default=datetime.now(timezone.utc),
         )
         data: dict[datetime, SpanAnnotationScoreTimeSeriesDataPoint] = {
             timestamp: SpanAnnotationScoreTimeSeriesDataPoint(
@@ -1313,8 +1328,17 @@ class Project(Node):
 
 
 @strawberry.type
-class SpanCountTimeSeries(TimeSeries):
-    """A time series of span count"""
+class SpanCountTimeSeriesDataPoint:
+    timestamp: datetime
+    ok_count: Optional[int] = None
+    error_count: Optional[int] = None
+    unset_count: Optional[int] = None
+    total_count: Optional[int] = None
+
+
+@strawberry.type
+class SpanCountTimeSeries:
+    data: list[SpanCountTimeSeriesDataPoint]
 
 
 @strawberry.type

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

@@ -1,67 +1,67 @@
 {
-  "_components-IBd-PDxA.js": {
-    "file": "assets/components-IBd-PDxA.js",
+  "_components-XAeml0-1.js": {
+    "file": "assets/components-XAeml0-1.js",
     "name": "components",
     "imports": [
-      "_vendor-BzZ0oklU.js",
-      "_pages-6D1duYIe.js",
-      "_vendor-arizeai-CvjUqTrl.js",
-      "_vendor-codemirror-CKK25Gd7.js",
+      "_vendor-DhvamIr8.js",
+      "_pages-CPfaxiKa.js",
+      "_vendor-arizeai-4fVwwnrI.js",
+      "_vendor-codemirror-DRfFHb57.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-6D1duYIe.js": {
-    "file": "assets/pages-6D1duYIe.js",
+  "_pages-CPfaxiKa.js": {
+    "file": "assets/pages-CPfaxiKa.js",
     "name": "pages",
     "imports": [
-      "_vendor-BzZ0oklU.js",
-      "_vendor-arizeai-CvjUqTrl.js",
-      "_components-IBd-PDxA.js",
-      "_vendor-codemirror-CKK25Gd7.js",
-      "_vendor-recharts-CWtaRhQC.js"
+      "_vendor-DhvamIr8.js",
+      "_vendor-arizeai-4fVwwnrI.js",
+      "_components-XAeml0-1.js",
+      "_vendor-codemirror-DRfFHb57.js",
+      "_vendor-recharts-w6bSawXG.js"
     ]
   },
-  "_vendor-BzZ0oklU.js": {
-    "file": "assets/vendor-BzZ0oklU.js",
+  "_vendor-CqDb5u4o.css": {
+    "file": "assets/vendor-CqDb5u4o.css",
+    "src": "_vendor-CqDb5u4o.css"
+  },
+  "_vendor-DhvamIr8.js": {
+    "file": "assets/vendor-DhvamIr8.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-C5WAXd5r.js"
     ],
     "css": [
-      "assets/vendor-WIZid84E.css"
+      "assets/vendor-CqDb5u4o.css"
     ]
   },
-  "_vendor-WIZid84E.css": {
-    "file": "assets/vendor-WIZid84E.css",
-    "src": "_vendor-WIZid84E.css"
-  },
-  "_vendor-arizeai-CvjUqTrl.js": {
-    "file": "assets/vendor-arizeai-CvjUqTrl.js",
+  "_vendor-arizeai-4fVwwnrI.js": {
+    "file": "assets/vendor-arizeai-4fVwwnrI.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-BzZ0oklU.js"
+      "_vendor-DhvamIr8.js"
     ]
   },
-  "_vendor-codemirror-CKK25Gd7.js": {
-    "file": "assets/vendor-codemirror-CKK25Gd7.js",
+  "_vendor-codemirror-DRfFHb57.js": {
+    "file": "assets/vendor-codemirror-DRfFHb57.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-BzZ0oklU.js",
-      "_vendor-shiki-D30GF-p9.js"
+      "_vendor-DhvamIr8.js",
+      "_vendor-shiki-CplrhwOk.js"
     ]
   },
-  "_vendor-recharts-CWtaRhQC.js": {
-    "file": "assets/vendor-recharts-CWtaRhQC.js",
+  "_vendor-recharts-w6bSawXG.js": {
+    "file": "assets/vendor-recharts-w6bSawXG.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-BzZ0oklU.js"
+      "_vendor-DhvamIr8.js"
     ]
   },
-  "_vendor-shiki-D30GF-p9.js": {
-    "file": "assets/vendor-shiki-D30GF-p9.js",
+  "_vendor-shiki-CplrhwOk.js": {
+    "file": "assets/vendor-shiki-CplrhwOk.js",
     "name": "vendor-shiki",
     "imports": [
-      "_vendor-BzZ0oklU.js"
+      "_vendor-DhvamIr8.js"
     ]
   },
   "_vendor-three-C5WAXd5r.js": {
@@ -69,19 +69,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-B8EBC_Z5.js",
+    "file": "assets/index-D7EtHUpz.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-BzZ0oklU.js",
-      "_vendor-arizeai-CvjUqTrl.js",
-      "_pages-6D1duYIe.js",
-      "_components-IBd-PDxA.js",
+      "_vendor-DhvamIr8.js",
+      "_vendor-arizeai-4fVwwnrI.js",
+      "_pages-CPfaxiKa.js",
+      "_components-XAeml0-1.js",
       "_vendor-three-C5WAXd5r.js",
-      "_vendor-codemirror-CKK25Gd7.js",
-      "_vendor-shiki-D30GF-p9.js",
-      "_vendor-recharts-CWtaRhQC.js"
+      "_vendor-codemirror-DRfFHb57.js",
+      "_vendor-shiki-CplrhwOk.js",
+      "_vendor-recharts-w6bSawXG.js"
     ]
   }
 }