phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat) support headers for generic API logger

Browse source
This commit is contained in:
ishaan-jaff 2024-02-15 13:50:01 -08:00
parent b3f5402017
commit 3e90acb750
2 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
enterprise/callbacks/api_callback.py
View file

@ -31,10 +31,24 @@ class GenericAPILogger:
# Class variables or attributes
def __init__(self, endpoint=None, headers=None):
try:
verbose_logger.debug(f"in init GenericAPILogger, endpoint {endpoint}")
if endpoint == None:
# check env for "GENERIC_LOGGER_ENDPOINT"
if os.getenv("GENERIC_LOGGER_ENDPOINT"):
# Do something with the endpoint
endpoint = os.getenv("GENERIC_LOGGER_ENDPOINT")
else:
# Handle the case when the endpoint is not found in the environment variables
raise ValueError(
f"endpoint not set for GenericAPILogger, GENERIC_LOGGER_ENDPOINT not found in environment variables"
)
headers = headers or litellm.generic_logger_headers
self.endpoint = endpoint
self.headers = headers
verbose_logger.debug(
f"in init GenericAPILogger, endpoint {self.endpoint}, headers {self.headers}"
)
pass
except Exception as e: