phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix arize config

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-18 16:54:31 -07:00
parent 9144f25a69
commit c153fd3656
2 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/arize/arize.py
View file

@ -51,11 +51,6 @@ class ArizeLogger(OpenTelemetry):
space_key = os.environ.get("ARIZE_SPACE_KEY") space_key = os.environ.get("ARIZE_SPACE_KEY")
api_key = os.environ.get("ARIZE_API_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") grpc_endpoint = os.environ.get("ARIZE_ENDPOINT")
http_endpoint = os.environ.get("ARIZE_HTTP_ENDPOINT") http_endpoint = os.environ.get("ARIZE_HTTP_ENDPOINT")

9
litellm/types/integrations/arize.py
View file

@ -1,4 +1,4 @@
from typing import TYPE_CHECKING, Literal, Any from typing import TYPE_CHECKING, Any, Literal, Optional
from pydantic import BaseModel from pydantic import BaseModel
@ -6,9 +6,10 @@ if TYPE_CHECKING:
Protocol = Literal["otlp_grpc", "otlp_http"] Protocol = Literal["otlp_grpc", "otlp_http"]
else: else:
Protocol = Any Protocol = Any
class ArizeConfig(BaseModel): class ArizeConfig(BaseModel):
space_key: str space_key: Optional[str] = None
api_key: str api_key: Optional[str] = None
protocol: Protocol protocol: Protocol
endpoint: str endpoint: str