arize-phoenix 11.35.0__py3-none-any.whl → 11.37.0__py3-none-any.whl

{arize_phoenix-11.35.0.dist-info → arize_phoenix-11.37.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 11.35.0
+Version: 11.37.0
 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.35.0.dist-info → arize_phoenix-11.37.0.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=RJFnwmqSmKmJQDpkbU6omf53SgvFxHMmwQOK75tvJYM,24
+phoenix/version.py,sha256=51ny777jKy9SIcIn2Vr7pZhCjHVXPehMhak2EG9U07g,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
@@ -183,7 +183,7 @@ 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=fG97GnJPxOYMd0lNHdQfML_ZmSZfLnY2s7_GAiAecOc,72327
+phoenix/server/api/helpers/playground_clients.py,sha256=7PuGc5s9uKZjM6uGO85yc0F0HWke3Yn1F6N3dBaukOY,73757
 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
@@ -215,7 +215,7 @@ phoenix/server/api/input_types/DeleteExperimentsInput.py,sha256=4d9N0vSLYbuysAam
 phoenix/server/api/input_types/DimensionFilter.py,sha256=eBYcn7ECSJQlEePvbStqkHBRicbIL4vEAzFJwX7bacQ,3137
 phoenix/server/api/input_types/DimensionInput.py,sha256=Vfx5FmiMKey4-EHDQsQRPzSAMRJMN5oVMLDUl4NKAa8,164
 phoenix/server/api/input_types/GenerativeCredentialInput.py,sha256=sEM9UtgDMMuhImLwGXctD8BWDs2V3hNQ1mosoklRZvc,219
-phoenix/server/api/input_types/GenerativeModelInput.py,sha256=ceJ65f7Mf1JmOuH9TV3KJYbNnQ-_Z85Dl4_vPMB1K_o,713
+phoenix/server/api/input_types/GenerativeModelInput.py,sha256=M9R5LmLjyxOYPEIiann4ln23degOWJ2GTIk64bi73yI,841
 phoenix/server/api/input_types/Granularity.py,sha256=dbBlD_GsIBa8_xrx4JlLuR59bQ0NRB5H-cv1zvcb-cw,2299
 phoenix/server/api/input_types/InvocationParameters.py,sha256=62xL0iIKvuQherkuJaJ6Lha4TTEoYLpvH-pEP9awK6k,5260
 phoenix/server/api/input_types/PatchAnnotationInput.py,sha256=2wxC-ibQU59I28amTxLMYMo9SSvJwAIor0w0YvS0e48,676
@@ -230,7 +230,7 @@ phoenix/server/api/input_types/PromptTemplateOptions.py,sha256=8ZJdH1F9fExcdH9dF
 phoenix/server/api/input_types/PromptVersionInput.py,sha256=6iFWf2Ye9K1dwL1810L-wUTLkMtOIHemJTxar4xF4kw,3911
 phoenix/server/api/input_types/SpanAnnotationFilter.py,sha256=-djfIXYCxV6sV3GPOZQUV0SPfiWDhRlTORfeQ7tCBgQ,2671
 phoenix/server/api/input_types/SpanAnnotationSort.py,sha256=T5pAGzmh4MiJp9JMAzNDByFVTczfw02FH4WFWwFezyI,361
-phoenix/server/api/input_types/SpanSort.py,sha256=GReQx9yOo0Kehi2y4AtY69aZhRtcqvcg-9bSIFru69U,7540
+phoenix/server/api/input_types/SpanSort.py,sha256=1AHrqx3HuVKsp9DSaDsQMAyid4Au3kqY38mE9x0nPBU,7552
 phoenix/server/api/input_types/TimeBinConfig.py,sha256=s4p0SNTGRY6rbHfZhEMwmon5zX85j-DbN06GkapF6jg,516
 phoenix/server/api/input_types/TimeRange.py,sha256=pwhC2jx6dFIgT0qFfnO68qiJp9-m7Je5QkNscNsuVxA,700
 phoenix/server/api/input_types/TraceAnnotationSort.py,sha256=BzwiUnMh2VsgQYnhDlbJ6ljHugqIS4YDUlYzvq_tl3o,365
@@ -311,7 +311,7 @@ phoenix/server/api/types/EvaluationSummary.py,sha256=vILYejnfPvMwWEXOwhQZsANvYe3
 phoenix/server/api/types/Event.py,sha256=iYt_Jx1Roioo0vZ0iPeJTHcTu6NSm4ilVMJ-IMUHAKk,3970
 phoenix/server/api/types/EventMetadata.py,sha256=-J0tYF9eZTHwCjwxQHY7Gckr2_MNW5OoWT1mydweZNM,635
 phoenix/server/api/types/ExampleRevisionInterface.py,sha256=gV3Gt9-3Oi5wjaVtepC6nOt3FzTzZFD1KebNnqiw56E,294
-phoenix/server/api/types/Experiment.py,sha256=URdD3Ow0gTlJRd74duLdGFs-E3gKy2-UAah1AIaFKGk,8230
+phoenix/server/api/types/Experiment.py,sha256=GdTAS-p5d1GyekwGItQZClHS1Hz1u9IfTcnAeZUaCwU,8295
 phoenix/server/api/types/ExperimentAnnotationSummary.py,sha256=Uk3JtxIrsMoZT5tqc4nJdUOM3XegVzjUyoV3pkjNotE,256
 phoenix/server/api/types/ExperimentComparison.py,sha256=fj4KoAPFNJvfrHBMloGkHz_-7Lf403IMe0OwDDJyZWk,383
 phoenix/server/api/types/ExperimentRepeatedRunGroup.py,sha256=a2-C6ZWyFm-THvus6n4WDYGtmapm-ZWFaDahW7A2i28,5434
@@ -332,7 +332,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=XJUVEGCmEgToIyoVNWpNgcgtV1g24DvAKzFHpDNpCJY,72020
+phoenix/server/api/types/Project.py,sha256=klaowPuWXVz6tFQEA10gL1pY554H-nPFNtLcUSGGMxk,72289
 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=ad52KVQ1yJrAWXd3HdOKKHzyaUJqrp2mtFEe1QGEE4E,5455
@@ -368,7 +368,7 @@ phoenix/server/api/types/ValidationResult.py,sha256=pHwdYk4J7SJ5xhlWWHg_6qWkfk4r
 phoenix/server/api/types/VectorDriftMetricEnum.py,sha256=etiJM5ZjQuD-oE7sY-FbdIKY050jk3IU49IMwmfJbEc,188
 phoenix/server/api/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/types/node.py,sha256=ya1kJz6ok0LEiMMJOvV6s6nr_biWBZWAiC9JEtR-ki8,1073
-phoenix/server/api/types/pagination.py,sha256=BXm46gXZfrBS4hpiLvVSEdsbb29ctUMVJYjKXlOLxUA,9064
+phoenix/server/api/types/pagination.py,sha256=GJktCIGRfbuFh8kHJzJ0QTzs6ZDtAJizgjSSObnBCeM,9342
 phoenix/server/cost_tracking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/cost_tracking/cost_details_calculator.py,sha256=Tt0YcuLhgPuXKWJemWVmYQfG0xQUvH4VziIj6KcDnoA,8945
 phoenix/server/cost_tracking/cost_model_lookup.py,sha256=jhtVdnQBzrTUHeOGPWgOebk-Io5hpJ1vAgWOu8ojeJ4,6801
@@ -402,16 +402,16 @@ 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=UKiCtA5rvKMr8U4zLFRXo5S5Fdvy8dKouD1kF3GESko,2328
-phoenix/server/static/assets/components-CdQiQTvs.js,sha256=ffg4jmfz-Bc9PTLXXAz7OduRazQDl5KZyeR_3S2aILs,675981
-phoenix/server/static/assets/index-B1VuXYRI.js,sha256=jOk_qo0Q0SRXWfvHVlValebC3vs1mUfV1v-Kf94B2rc,62934
-phoenix/server/static/assets/pages-CnfZ3RhB.js,sha256=XYLansUm9qmewwVwsW-dskFdfP72RCMzRAdzlLMJMhg,1298004
+phoenix/server/static/.vite/manifest.json,sha256=eL7fSMB_ukWrI1_haVcKcon1yQTo89vwyOJmSZar_qU,2328
+phoenix/server/static/assets/components-CFzdBkk_.js,sha256=TPy0Mz14mPKzLVdxczNBA9dAvPE9TkdOZBMu8xkJqMY,683709
+phoenix/server/static/assets/index-DayUA9lQ.js,sha256=4QQWs2ubciFLQwVnRPTRHZkzTCW_HGvtNjJfVcbi-pI,63435
+phoenix/server/static/assets/pages-CvUhOO9h.js,sha256=x30rhunnQtnuzZnX-HArUNQ-Uii4sdsL8tFFmoWEjsg,1303234
 phoenix/server/static/assets/vendor-BGzfc4EU.css,sha256=Nx5Lmx-bqYR7nsO_O4kEBcrJ8cwknWjZ6seHN3_s4UQ,3171
-phoenix/server/static/assets/vendor-Cfrr9FCF.js,sha256=e5sXxTzHgKwjsNy7bFWp19fM9QolSJBQ9FvE4hmxrPs,2575393
-phoenix/server/static/assets/vendor-arizeai-Dz0kN-lQ.js,sha256=hPa1rpMUjztQaAsf5h87E_RCDhcuWowkoTrfMaTS5_g,121519
-phoenix/server/static/assets/vendor-codemirror-ClqtONZQ.js,sha256=zPD9V1TJW7mlnmIQcpjjaajHLATsDb0dxXnGsq24Qcg,413205
-phoenix/server/static/assets/vendor-recharts-D6kvOpmb.js,sha256=AcaZOFH-ggGi-znCeAP_b072ALbIcPqhqYDZn63Hjps,231653
-phoenix/server/static/assets/vendor-shiki-xSOiKxt0.js,sha256=itTnSRR18G_YLCygdVlLLqMrXHaPqjnD2Z6WcS4O9ro,305160
+phoenix/server/static/assets/vendor-BdjZxMii.js,sha256=CEEWyhdR8PcZfr9EntiJVBEN3hoFmS3hnvxQapcB57M,2591438
+phoenix/server/static/assets/vendor-arizeai-CHYlS8jV.js,sha256=AEKxh9WhnE-QrWus26WBBR0Fq1fh2r1iXPl3BFU7quk,112652
+phoenix/server/static/assets/vendor-codemirror-Di6t4HnH.js,sha256=4grodwIcHPZTnaKyCM8LqhYr7xQh6sQfICc-eQ_5l2c,413213
+phoenix/server/static/assets/vendor-recharts-C9wCDYj3.js,sha256=_vwHqxWoiaeY2yKMKGQxyb62QorGUX0Y9kv-jE0JAMA,231654
+phoenix/server/static/assets/vendor-shiki-MNnmOotP.js,sha256=gXLCqndUHWJUssTPN2s4nOnxBMlnGyWb2XmIwogQfLs,305160
 phoenix/server/static/assets/vendor-three-BLWp5bic.js,sha256=vfSCVXS20jA0Ceo_O0mDxYBcROinWMdPE6RR4JXmtec,620972
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=QAYh0TG5mg-GvDQUR09aD9ebl9Sfq0fYAcfIa5G7J6E,7148
@@ -451,9 +451,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.35.0.dist-info/METADATA,sha256=xS7hzRKWSqDhACjwfYoslteLqm7z6LzDigKI74ei-rg,34045
-arize_phoenix-11.35.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-11.35.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-11.35.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-11.35.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-11.35.0.dist-info/RECORD,,
+arize_phoenix-11.37.0.dist-info/METADATA,sha256=1taqXAi5gQonzOQVvGR3ibHNHhjlVTXK-hptSPCpL8M,34045
+arize_phoenix-11.37.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-11.37.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-11.37.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-11.37.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-11.37.0.dist-info/RECORD,,

phoenix/server/api/helpers/playground_clients.py

@@ -57,6 +57,7 @@ from phoenix.server.api.types.GenerativeProvider import GenerativeProviderKey
 if TYPE_CHECKING:
     import httpx
     from anthropic.types import MessageParam, TextBlockParam, ToolResultBlockParam
+    from botocore.awsrequest import AWSPreparedRequest  # type: ignore[import-untyped]
     from google.generativeai.types import ContentType
     from openai import AsyncAzureOpenAI, AsyncOpenAI
     from openai.types import CompletionUsage
@@ -546,6 +547,7 @@ class DeepSeekStreamingClient(OpenAIBaseStreamingClient):
         client = AsyncOpenAI(
             api_key=api_key,
             base_url=base_url or "https://api.deepseek.com",
+            default_headers=model.custom_headers or None,
         )
         super().__init__(client=client, model=model, credentials=credentials)
         # DeepSeek uses OpenAI-compatible API but we'll track it as a separate provider
@@ -587,6 +589,7 @@ class XAIStreamingClient(OpenAIBaseStreamingClient):
         client = AsyncOpenAI(
             api_key=api_key,
             base_url=base_url or "https://api.x.ai/v1",
+            default_headers=model.custom_headers or None,
         )
         super().__init__(client=client, model=model, credentials=credentials)
         # xAI uses OpenAI-compatible API but we'll track it as a separate provider
@@ -624,7 +627,11 @@ class OllamaStreamingClient(OpenAIBaseStreamingClient):
         if not base_url:
             raise BadRequest("An Ollama base URL is required for Ollama models")
         api_key = "ollama"
-        client = AsyncOpenAI(api_key=api_key, base_url=base_url)
+        client = AsyncOpenAI(
+            api_key=api_key,
+            base_url=base_url,
+            default_headers=model.custom_headers or None,
+        )
         super().__init__(client=client, model=model, credentials=credentials)
         # Ollama uses OpenAI-compatible API but we'll track it as a separate provider
         # Adding a custom "ollama" provider value to make it distinguishable in traces
@@ -679,6 +686,7 @@ class BedrockStreamingClient(PlaygroundStreamingClient):
         super().__init__(model=model, credentials=credentials)
         self.region = model.region or "us-east-1"
         self.api = "converse"
+        self.custom_headers = model.custom_headers or {}
         self.aws_access_key_id = _get_credential_value(credentials, "AWS_ACCESS_KEY_ID") or getenv(
             "AWS_ACCESS_KEY_ID"
         )
@@ -697,9 +705,24 @@ class BedrockStreamingClient(PlaygroundStreamingClient):
             aws_session_token=self.aws_session_token,
         )
 
+        # Add custom headers support via boto3 event system
+        if self.custom_headers:
+            self._setup_custom_headers(self.client, self.custom_headers)
+
         self._attributes[LLM_PROVIDER] = "aws"
         self._attributes[LLM_SYSTEM] = "aws"
 
+    @staticmethod
+    def _setup_custom_headers(client: Any, custom_headers: Mapping[str, str]) -> None:
+        """Setup custom headers using boto3's event system."""
+        if not custom_headers:
+            return
+
+        def add_custom_headers(request: "AWSPreparedRequest", **kwargs: Any) -> None:
+            request.headers.update(custom_headers)
+
+        client.meta.events.register("before-send.*", add_custom_headers)
+
     @classmethod
     def dependencies(cls) -> list[Dependency]:
         return [Dependency(name="boto3")]
@@ -756,6 +779,9 @@ class BedrockStreamingClient(PlaygroundStreamingClient):
                 aws_secret_access_key=self.aws_secret_access_key,
                 aws_session_token=self.aws_session_token,
             )
+            # Re-setup custom headers for the new client
+            if self.custom_headers:
+                self._setup_custom_headers(self.client, self.custom_headers)
         if self.api == "invoke":
             async for chunk in self._handle_invoke_api(messages, tools, invocation_parameters):
                 yield chunk
@@ -1140,7 +1166,12 @@ class OpenAIStreamingClient(OpenAIBaseStreamingClient):
                 raise BadRequest("An API key is required for OpenAI models")
             api_key = "sk-fake-api-key"
 
-        client = AsyncOpenAI(api_key=api_key, base_url=base_url, timeout=30)
+        client = AsyncOpenAI(
+            api_key=api_key,
+            base_url=base_url,
+            default_headers=model.custom_headers or None,
+            timeout=30,
+        )
         super().__init__(client=client, model=model, credentials=credentials)
         self._attributes[LLM_PROVIDER] = OpenInferenceLLMProviderValues.OPENAI.value
         self._attributes[LLM_SYSTEM] = OpenInferenceLLMSystemValues.OPENAI.value
@@ -1295,6 +1326,7 @@ class AzureOpenAIStreamingClient(OpenAIBaseStreamingClient):
                 api_key=api_key,
                 azure_endpoint=endpoint,
                 api_version=api_version,
+                default_headers=model.custom_headers or None,
             )
         else:
             try:
@@ -1312,6 +1344,7 @@ class AzureOpenAIStreamingClient(OpenAIBaseStreamingClient):
                 ),
                 azure_endpoint=endpoint,
                 api_version=api_version,
+                default_headers=model.custom_headers or None,
             )
         super().__init__(client=client, model=model, credentials=credentials)
         self._attributes[LLM_PROVIDER] = OpenInferenceLLMProviderValues.AZURE.value
@@ -1459,7 +1492,10 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
         if not api_key:
             raise BadRequest("An API key is required for Anthropic models")
 
-        self.client = anthropic.AsyncAnthropic(api_key=api_key)
+        self.client = anthropic.AsyncAnthropic(
+            api_key=api_key,
+            default_headers=model.custom_headers or None,
+        )
         self.model_name = model.name
         self.rate_limiter = PlaygroundRateLimiter(model.provider_key, anthropic.RateLimitError)
         self.client._client = _HttpxClient(self.client._client, self._attributes)

phoenix/server/api/input_types/GenerativeModelInput.py

@@ -2,6 +2,7 @@ from typing import Optional
 
 import strawberry
 from strawberry import UNSET
+from strawberry.scalars import JSON
 
 from phoenix.server.api.types.GenerativeProvider import GenerativeProviderKey
 
@@ -19,3 +20,5 @@ class GenerativeModelInput:
     """ The API version to use for the model. """
     region: Optional[str] = UNSET
     """ The region to use for the model. """
+    custom_headers: Optional[JSON] = UNSET
+    """ Custom headers to use for the model. """

phoenix/server/api/input_types/SpanSort.py

@@ -199,7 +199,7 @@ class SpanSort:
                     models.SpanAnnotation.span_rowid == models.Span.id,
                     models.SpanAnnotation.name == eval_name,
                 ),
-            ).order_by(expr)
+            ).order_by(nulls_last(expr))
             return SpanSortConfig(
                 stmt=stmt,
                 orm_expression=eval_result_key.attr.orm_expression,

phoenix/server/api/types/Experiment.py

@@ -33,6 +33,7 @@ class Experiment(Node):
     name: str
     project_name: Optional[str]
     description: Optional[str]
+    repetitions: int
     dataset_version_id: GlobalID
     metadata: JSON
     created_at: datetime
@@ -210,6 +211,7 @@ def to_gql_experiment(
         name=experiment.name,
         project_name=experiment.project_name,
         description=experiment.description,
+        repetitions=experiment.repetitions,
         dataset_version_id=GlobalID(DatasetVersion.__name__, str(experiment.dataset_version_id)),
         metadata=experiment.metadata_,
         created_at=experiment.created_at,

phoenix/server/api/types/Project.py

@@ -355,12 +355,16 @@ class Project(Node):
             if sort_config and cursor.sort_column:
                 sort_column = cursor.sort_column
                 compare = operator.lt if sort_config.dir is SortDir.desc else operator.gt
-                stmt = stmt.where(
-                    compare(
-                        tuple_(sort_config.orm_expression, models.Span.id),
-                        (sort_column.value, cursor.rowid),
+                if sort_column.type is CursorSortColumnDataType.NULL:
+                    stmt = stmt.where(sort_config.orm_expression.is_(None))
+                    stmt = stmt.where(compare(models.Span.id, cursor.rowid))
+                else:
+                    stmt = stmt.where(
+                        compare(
+                            tuple_(sort_config.orm_expression, models.Span.id),
+                            (sort_column.value, cursor.rowid),
+                        )
                     )
-                )
             else:
                 stmt = stmt.where(models.Span.id > cursor.rowid)
         stmt = stmt.order_by(cursor_rowid_column)

phoenix/server/api/types/pagination.py

@@ -9,7 +9,7 @@ from strawberry.relay.types import Connection, Edge, NodeType, PageInfo
 from typing_extensions import TypeAlias, assert_never
 
 ID: TypeAlias = int
-CursorSortColumnValue: TypeAlias = Union[str, int, float, datetime]
+CursorSortColumnValue: TypeAlias = Union[str, int, float, datetime, None]
 
 # A type alias for the connection cursor implementation
 CursorString = str
@@ -19,6 +19,7 @@ CURSOR_PREFIX = "connection:"
 
 
 class CursorSortColumnDataType(Enum):
+    NULL = auto()
     STRING = auto()
     INT = auto()
     FLOAT = auto()
@@ -30,6 +31,10 @@ class CursorSortColumn:
     type: CursorSortColumnDataType
     value: CursorSortColumnValue
 
+    def __post_init__(self) -> None:
+        if self.value is None:
+            self.type = CursorSortColumnDataType.NULL
+
     def __str__(self) -> str:
         if isinstance(self.value, str):
             return self.value
@@ -37,7 +42,9 @@ class CursorSortColumn:
             return str(self.value)
         if isinstance(self.value, datetime):
             return self.value.isoformat()
-        assert_never(self.type)
+        if self.value is None:
+            return ""
+        assert_never(self.value)
 
     @classmethod
     def from_string(cls, type: CursorSortColumnDataType, cursor_string: str) -> "CursorSortColumn":
@@ -50,6 +57,8 @@ class CursorSortColumn:
             value = float(cursor_string)
         elif type is CursorSortColumnDataType.DATETIME:
             value = datetime.fromisoformat(cursor_string)
+        elif type is CursorSortColumnDataType.NULL:
+            value = None
         else:
             assert_never(type)
         return cls(type=type, value=value)

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

@@ -1,32 +1,32 @@
 {
-  "_components-CdQiQTvs.js": {
-    "file": "assets/components-CdQiQTvs.js",
+  "_components-CFzdBkk_.js": {
+    "file": "assets/components-CFzdBkk_.js",
     "name": "components",
     "imports": [
-      "_vendor-Cfrr9FCF.js",
-      "_pages-CnfZ3RhB.js",
-      "_vendor-arizeai-Dz0kN-lQ.js",
-      "_vendor-codemirror-ClqtONZQ.js",
+      "_vendor-BdjZxMii.js",
+      "_vendor-arizeai-CHYlS8jV.js",
+      "_pages-CvUhOO9h.js",
+      "_vendor-codemirror-Di6t4HnH.js",
       "_vendor-three-BLWp5bic.js"
     ]
   },
-  "_pages-CnfZ3RhB.js": {
-    "file": "assets/pages-CnfZ3RhB.js",
+  "_pages-CvUhOO9h.js": {
+    "file": "assets/pages-CvUhOO9h.js",
     "name": "pages",
     "imports": [
-      "_vendor-Cfrr9FCF.js",
-      "_vendor-arizeai-Dz0kN-lQ.js",
-      "_components-CdQiQTvs.js",
-      "_vendor-codemirror-ClqtONZQ.js",
-      "_vendor-recharts-D6kvOpmb.js"
+      "_vendor-BdjZxMii.js",
+      "_vendor-arizeai-CHYlS8jV.js",
+      "_components-CFzdBkk_.js",
+      "_vendor-codemirror-Di6t4HnH.js",
+      "_vendor-recharts-C9wCDYj3.js"
     ]
   },
   "_vendor-BGzfc4EU.css": {
     "file": "assets/vendor-BGzfc4EU.css",
     "src": "_vendor-BGzfc4EU.css"
   },
-  "_vendor-Cfrr9FCF.js": {
-    "file": "assets/vendor-Cfrr9FCF.js",
+  "_vendor-BdjZxMii.js": {
+    "file": "assets/vendor-BdjZxMii.js",
     "name": "vendor",
     "imports": [
       "_vendor-three-BLWp5bic.js"
@@ -35,39 +35,39 @@
       "assets/vendor-BGzfc4EU.css"
     ]
   },
-  "_vendor-arizeai-Dz0kN-lQ.js": {
-    "file": "assets/vendor-arizeai-Dz0kN-lQ.js",
+  "_vendor-arizeai-CHYlS8jV.js": {
+    "file": "assets/vendor-arizeai-CHYlS8jV.js",
     "name": "vendor-arizeai",
     "imports": [
-      "_vendor-Cfrr9FCF.js"
+      "_vendor-BdjZxMii.js"
     ]
   },
-  "_vendor-codemirror-ClqtONZQ.js": {
-    "file": "assets/vendor-codemirror-ClqtONZQ.js",
+  "_vendor-codemirror-Di6t4HnH.js": {
+    "file": "assets/vendor-codemirror-Di6t4HnH.js",
     "name": "vendor-codemirror",
     "imports": [
-      "_vendor-Cfrr9FCF.js",
-      "_vendor-shiki-xSOiKxt0.js"
+      "_vendor-BdjZxMii.js",
+      "_vendor-shiki-MNnmOotP.js"
     ],
     "dynamicImports": [
-      "_vendor-shiki-xSOiKxt0.js",
-      "_vendor-shiki-xSOiKxt0.js",
-      "_vendor-shiki-xSOiKxt0.js"
+      "_vendor-shiki-MNnmOotP.js",
+      "_vendor-shiki-MNnmOotP.js",
+      "_vendor-shiki-MNnmOotP.js"
     ]
   },
-  "_vendor-recharts-D6kvOpmb.js": {
-    "file": "assets/vendor-recharts-D6kvOpmb.js",
+  "_vendor-recharts-C9wCDYj3.js": {
+    "file": "assets/vendor-recharts-C9wCDYj3.js",
     "name": "vendor-recharts",
     "imports": [
-      "_vendor-Cfrr9FCF.js"
+      "_vendor-BdjZxMii.js"
     ]
   },
-  "_vendor-shiki-xSOiKxt0.js": {
-    "file": "assets/vendor-shiki-xSOiKxt0.js",
+  "_vendor-shiki-MNnmOotP.js": {
+    "file": "assets/vendor-shiki-MNnmOotP.js",
     "name": "vendor-shiki",
     "isDynamicEntry": true,
     "imports": [
-      "_vendor-Cfrr9FCF.js"
+      "_vendor-BdjZxMii.js"
     ]
   },
   "_vendor-three-BLWp5bic.js": {
@@ -75,19 +75,19 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-B1VuXYRI.js",
+    "file": "assets/index-DayUA9lQ.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
-      "_vendor-Cfrr9FCF.js",
-      "_vendor-arizeai-Dz0kN-lQ.js",
-      "_pages-CnfZ3RhB.js",
-      "_components-CdQiQTvs.js",
+      "_vendor-BdjZxMii.js",
+      "_vendor-arizeai-CHYlS8jV.js",
+      "_pages-CvUhOO9h.js",
+      "_components-CFzdBkk_.js",
       "_vendor-three-BLWp5bic.js",
-      "_vendor-codemirror-ClqtONZQ.js",
-      "_vendor-shiki-xSOiKxt0.js",
-      "_vendor-recharts-D6kvOpmb.js"
+      "_vendor-codemirror-Di6t4HnH.js",
+      "_vendor-shiki-MNnmOotP.js",
+      "_vendor-recharts-C9wCDYj3.js"
     ]
   }
 }