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