From b32062a77c83e8c23189b8e9606a79cf07404dbd Mon Sep 17 00:00:00 2001 From: Ilya Sibiryakov Date: Thu, 1 Oct 2026 00:02:34 +0300 Subject: [PATCH] fix(types): narrow metadata to dict to satisfy basedpyright reportIndexIssue gate --- litellm/proxy/guardrails/guardrail_hooks/ztds.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/ztds.py b/litellm/proxy/guardrails/guardrail_hooks/ztds.py index dcafc860e64..bf8f1da7bf8 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/ztds.py +++ b/litellm/proxy/guardrails/guardrail_hooks/ztds.py @@ -219,11 +219,13 @@ class ZTDSGuardrail(CustomGuardrail): ] # Attach ZTDS audit receipt to metadata - if "metadata" not in data or data["metadata"] is None: - data["metadata"] = {} - data["metadata"]["ztds_sanitized"] = True - data["metadata"]["ztds_standard"] = "RFC v1.0 (IETF draft-sibiryakov-ztds-protocol-02)" - data["metadata"]["ztds_invariants_verified"] = [1, 2, 3, 4] + metadata = data.get("metadata") + if not isinstance(metadata, dict): + metadata = {} + data["metadata"] = metadata + metadata["ztds_sanitized"] = True + metadata["ztds_standard"] = "RFC v1.0 (IETF draft-sibiryakov-ztds-protocol-02)" + metadata["ztds_invariants_verified"] = [1, 2, 3, 4] return data