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

dd logger, inherit from .envs

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-26 22:01:06 -08:00
parent fe151db27c
commit 673fd7e257
2 changed files with 33 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

44
litellm/integrations/datadog/datadog.py
View file

@ -279,11 +279,11 @@ class DataDogLogger(CustomBatchLogger):
verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload) verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)
dd_payload = DatadogPayload( dd_payload = DatadogPayload(
ddsource=os.getenv("DD_SOURCE", "litellm"), ddsource=self._get_datadog_source(),
ddtags="", ddtags=self._get_datadog_tags(),
hostname="", hostname=self._get_datadog_hostname(),
message=json_payload, message=json_payload,
service="litellm-server", service=self._get_datadog_service(),
status=status, status=status,
) )
return dd_payload return dd_payload
@ -387,11 +387,11 @@ class DataDogLogger(CustomBatchLogger):
json_payload = json.dumps(_exception_payload) json_payload = json.dumps(_exception_payload)
verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload) verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)
dd_payload = DatadogPayload( dd_payload = DatadogPayload(
ddsource=os.getenv("DD_SOURCE", "litellm"), ddsource=self._get_datadog_source(),
ddtags="", ddtags=self._get_datadog_tags(),
hostname="", hostname=self._get_datadog_hostname(),
message=json_payload, message=json_payload,
service="litellm-server", service=self._get_datadog_service(),
status=DataDogStatus.ERROR, status=DataDogStatus.ERROR,
) )
@ -473,11 +473,31 @@ class DataDogLogger(CustomBatchLogger):
verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload) verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)
dd_payload = DatadogPayload( dd_payload = DatadogPayload(
ddsource=os.getenv("DD_SOURCE", "litellm"), ddsource=self._get_datadog_source(),
ddtags="", ddtags=self._get_datadog_tags(),
hostname="", hostname=self._get_datadog_hostname(),
message=json_payload, message=json_payload,
service="litellm-server", service=self._get_datadog_service(),
status=DataDogStatus.INFO, status=DataDogStatus.INFO,
) )
return dd_payload return dd_payload
@staticmethod
def _get_datadog_tags():
return f"env:{os.getenv('DD_ENV', 'unknown')},service:{os.getenv('DD_SERVICE', 'litellm')},version:{os.getenv('DD_VERSION', 'unknown')}"
@staticmethod
def _get_datadog_source():
return os.getenv("DD_SOURCE", "litellm")
@staticmethod
def _get_datadog_service():
return os.getenv("DD_SERVICE", "litellm")
@staticmethod
def _get_datadog_hostname():
return ""
@staticmethod
def _get_datadog_env():
return os.getenv("DD_ENV", "unknown")

2
litellm/proxy/proxy_config.yaml
View file

@ -2,4 +2,4 @@ include:
- model_config.yaml - model_config.yaml
litellm_settings: litellm_settings:
callbacks: ["prometheus"] callbacks: ["datadog"]