From c153fd365684f76db0ba2f6888e71ffdb40d29ce Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 18 Mar 2025 16:54:31 -0700 Subject: [PATCH] fix arize config --- litellm/integrations/arize/arize.py | 5 ----- litellm/types/integrations/arize.py | 9 +++++---- 2 files changed, 5 insertions(+), 9 deletions(-) diff --git a/litellm/integrations/arize/arize.py b/litellm/integrations/arize/arize.py index 768f8a655d..1a1b8f9979 100644 --- a/litellm/integrations/arize/arize.py +++ b/litellm/integrations/arize/arize.py @@ -51,11 +51,6 @@ class ArizeLogger(OpenTelemetry): space_key = os.environ.get("ARIZE_SPACE_KEY") api_key = os.environ.get("ARIZE_API_KEY") - if not space_key: - raise ValueError("ARIZE_SPACE_KEY not found in environment variables") - if not api_key: - raise ValueError("ARIZE_API_KEY not found in environment variables") - grpc_endpoint = os.environ.get("ARIZE_ENDPOINT") http_endpoint = os.environ.get("ARIZE_HTTP_ENDPOINT") diff --git a/litellm/types/integrations/arize.py b/litellm/types/integrations/arize.py index 24298fc363..e1ec1755f8 100644 --- a/litellm/types/integrations/arize.py +++ b/litellm/types/integrations/arize.py @@ -1,4 +1,4 @@ -from typing import TYPE_CHECKING, Literal, Any +from typing import TYPE_CHECKING, Any, Literal, Optional from pydantic import BaseModel @@ -6,9 +6,10 @@ if TYPE_CHECKING: Protocol = Literal["otlp_grpc", "otlp_http"] else: Protocol = Any - + + class ArizeConfig(BaseModel): - space_key: str - api_key: str + space_key: Optional[str] = None + api_key: Optional[str] = None protocol: Protocol endpoint: str