arize-phoenix 7.10.3__py3-none-any.whl → 7.12.0__py3-none-any.whl

{arize_phoenix-7.10.3.dist-info → arize_phoenix-7.12.0.dist-info}/METADATA

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

{arize_phoenix-7.10.3.dist-info → arize_phoenix-7.12.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=kpW1WL0kiB8XJsO6XycvZVJ-lBkNoenhQ7atCvBoSe8,5365
 phoenix/settings.py,sha256=ht-0oN-sMV6SPXrk7Tu1EZlngpAYkGNLYPhO8DyrdQI,661
-phoenix/version.py,sha256=kBzXRB3iFFczA8Xv7GfR7zf-a7LC9V3kyUu5l-giCpE,23
+phoenix/version.py,sha256=epZfTrsMWfpY2no1bS7o-MeQ23JncNk7yI5RH-5Pqlg,23
 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
@@ -135,7 +135,7 @@ phoenix/server/api/dataloaders/cache/two_tier_cache.py,sha256=cmo8FUT3E91R139IEz
 phoenix/server/api/helpers/__init__.py,sha256=m2-xaSPqUiSs91k62JaRDjFNfl-1byxBfY-m_Vxw16U,272
 phoenix/server/api/helpers/dataset_helpers.py,sha256=DoMBTg-qXTnC_K4Evx1WKpCCYgRbITpVqyY-8efJRf0,8984
 phoenix/server/api/helpers/experiment_run_filters.py,sha256=DOnVwrmn39eAkk2mwuZP8kIcAnR5jrOgllEwWSjsw94,29893
-phoenix/server/api/helpers/playground_clients.py,sha256=AUZCc3lgHFi_hoipgK0tBr9eFA5ZEhLpNm7ZRoz7HeU,39035
+phoenix/server/api/helpers/playground_clients.py,sha256=U9EwldYQtQrubT0cJDzUAo4Y7n18_xoU9uIKPYHiBHs,39890
 phoenix/server/api/helpers/playground_registry.py,sha256=CPLMziFB2wmr-dfbx7VbzO2f8YIG_k5RftzvGXYGQ1w,2570
 phoenix/server/api/helpers/playground_spans.py,sha256=qGk7V7IZK7EkRE1mvZyROpLN5kgOahOZifFzUWmqYFc,16546
 phoenix/server/api/input_types/AddExamplesToDatasetInput.py,sha256=mIQz0S_z8YdrktKIY6RCvtNJ2yZF9pYvTGgasUsI-54,430
@@ -158,7 +158,7 @@ phoenix/server/api/input_types/DeleteDatasetInput.py,sha256=p7xjCyWnVCIXHnezmDiW
 phoenix/server/api/input_types/DeleteExperimentsInput.py,sha256=4d9N0vSLYbuysAamGoPUP_m8vdVhwrZmXoi2vhy_HdI,141
 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/GenerativeModelInput.py,sha256=8eRKLMVwXbfc6QVQySN43BDA1JEY09a2x0Egy4K0jAQ,815
+phoenix/server/api/input_types/GenerativeModelInput.py,sha256=n6OCkX44I1AIovMAHCxy8SvqPKDb_BYDPA-fn_JnckQ,634
 phoenix/server/api/input_types/Granularity.py,sha256=dbBlD_GsIBa8_xrx4JlLuR59bQ0NRB5H-cv1zvcb-cw,2299
 phoenix/server/api/input_types/InvocationParameters.py,sha256=I_FvPvY2sXxNd-kFSB96-1DOW7lo-VbqVWzppU_5hy8,5196
 phoenix/server/api/input_types/PatchAnnotationInput.py,sha256=NWhkcbcGNPwfOYsN3wm5YFNNrSc5T-8Y5my74RK99HE,520
@@ -238,7 +238,7 @@ phoenix/server/api/types/ExperimentRunAnnotation.py,sha256=HEbPygBwbWg10fwY3VZyM
 phoenix/server/api/types/ExportedFile.py,sha256=e3GTn7B5LgsTbqiwjhMCQH7VsiqXitrBO4aCMS1lHsg,163
 phoenix/server/api/types/Functionality.py,sha256=tzV9xdhB8zqfsjWxP66NDC7EZsplYkYO7jRbLWJIeeg,382
 phoenix/server/api/types/GenerativeModel.py,sha256=P7eBUMXbeqaLwSSGBKdZy3a5gOLd9I0fuP8o1st6H08,193
-phoenix/server/api/types/GenerativeProvider.py,sha256=3zOR7SssR3mkGAaj8j-_qVzN6H7Qll52J_i98jVyEpA,3127
+phoenix/server/api/types/GenerativeProvider.py,sha256=Zg2WcCPiM5xmQrOrfm8-Lp9srRiqNaVGeBnyNBTr390,3925
 phoenix/server/api/types/Inferences.py,sha256=wv88PjcK-KwnzmTdukiAX9EV2KX4GqsKXVAUm1JtnDA,3383
 phoenix/server/api/types/InferencesRole.py,sha256=mLfeHpyhGUVX1-tWzT9IwC_cD18BZrD3RA4YsHYuSpA,595
 phoenix/server/api/types/LabelFraction.py,sha256=zsDxdFALrNiGA1eNykeP8o65gbA0HOhRp54MPH_iRAM,93
@@ -287,10 +287,10 @@ 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=0X2PoKEznH0bU3560si0-LLCP2xUT-69XmW7uKVnKu8,2163
-phoenix/server/static/assets/components-CMB_kvRq.js,sha256=i3FlgtH129YHPfsg8a4dYi06Wn1ldoAPjLEjrlyLJmE,390293
-phoenix/server/static/assets/index-baAGkuLZ.js,sha256=mhvaZZ3cX7u00MOah0DciE4_LFdrYfuZ5-4fn3fZ8Zw,8711
-phoenix/server/static/assets/pages-Cw_XOLII.js,sha256=in4giNaXszn2Dn9mkqWLVu4gk4Km5H6xUwxnV4gq2cg,681219
+phoenix/server/static/.vite/manifest.json,sha256=-112B2hz7FbbbyjgMlsRk8-lZneQvQP4g-tM-IoZty0,2163
+phoenix/server/static/assets/components-Caw_U14D.js,sha256=dlWsRhD1_3PxoeVVITya9Yp9IkfgDdo4dO21yKDZF5g,393511
+phoenix/server/static/assets/index-EgdbaF48.js,sha256=FEa1qyVq5vNUWSUvLMoUWK2j1Ra7EdEOIWGLp8pXrgY,8729
+phoenix/server/static/assets/pages-BFZvquxr.js,sha256=Y0tRNqTRt220FpZdHW7vQTOcCi87p8O3zSDS6BftByE,686674
 phoenix/server/static/assets/vendor-DvC8cT4X.js,sha256=-EzJrg6BeYa58JPfaZC6KS5gvkJ8gC1n-Sb7lP-pOkw,2159677
 phoenix/server/static/assets/vendor-DxkFTwjz.css,sha256=nZrkr0u6NNElFGvpWHk9GTHeGoibCXCli1bE7mXZGZg,1816
 phoenix/server/static/assets/vendor-arizeai-Do1793cv.js,sha256=IFADVVEZ8VY9o4iqYighRedZTXSyi1uZMTFFsBeIYis,305184
@@ -337,9 +337,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-7.10.3.dist-info/METADATA,sha256=3hZYS29Yh4Fn6jPQ3oycih7cmB2ihkQeMXnu_74wj6k,23390
-arize_phoenix-7.10.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize_phoenix-7.10.3.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-7.10.3.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-7.10.3.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-7.10.3.dist-info/RECORD,,
+arize_phoenix-7.12.0.dist-info/METADATA,sha256=PJlLicxrJuM_QJI6MgWWu0L29jZxMZEmq1GVREn6Jck,23390
+arize_phoenix-7.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize_phoenix-7.12.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-7.12.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-7.12.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-7.12.0.dist-info/RECORD,,

phoenix/server/api/helpers/playground_clients.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import importlib.util
 import inspect
@@ -7,8 +9,9 @@ import time
 from abc import ABC, abstractmethod
 from collections.abc import AsyncIterator, Callable, Iterator
 from functools import wraps
-from typing import TYPE_CHECKING, Any, Hashable, Mapping, Optional, Union
+from typing import TYPE_CHECKING, Any, Hashable, Mapping, MutableMapping, Optional, Union
 
+import wrapt
 from openinference.instrumentation import safe_json_dumps
 from openinference.semconv.trace import (
     OpenInferenceLLMProviderValues,
@@ -51,11 +54,13 @@ from phoenix.server.api.types.ChatCompletionSubscriptionPayload import (
 from phoenix.server.api.types.GenerativeProvider import GenerativeProviderKey
 
 if TYPE_CHECKING:
+    import httpx
     from anthropic.types import MessageParam, TextBlockParam, ToolResultBlockParam
     from google.generativeai.types import ContentType
     from openai import AsyncAzureOpenAI, AsyncOpenAI
     from openai.types import CompletionUsage
     from openai.types.chat import ChatCompletionMessageParam, ChatCompletionMessageToolCallParam
+    from opentelemetry.util.types import AttributeValue
 
 SetSpanAttributesFn: TypeAlias = Callable[[Mapping[str, Any]], None]
 ChatCompletionChunk: TypeAlias = Union[TextChunk, ToolCallChunk]
@@ -169,7 +174,7 @@ class PlaygroundStreamingClient(ABC):
         model: GenerativeModelInput,
         api_key: Optional[str] = None,
     ) -> None:
-        self._attributes: dict[str, Any] = dict()
+        self._attributes: dict[str, AttributeValue] = dict()
 
     @classmethod
     @abstractmethod
@@ -246,6 +251,7 @@ class OpenAIBaseStreamingClient(PlaygroundStreamingClient):
         self.client = client
         self.model_name = model.name
         self.rate_limiter = PlaygroundRateLimiter(model.provider_key, OpenAIRateLimitError)
+        self.client._client = _HttpxClient(self.client._client, self._attributes)
 
     @classmethod
     def dependencies(cls) -> list[Dependency]:
@@ -696,6 +702,7 @@ class AnthropicStreamingClient(PlaygroundStreamingClient):
         self.client = anthropic.AsyncAnthropic(api_key=api_key)
         self.model_name = model.name
         self.rate_limiter = PlaygroundRateLimiter(model.provider_key, anthropic.RateLimitError)
+        self.client._client = _HttpxClient(self.client._client, self._attributes)
 
     @classmethod
     def dependencies(cls) -> list[Dependency]:
@@ -1005,3 +1012,15 @@ LLM_SYSTEM = SpanAttributes.LLM_SYSTEM
 LLM_TOKEN_COUNT_PROMPT = SpanAttributes.LLM_TOKEN_COUNT_PROMPT
 LLM_TOKEN_COUNT_COMPLETION = SpanAttributes.LLM_TOKEN_COUNT_COMPLETION
 LLM_TOKEN_COUNT_TOTAL = SpanAttributes.LLM_TOKEN_COUNT_TOTAL
+
+
+class _HttpxClient(wrapt.ObjectProxy):  # type: ignore
+    def __init__(self, wrapped: httpx.AsyncClient, attributes: MutableMapping[str, Any]):
+        super().__init__(wrapped)
+        self._self_attributes = attributes
+
+    async def send(self, request: httpx.Request, **kwargs: Any) -> Any:
+        self._self_attributes["url.full"] = str(request.url)
+        self._self_attributes["url.path"] = request.url.path.removeprefix(self.base_url.path)
+        response = await self.__wrapped__.send(request, **kwargs)
+        return response

phoenix/server/api/input_types/GenerativeModelInput.py

@@ -17,7 +17,3 @@ class GenerativeModelInput:
     """ The endpoint to use for the model. Only required for Azure OpenAI models. """
     api_version: Optional[str] = UNSET
     """ The API version to use for the model. """
-
-    def __post_init__(self) -> None:
-        if self.base_url and self.endpoint:
-            raise ValueError("Must not provide both a base URL and an endpoint at the same time.")

phoenix/server/api/types/GenerativeProvider.py

@@ -1,3 +1,4 @@
+import os
 from enum import Enum
 from typing import Any, ClassVar, Optional, Union
 
@@ -34,6 +35,13 @@ class GenerativeProvider:
         OpenInferenceLLMProviderValues.GOOGLE.value: GenerativeProviderKey.GEMINI,
     }
 
+    model_provider_to_api_key_env_var_map: ClassVar[dict[GenerativeProviderKey, str]] = {
+        GenerativeProviderKey.AZURE_OPENAI: "AZURE_OPENAI_API_KEY",
+        GenerativeProviderKey.ANTHROPIC: "ANTHROPIC_API_KEY",
+        GenerativeProviderKey.OPENAI: "OPENAI_API_KEY",
+        GenerativeProviderKey.GEMINI: "GEMINI_API_KEY",
+    }
+
     @strawberry.field
     async def dependencies(self) -> list[str]:
         from phoenix.server.api.helpers.playground_registry import (
@@ -58,6 +66,14 @@ class GenerativeProvider:
             return default_client.dependencies_are_installed()
         return False
 
+    @strawberry.field(description="The API key for the provider")  # type: ignore
+    async def api_key_env_var(self) -> str:
+        return self.model_provider_to_api_key_env_var_map[self.key]
+
+    @strawberry.field(description="Whether the credentials are set on the server for the provider")  # type: ignore
+    async def api_key_set(self) -> bool:
+        return os.environ.get(self.model_provider_to_api_key_env_var_map[self.key]) is not None
+
     @classmethod
     def _infer_model_provider_from_model_name(
         cls,

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

@@ -1,22 +1,22 @@
 {
-  "_components-CMB_kvRq.js": {
-    "file": "assets/components-CMB_kvRq.js",
+  "_components-Caw_U14D.js": {
+    "file": "assets/components-Caw_U14D.js",
     "name": "components",
     "imports": [
       "_vendor-DvC8cT4X.js",
-      "_pages-Cw_XOLII.js",
+      "_pages-BFZvquxr.js",
       "_vendor-arizeai-Do1793cv.js",
       "_vendor-codemirror-BzwZPyJM.js",
       "_vendor-three-DwGkEfCM.js"
     ]
   },
-  "_pages-Cw_XOLII.js": {
-    "file": "assets/pages-Cw_XOLII.js",
+  "_pages-BFZvquxr.js": {
+    "file": "assets/pages-BFZvquxr.js",
     "name": "pages",
     "imports": [
       "_vendor-DvC8cT4X.js",
       "_vendor-arizeai-Do1793cv.js",
-      "_components-CMB_kvRq.js",
+      "_components-Caw_U14D.js",
       "_vendor-codemirror-BzwZPyJM.js",
       "_vendor-recharts-_Jb7JjhG.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-baAGkuLZ.js",
+    "file": "assets/index-EgdbaF48.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-DvC8cT4X.js",
       "_vendor-arizeai-Do1793cv.js",
-      "_pages-Cw_XOLII.js",
-      "_components-CMB_kvRq.js",
+      "_pages-BFZvquxr.js",
+      "_components-Caw_U14D.js",
       "_vendor-three-DwGkEfCM.js",
       "_vendor-codemirror-BzwZPyJM.js",
       "_vendor-shiki-Cl9QBraO.js",