From 245d705e6ca7ee9c8cda2dccb2d40256480726f2 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Thu, 5 Feb 2026 14:28:28 +0530
Subject: [PATCH] Add new methods in chat completion

---
 litellm/integrations/custom_logger.py | 31 +++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/litellm/integrations/custom_logger.py b/litellm/integrations/custom_logger.py
index 07d237c475..4a341863d4 100644
--- a/litellm/integrations/custom_logger.py
+++ b/litellm/integrations/custom_logger.py
@@ -664,6 +664,37 @@ class CustomLogger:  # https://docs.litellm.ai/docs/observability/custom_callbac
             return final_response
         """
         pass
+    
+    async def async_should_run_chat_completion_agentic_loop(
+        self,
+        response: Any,
+        model: str,
+        messages: List[Dict],
+        tools: Optional[List[Dict]],
+        stream: bool,
+        custom_llm_provider: str,
+        kwargs: Dict,
+    ) -> Tuple[bool, Dict]:
+        """
+        Hook to determine if chat completion agentic loop should be executed.
+        """
+        return False, {}
+
+    async def async_run_chat_completion_agentic_loop(
+        self,
+        tools: Dict,
+        model: str,
+        messages: List[Dict],
+        response: Any,
+        optional_params: Dict,
+        logging_obj: "LiteLLMLoggingObj",
+        stream: bool,
+        kwargs: Dict,
+    ) -> Any:
+        """
+        Hook to execute chat completion agentic loop based on context from should_run hook.
+        """
+        pass
 
     # Useful helpers for custom logger classes