arize-phoenix 10.1.0__tar.gz → 10.2.1__tar.gz

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize-phoenix
-Version: 10.1.0
+Version: 10.2.1
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -22,7 +22,7 @@ Requires-Dist: aiosqlite
 Requires-Dist: alembic<2,>=1.3.0
 Requires-Dist: arize-phoenix-client
 Requires-Dist: arize-phoenix-evals>=0.20.6
-Requires-Dist: arize-phoenix-otel>=0.9.2
+Requires-Dist: arize-phoenix-otel>=0.10.1
 Requires-Dist: authlib
 Requires-Dist: cachetools
 Requires-Dist: email-validator

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/pyproject.toml

@@ -53,7 +53,7 @@ dependencies = [
   "cachetools",
   "python-multipart",  # see https://www.starlette.io/#dependencies
   "arize-phoenix-evals>=0.20.6",
-  "arize-phoenix-otel>=0.9.2",
+  "arize-phoenix-otel>=0.10.1",
   "fastapi",
   "pydantic>=2.1.0", # exclude 2.0.* since it does not support the `json_encoders` configuration setting
   "authlib",

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/db/insertion/dataset.py

@@ -35,6 +35,7 @@ Examples: TypeAlias = Iterable[ExampleContent]
 @dataclass(frozen=True)
 class DatasetExampleAdditionEvent(DataManipulationEvent):
     dataset_id: DatasetId
+    dataset_version_id: DatasetVersionId
 
 
 async def insert_dataset(
@@ -205,7 +206,7 @@ async def add_dataset_examples(
                 f"{dataset_example_id=}"
             )
             raise
-    return DatasetExampleAdditionEvent(dataset_id=dataset_id)
+    return DatasetExampleAdditionEvent(dataset_id=dataset_id, dataset_version_id=dataset_version_id)
 
 
 @dataclass(frozen=True)

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/db/types/model_provider.py

@@ -6,3 +6,4 @@ class ModelProvider(Enum):
     AZURE_OPENAI = "AZURE_OPENAI"
     ANTHROPIC = "ANTHROPIC"
     GOOGLE = "GOOGLE"
+    DEEPSEEK = "DEEPSEEK"

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/server/api/helpers/playground_clients.py

@@ -453,6 +453,35 @@ class OpenAIBaseStreamingClient(PlaygroundStreamingClient):
         yield LLM_TOKEN_COUNT_TOTAL, usage.total_tokens
 
 
+@register_llm_client(
+    provider_key=GenerativeProviderKey.DEEPSEEK,
+    model_names=[
+        PROVIDER_DEFAULT,
+        "deepseek-chat",
+        "deepseek-reasoner",
+    ],
+)
+class DeepSeekStreamingClient(OpenAIBaseStreamingClient):
+    def __init__(
+        self,
+        model: GenerativeModelInput,
+        api_key: Optional[str] = None,
+    ) -> None:
+        from openai import AsyncOpenAI
+
+        base_url = model.base_url or getenv("DEEPSEEK_BASE_URL")
+        if not (api_key := api_key or getenv("DEEPSEEK_API_KEY")):
+            if not base_url:
+                raise BadRequest("An API key is required for DeepSeek models")
+            api_key = "sk-fake-api-key"
+        client = AsyncOpenAI(api_key=api_key, base_url=base_url or "https://api.deepseek.com")
+        super().__init__(client=client, model=model, api_key=api_key)
+        # DeepSeek uses OpenAI-compatible API but we'll track it as a separate provider
+        # Adding a custom "deepseek" provider value to make it distinguishable in traces
+        self._attributes[LLM_PROVIDER] = "deepseek"
+        self._attributes[LLM_SYSTEM] = OpenInferenceLLMSystemValues.OPENAI.value
+
+
 @register_llm_client(
     provider_key=GenerativeProviderKey.OPENAI,
     model_names=[

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/server/api/helpers/prompts/models.py

@@ -332,11 +332,20 @@ class PromptAzureOpenAIInvocationParametersContent(PromptOpenAIInvocationParamet
     pass
 
 
+class PromptDeepSeekInvocationParametersContent(PromptOpenAIInvocationParametersContent):
+    pass
+
+
 class PromptAzureOpenAIInvocationParameters(DBBaseModel):
     type: Literal["azure_openai"]
     azure_openai: PromptAzureOpenAIInvocationParametersContent
 
 
+class PromptDeepSeekInvocationParameters(DBBaseModel):
+    type: Literal["deepseek"]
+    deepseek: PromptDeepSeekInvocationParametersContent
+
+
 class PromptAnthropicThinkingConfigDisabled(DBBaseModel):
     type: Literal["disabled"]
 
@@ -391,6 +400,7 @@ PromptInvocationParameters: TypeAlias = Annotated[
         PromptAzureOpenAIInvocationParameters,
         PromptAnthropicInvocationParameters,
         PromptGoogleInvocationParameters,
+        PromptDeepSeekInvocationParameters,
     ],
     Field(..., discriminator="type"),
 ]
@@ -407,6 +417,8 @@ def get_raw_invocation_parameters(
         return invocation_parameters.anthropic.model_dump()
     if isinstance(invocation_parameters, PromptGoogleInvocationParameters):
         return invocation_parameters.google.model_dump()
+    if isinstance(invocation_parameters, PromptDeepSeekInvocationParameters):
+        return invocation_parameters.deepseek.model_dump()
     assert_never(invocation_parameters)
 
 
@@ -420,6 +432,7 @@ def is_prompt_invocation_parameters(
             PromptAzureOpenAIInvocationParameters,
             PromptAnthropicInvocationParameters,
             PromptGoogleInvocationParameters,
+            PromptDeepSeekInvocationParameters,
         ),
     )
 
@@ -444,6 +457,13 @@ def validate_invocation_parameters(
                 invocation_parameters
             ),
         )
+    elif model_provider is ModelProvider.DEEPSEEK:
+        return PromptDeepSeekInvocationParameters(
+            type="deepseek",
+            deepseek=PromptDeepSeekInvocationParametersContent.model_validate(
+                invocation_parameters
+            ),
+        )
     elif model_provider is ModelProvider.ANTHROPIC:
         return PromptAnthropicInvocationParameters(
             type="anthropic",

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/server/api/routers/v1/datasets.py

@@ -312,6 +312,7 @@ async def list_dataset_versions(
 
 class UploadDatasetData(V1RoutesBaseModel):
     dataset_id: str
+    version_id: str
 
 
 class UploadDatasetResponseBody(ResponseBody[UploadDatasetData]):
@@ -468,10 +469,15 @@ async def upload_dataset(
     )
     if sync:
         async with request.app.state.db() as session:
-            dataset_id = (await operation(session)).dataset_id
+            event = await operation(session)
+            dataset_id = event.dataset_id
+            version_id = event.dataset_version_id
         request.state.event_queue.put(DatasetInsertEvent((dataset_id,)))
         return UploadDatasetResponseBody(
-            data=UploadDatasetData(dataset_id=str(GlobalID(Dataset.__name__, str(dataset_id))))
+            data=UploadDatasetData(
+                dataset_id=str(GlobalID(Dataset.__name__, str(dataset_id))),
+                version_id=str(GlobalID(DatasetVersion.__name__, str(version_id))),
+            )
         )
     try:
         request.state.enqueue_operation(operation)

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/server/api/types/GenerativeProvider.py

@@ -14,6 +14,7 @@ class GenerativeProviderKey(Enum):
     ANTHROPIC = "Anthropic"
     AZURE_OPENAI = "Azure OpenAI"
     GOOGLE = "Google AI Studio"
+    DEEPSEEK = "DeepSeek"
 
 
 @strawberry.type
@@ -26,6 +27,7 @@ class GenerativeProvider:
         GenerativeProviderKey.ANTHROPIC: ["claude"],
         GenerativeProviderKey.OPENAI: ["gpt", "o1"],
         GenerativeProviderKey.GOOGLE: ["gemini"],
+        GenerativeProviderKey.DEEPSEEK: ["deepseek"],
     }
 
     attribute_provider_to_generative_provider_map: ClassVar[dict[str, GenerativeProviderKey]] = {
@@ -33,6 +35,8 @@ class GenerativeProvider:
         OpenInferenceLLMProviderValues.ANTHROPIC.value: GenerativeProviderKey.ANTHROPIC,
         OpenInferenceLLMProviderValues.AZURE.value: GenerativeProviderKey.AZURE_OPENAI,
         OpenInferenceLLMProviderValues.GOOGLE.value: GenerativeProviderKey.GOOGLE,
+        # Note: DeepSeek uses OpenAI compatibility but we can't duplicate the key in the dict
+        # The provider will be determined through model name prefix matching instead
     }
 
     model_provider_to_api_key_env_var_map: ClassVar[dict[GenerativeProviderKey, str]] = {
@@ -40,6 +44,7 @@ class GenerativeProvider:
         GenerativeProviderKey.ANTHROPIC: "ANTHROPIC_API_KEY",
         GenerativeProviderKey.OPENAI: "OPENAI_API_KEY",
         GenerativeProviderKey.GOOGLE: "GEMINI_API_KEY",
+        GenerativeProviderKey.DEEPSEEK: "DEEPSEEK_API_KEY",
     }
 
     @strawberry.field

{arize_phoenix-10.1.0 → arize_phoenix-10.2.1}/src/phoenix/server/static/.vite/manifest.json

@@ -1,22 +1,22 @@
 {
-  "_components-BHJEWQsc.js": {
-    "file": "assets/components-BHJEWQsc.js",
+  "_components-ClD3sHta.js": {
+    "file": "assets/components-ClD3sHta.js",
     "name": "components",
     "imports": [
       "_vendor-DOUbLVp5.js",
-      "_pages-kpw8RDmo.js",
+      "_pages-BFtNRfTL.js",
       "_vendor-arizeai-DHqMQzfV.js",
       "_vendor-codemirror-DWdZV1Is.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-kpw8RDmo.js": {
-    "file": "assets/pages-kpw8RDmo.js",
+  "_pages-BFtNRfTL.js": {
+    "file": "assets/pages-BFtNRfTL.js",
     "name": "pages",
     "imports": [
       "_vendor-DOUbLVp5.js",
       "_vendor-arizeai-DHqMQzfV.js",
-      "_components-BHJEWQsc.js",
+      "_components-ClD3sHta.js",
       "_vendor-codemirror-DWdZV1Is.js",
       "_vendor-recharts-BfHdRd1Y.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-CAt4FOOX.js",
+    "file": "assets/index-CXawXHw0.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-DOUbLVp5.js",
       "_vendor-arizeai-DHqMQzfV.js",
-      "_pages-kpw8RDmo.js",
-      "_components-BHJEWQsc.js",
+      "_pages-BFtNRfTL.js",
+      "_components-ClD3sHta.js",
       "_vendor-three-C5WAXd5r.js",
       "_vendor-codemirror-DWdZV1Is.js",
       "_vendor-shiki-CHu75YVL.js",