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

fix(openmeter.py): fix get from env

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-02 16:34:08 -07:00
parent 7e04447159
commit 115eb1abe7
1 changed files with 6 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/integrations/openmeter.py
View file

@ -38,7 +38,7 @@ class OpenMeterLogger(CustomLogger):
in the environment in the environment
""" """
missing_keys = [] missing_keys = []
if litellm.get_secret("OPENMETER_API_KEY", None) is None: if os.getenv("OPENMETER_API_KEY", None) is None:
missing_keys.append("OPENMETER_API_KEY") missing_keys.append("OPENMETER_API_KEY")
if len(missing_keys) > 0: if len(missing_keys) > 0:
@ -71,15 +71,13 @@ class OpenMeterLogger(CustomLogger):
} }
def log_success_event(self, kwargs, response_obj, start_time, end_time): def log_success_event(self, kwargs, response_obj, start_time, end_time):
_url = litellm.get_secret( _url = os.getenv("OPENMETER_API_ENDPOINT", "https://openmeter.cloud")
"OPENMETER_API_ENDPOINT", default_value="https://openmeter.cloud"
)
if _url.endswith("/"): if _url.endswith("/"):
_url += "api/v1/events" _url += "api/v1/events"
else: else:
_url += "/api/v1/events" _url += "/api/v1/events"
api_key = litellm.get_secret("OPENMETER_API_KEY") api_key = os.getenv("OPENMETER_API_KEY")
_data = self._common_logic(kwargs=kwargs, response_obj=response_obj) _data = self._common_logic(kwargs=kwargs, response_obj=response_obj)
self.sync_http_handler.post( self.sync_http_handler.post(
@ -92,15 +90,13 @@ class OpenMeterLogger(CustomLogger):
) )
async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
_url = litellm.get_secret( _url = os.getenv("OPENMETER_API_ENDPOINT", "https://openmeter.cloud")
"OPENMETER_API_ENDPOINT", default_value="https://openmeter.cloud"
)
if _url.endswith("/"): if _url.endswith("/"):
_url += "api/v1/events" _url += "api/v1/events"
else: else:
_url += "/api/v1/events" _url += "/api/v1/events"
api_key = litellm.get_secret("OPENMETER_API_KEY") api_key = os.getenv("OPENMETER_API_KEY")
_data = self._common_logic(kwargs=kwargs, response_obj=response_obj) _data = self._common_logic(kwargs=kwargs, response_obj=response_obj)
_headers = { _headers = {
@ -117,7 +113,6 @@ class OpenMeterLogger(CustomLogger):
response.raise_for_status() response.raise_for_status()
except Exception as e: except Exception as e:
print(f"\nAn Exception Occurred - {str(e)}")
if hasattr(response, "text"): if hasattr(response, "text"):
print(f"\nError Message: {response.text}") litellm.print_verbose(f"\nError Message: {response.text}")
raise e raise e