From 2823d09e1bcfc99cee126583f3552e5d5b32dac5 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 8 Sep 2026 12:30:05 -0700
Subject: [PATCH] refactor(policy_engine): type the request-state parameter of
 the retrieval hook

---
 litellm/proxy/policy_engine/response_retrieval.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/proxy/policy_engine/response_retrieval.py b/litellm/proxy/policy_engine/response_retrieval.py
index 54b9acf6150..0aa9579ca71 100644
--- a/litellm/proxy/policy_engine/response_retrieval.py
+++ b/litellm/proxy/policy_engine/response_retrieval.py
@@ -71,7 +71,7 @@ def _post_call_pipelines_for_context(context: PolicyMatchContext) -> tuple[Polic
 
 
 def attach_post_call_pipelines_to_retrieval(
-    data: dict,  # mutable-ok: the proxy's request-state dict, written in place like every other policy engine hook
+    data: dict[str, object],  # mutable-ok: request-state dict the policy engine hooks all write in place
     user_api_key_dict: "UserAPIKeyAuth",
     llm_router: "Router | None",
 ) -> None: