From f023f7ff77488c304a14762d5979a59740eae63e Mon Sep 17 00:00:00 2001
From: devarakondasrikanth <devarakondasrikanth@ymail.com>
Date: Fri, 13 Mar 2026 20:15:44 -0700
Subject: [PATCH] fixing PR comments

---
 litellm/proxy/proxy_server.py | 6 ++++--
 litellm/proxy/utils.py        | 8 ++++----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 7f1c06e32a0..d3fe1406abd 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -5614,12 +5614,14 @@ async def async_data_generator(
             def _discard_task(t: asyncio.Task[None]) -> None:
                 _post_guardrail_log_tasks.discard(t)
 
+            logging_obj_for_log = request_data.get("litellm_logging_obj")
+            request_data_snapshot = dict(request_data)
             _task = asyncio.create_task(
                 _async_data_generator_fire_post_guardrail_log(
-                    request_data=request_data,
+                    request_data=request_data_snapshot,
                     user_api_key_dict=user_api_key_dict,
                     chunks_for_log=_streaming_chunks_for_log,
-                    logging_obj=request_data.get("litellm_logging_obj"),
+                    logging_obj=logging_obj_for_log,
                 )
             )
             _post_guardrail_log_tasks.add(_task)
diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 4771fbd9691..e269d5a4700 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -1995,11 +1995,11 @@ class ProxyLogging:
                 logging_obj, "model_call_details", None
             ):
                 # Merge so request data wins overall, but model_call_details wins for
-                # LLM-lifecycle timing fields so callbacks get accurate start/end times.
+                # start_time so callbacks get accurate LLM start. end_time in kwargs is
+                # left to match the end_time argument (hook run time); use llm_end_time for LLM-only end.
                 kwargs = {**logging_obj.model_call_details, **kwargs}
-                for key in ("start_time", "end_time"):
-                    if key in logging_obj.model_call_details:
-                        kwargs[key] = logging_obj.model_call_details[key]
+                if "start_time" in logging_obj.model_call_details:
+                    kwargs["start_time"] = logging_obj.model_call_details["start_time"]
             kwargs["user_api_key_dict"] = user_api_key_dict
             start_time = None
             if logging_obj is not None: