scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

address greptile issues - improved error handling and logging, move the test file to correct dir

Browse source
This commit is contained in:
rzeta-10 2026-03-23 14:13:39 +05:30
parent a5161e6cb2
commit b12ba12749
3 changed files with 29 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
litellm/integrations/akto/akto_logger.py
View file

@ -197,7 +197,8 @@ class AktoLogger(CustomLogger):
"status": "unhealthy",
"error_message": f"Akto returned status {response.status_code}",
}
except Exception:
except Exception as e:
verbose_logger.warning("Akto health check failed: %s", e)
return {"status": "unhealthy", "error_message": "Akto health check failed"}
# ── Logging callbacks ──
@ -206,7 +207,11 @@ class AktoLogger(CustomLogger):
try:
data = self.extract_logging_data(kwargs)
payload = self.build_akto_payload(data, response_obj=response_obj)
self.sync_http_handler.post(**self.request_kwargs(payload))
response = self.sync_http_handler.post(**self.request_kwargs(payload))
if response.status_code >= 400:
verbose_logger.warning(
"Akto ingestion returned %s", response.status_code
)
except Exception as e:
verbose_logger.error("Akto logging error: %s", e)
@ -214,7 +219,13 @@ class AktoLogger(CustomLogger):
try:
data = self.extract_logging_data(kwargs)
payload = self.build_akto_payload(data, response_obj=response_obj)
await self.async_http_handler.post(**self.request_kwargs(payload))
response = await self.async_http_handler.post(
**self.request_kwargs(payload)
)
if response.status_code >= 400:
verbose_logger.warning(
"Akto ingestion returned %s", response.status_code
)
except Exception as e:
verbose_logger.error("Akto logging error: %s", e)
@ -237,7 +248,11 @@ class AktoLogger(CustomLogger):
payload = self.build_akto_payload(
data, status_code=status, response_obj=response_obj
)
self.sync_http_handler.post(**self.request_kwargs(payload))
response = self.sync_http_handler.post(**self.request_kwargs(payload))
if response.status_code >= 400:
verbose_logger.warning(
"Akto ingestion returned %s", response.status_code
)
except Exception as e:
verbose_logger.error("Akto logging error (failure): %s", e)
@ -248,6 +263,12 @@ class AktoLogger(CustomLogger):
payload = self.build_akto_payload(
data, status_code=status, response_obj=response_obj
)
await self.async_http_handler.post(**self.request_kwargs(payload))
response = await self.async_http_handler.post(
**self.request_kwargs(payload)
)
if response.status_code >= 400:
verbose_logger.warning(
"Akto ingestion returned %s", response.status_code
)
except Exception as e:
verbose_logger.error("Akto logging error (failure): %s", e)

5
litellm/proxy/health_endpoints/_health_endpoints.py
View file

@ -341,9 +341,10 @@ async def health_services_endpoint( # noqa: PLR0915
"message": "Mock LLM request made - check langfuse.",
}
elif service == "akto":
from litellm.litellm_core_utils.litellm_logging import aktoLogger as _akto
from litellm.integrations.akto.akto_logger import AktoLogger
akto_logger = AktoLogger()
akto_logger = _akto if _akto is not None else AktoLogger()
response = await akto_logger.async_health_check()
return {
"status": response["status"],
@ -353,7 +354,7 @@ async def health_services_endpoint( # noqa: PLR0915
else "Akto is healthy"
),
}
if service == "webhook":
user_info = CallInfo(
token=user_api_key_dict.token or "",

0
tests/guardrails_tests/test_akto_logger.py → tests/logging_callback_tests/test_akto_logger.py
View file