arize-phoenix 10.6.2__py3-none-any.whl → 10.7.0__py3-none-any.whl

phoenix/server/api/input_types/GenerativeCredentialInput.py

@@ -0,0 +1,9 @@
+import strawberry
+
+
+@strawberry.input
+class GenerativeCredentialInput:
+    env_var_name: str
+    """The name of the environment variable to set."""
+    value: str
+    """The value of the environment variable to set."""

phoenix/server/api/mutations/chat_mutations.py

@@ -30,6 +30,7 @@ from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, CustomGraphQLError, NotFound
 from phoenix.server.api.helpers.dataset_helpers import get_dataset_example_output
 from phoenix.server.api.helpers.playground_clients import (
+    PlaygroundClientCredential,
     PlaygroundStreamingClient,
     initialize_playground_clients,
 )
@@ -132,9 +133,17 @@ class ChatCompletionMutationMixin:
         if llm_client_class is None:
             raise BadRequest(f"Unknown LLM provider: '{provider_key.value}'")
         try:
+            # Convert GraphQL credentials to PlaygroundCredential objects
+            credentials = None
+            if input.credentials:
+                credentials = [
+                    PlaygroundClientCredential(env_var_name=cred.env_var_name, value=cred.value)
+                    for cred in input.credentials
+                ]
+
             llm_client = llm_client_class(
                 model=input.model,
-                api_key=input.api_key,
+                credentials=credentials,
             )
         except CustomGraphQLError:
             raise
@@ -200,7 +209,7 @@ class ChatCompletionMutationMixin:
                         llm_client,
                         ChatCompletionInput(
                             model=input.model,
-                            api_key=input.api_key,
+                            credentials=input.credentials,
                             messages=input.messages,
                             tools=input.tools,
                             invocation_parameters=input.invocation_parameters,
@@ -281,9 +290,17 @@ class ChatCompletionMutationMixin:
         if llm_client_class is None:
             raise BadRequest(f"Unknown LLM provider: '{provider_key.value}'")
         try:
+            # Convert GraphQL credentials to PlaygroundCredential objects
+            credentials = None
+            if input.credentials:
+                credentials = [
+                    PlaygroundClientCredential(env_var_name=cred.env_var_name, value=cred.value)
+                    for cred in input.credentials
+                ]
+
             llm_client = llm_client_class(
                 model=input.model,
-                api_key=input.api_key,
+                credentials=credentials,
             )
         except CustomGraphQLError:
             raise

phoenix/server/api/subscriptions.py

@@ -30,6 +30,7 @@ from phoenix.server.api.auth import IsLocked, IsNotReadOnly
 from phoenix.server.api.context import Context
 from phoenix.server.api.exceptions import BadRequest, CustomGraphQLError, NotFound
 from phoenix.server.api.helpers.playground_clients import (
+    PlaygroundClientCredential,
     PlaygroundStreamingClient,
     initialize_playground_clients,
 )
@@ -98,9 +99,17 @@ class Subscription:
         if llm_client_class is None:
             raise BadRequest(f"Unknown LLM provider: '{provider_key.value}'")
         try:
+            # Convert GraphQL credentials to PlaygroundCredential objects
+            playground_credentials = None
+            if input.credentials:
+                playground_credentials = [
+                    PlaygroundClientCredential(env_var_name=cred.env_var_name, value=cred.value)
+                    for cred in input.credentials
+                ]
+
             llm_client = llm_client_class(
                 model=input.model,
-                api_key=input.api_key,
+                credentials=playground_credentials,
             )
         except CustomGraphQLError:
             raise
@@ -176,9 +185,17 @@ class Subscription:
         if llm_client_class is None:
             raise BadRequest(f"Unknown LLM provider: '{provider_key.value}'")
         try:
+            # Convert GraphQL credentials to PlaygroundCredential objects
+            playground_credentials = None
+            if input.credentials:
+                playground_credentials = [
+                    PlaygroundClientCredential(env_var_name=cred.env_var_name, value=cred.value)
+                    for cred in input.credentials
+                ]
+
             llm_client = llm_client_class(
                 model=input.model,
-                api_key=input.api_key,
+                credentials=playground_credentials,
             )
         except CustomGraphQLError:
             raise

phoenix/server/api/types/GenerativeProvider.py

@@ -16,6 +16,13 @@ class GenerativeProviderKey(Enum):
     GOOGLE = "Google AI Studio"
     DEEPSEEK = "DeepSeek"
     XAI = "xAI"
+    OLLAMA = "Ollama"
+
+
+@strawberry.type
+class GenerativeProviderCredentialConfig:
+    env_var_name: str
+    is_required: bool
 
 
 @strawberry.type
@@ -30,6 +37,7 @@ class GenerativeProvider:
         GenerativeProviderKey.GOOGLE: ["gemini"],
         GenerativeProviderKey.DEEPSEEK: ["deepseek"],
         GenerativeProviderKey.XAI: ["grok"],
+        GenerativeProviderKey.OLLAMA: ["llama", "mistral", "codellama", "phi", "qwen", "gemma"],
     }
 
     attribute_provider_to_generative_provider_map: ClassVar[dict[str, GenerativeProviderKey]] = {
@@ -41,15 +49,35 @@ class GenerativeProvider:
         # The provider will be determined through model name prefix matching instead
         # Note: xAI uses OpenAI compatibility but we can't duplicate the key in the dict
         # The provider will be determined through model name prefix matching instead
+        # Note: Ollama 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]] = {
-        GenerativeProviderKey.AZURE_OPENAI: "AZURE_OPENAI_API_KEY",
-        GenerativeProviderKey.ANTHROPIC: "ANTHROPIC_API_KEY",
-        GenerativeProviderKey.OPENAI: "OPENAI_API_KEY",
-        GenerativeProviderKey.GOOGLE: "GEMINI_API_KEY",
-        GenerativeProviderKey.DEEPSEEK: "DEEPSEEK_API_KEY",
-        GenerativeProviderKey.XAI: "XAI_API_KEY",
+    """
+    A map of model provider keys to their credential requirements.
+    E.x. OpenAI requires a single API key
+    """
+    model_provider_to_credential_requirements_map: ClassVar[
+        dict[GenerativeProviderKey, GenerativeProviderCredentialConfig]
+    ] = {
+        GenerativeProviderKey.AZURE_OPENAI: GenerativeProviderCredentialConfig(
+            env_var_name="AZURE_OPENAI_API_KEY", is_required=True
+        ),
+        GenerativeProviderKey.ANTHROPIC: GenerativeProviderCredentialConfig(
+            env_var_name="ANTHROPIC_API_KEY", is_required=True
+        ),
+        GenerativeProviderKey.OPENAI: GenerativeProviderCredentialConfig(
+            env_var_name="OPENAI_API_KEY", is_required=True
+        ),
+        GenerativeProviderKey.GOOGLE: GenerativeProviderCredentialConfig(
+            env_var_name="GEMINI_API_KEY", is_required=True
+        ),
+        GenerativeProviderKey.DEEPSEEK: GenerativeProviderCredentialConfig(
+            env_var_name="DEEPSEEK_API_KEY", is_required=True
+        ),
+        GenerativeProviderKey.XAI: GenerativeProviderCredentialConfig(
+            env_var_name="XAI_API_KEY", is_required=True
+        ),
     }
 
     @strawberry.field
@@ -76,13 +104,17 @@ 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="The credential requirements for the provider")  # type: ignore
+    async def credential_requirements(self) -> list[GenerativeProviderCredentialConfig]:
+        return [self.model_provider_to_credential_requirements_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 getenv(self.model_provider_to_api_key_env_var_map[self.key]) is not None
+    async def credentials_set(self) -> bool:
+        # Check if every required credential is set
+        return all(
+            getenv(credential_config.env_var_name) is not None or not credential_config.is_required
+            for credential_config in await self.credential_requirements()
+        )
 
     @classmethod
     def _infer_model_provider_from_model_name(

phoenix/server/main.py

@@ -459,6 +459,7 @@ def main() -> None:
         host=host,  # type: ignore[arg-type]
         port=port,
         root_path=host_root_path,
+        log_level=Settings.logging_level,
     )
 
     if tls_enabled_for_http:

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

@@ -1,22 +1,22 @@
 {
-  "_components-J06J_j9O.js": {
-    "file": "assets/components-J06J_j9O.js",
+  "_components-CuV9v4w3.js": {
+    "file": "assets/components-CuV9v4w3.js",
     "name": "components",
     "imports": [
       "_vendor-B52WHALA.js",
-      "_pages-nxs-tDxQ.js",
+      "_pages-DTLdnakm.js",
       "_vendor-arizeai-DGHetzZW.js",
       "_vendor-codemirror-QIdVJrP_.js",
       "_vendor-three-C5WAXd5r.js"
     ]
   },
-  "_pages-nxs-tDxQ.js": {
-    "file": "assets/pages-nxs-tDxQ.js",
+  "_pages-DTLdnakm.js": {
+    "file": "assets/pages-DTLdnakm.js",
     "name": "pages",
     "imports": [
       "_vendor-B52WHALA.js",
       "_vendor-arizeai-DGHetzZW.js",
-      "_components-J06J_j9O.js",
+      "_components-CuV9v4w3.js",
       "_vendor-codemirror-QIdVJrP_.js",
       "_vendor-recharts-GmWamXB4.js"
     ]
@@ -69,15 +69,15 @@
     "name": "vendor-three"
   },
   "index.tsx": {
-    "file": "assets/index-DfT39tc3.js",
+    "file": "assets/index-BeIqg4nl.js",
     "name": "index",
     "src": "index.tsx",
     "isEntry": true,
     "imports": [
       "_vendor-B52WHALA.js",
       "_vendor-arizeai-DGHetzZW.js",
-      "_pages-nxs-tDxQ.js",
-      "_components-J06J_j9O.js",
+      "_pages-DTLdnakm.js",
+      "_components-CuV9v4w3.js",
       "_vendor-three-C5WAXd5r.js",
       "_vendor-codemirror-QIdVJrP_.js",
       "_vendor-shiki-C8cTrXI5.js",