From 5e529e14629c85ca32fb178562db327df221398b Mon Sep 17 00:00:00 2001 From: Gary Caldwell Date: Sat, 26 Apr 2025 10:46:08 -0700 Subject: [PATCH] batch debugging --- litellm/integrations/SlackAlerting/slack_alerting.py | 1 + litellm/integrations/custom_batch_logger.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/litellm/integrations/SlackAlerting/slack_alerting.py b/litellm/integrations/SlackAlerting/slack_alerting.py index 5e82f1e76cf..39d6408bb63 100644 --- a/litellm/integrations/SlackAlerting/slack_alerting.py +++ b/litellm/integrations/SlackAlerting/slack_alerting.py @@ -1422,6 +1422,7 @@ Model Info: print(f"slack url before log_queue {slack_webhook_url}") print(f"slack payload {payload}") + print(f"slack batchsize {self.batch_size}") if isinstance(slack_webhook_url, list): for url in slack_webhook_url: self.log_queue.append( diff --git a/litellm/integrations/custom_batch_logger.py b/litellm/integrations/custom_batch_logger.py index f9d4496c21f..976b0ce1fec 100644 --- a/litellm/integrations/custom_batch_logger.py +++ b/litellm/integrations/custom_batch_logger.py @@ -42,6 +42,7 @@ class CustomBatchLogger(CustomLogger): await self.flush_queue() async def flush_queue(self): + print(f"flushing queue: {self.flush_lock}") if self.flush_lock is None: return @@ -50,6 +51,7 @@ class CustomBatchLogger(CustomLogger): verbose_logger.debug( "CustomLogger: Flushing batch of %s events", len(self.log_queue) ) + print("calling send_batch") await self.async_send_batch() self.log_queue.clear() self.last_flush_time = time.time()