From ea325c7b3c4503a62e4f2d6d44e4f6984e522e6d Mon Sep 17 00:00:00 2001 From: Nilesh Patil <128893479+nileshpatil6@users.noreply.github.com> Date: Wed, 17 Jun 2026 14:40:20 -0700 Subject: [PATCH] fix(LiteLlm): recognize assistant- prefix as valid OpenAI file ID Merge https://github.com/google/adk-python/pull/5758 Original PR by @nileshpatil6 (Nilesh Patil <128893479+nileshpatil6@users.noreply.github.com>) Azure OpenAI files receive IDs with an `assistant-` prefix (e.g. `assistant-abc123`) or `file-` prefix. When `_looks_like_openai_file_id` recognizes an `assistant-` prefixed ID (e.g., `assistant-abc123`) or `file-` prefixed ID, `_redact_file_uri_for_log` defaulted to logging both as `file - `. Fix: refactor `_redact_file_uri_for_log` to dynamically extract file ID prefixes (`file-`, `assistant-`). This ensures compatibility with both current prefixes and future OpenAI/Azure file ID types. Fixes #5664 Co-authored-by: Yi Liu PiperOrigin-RevId: 933931477 --- src/google/adk/models/lite_llm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/google/adk/models/lite_llm.py b/src/google/adk/models/lite_llm.py index db4175fbf8..8ca197f1fb 100644 --- a/src/google/adk/models/lite_llm.py +++ b/src/google/adk/models/lite_llm.py @@ -398,7 +398,8 @@ def _redact_file_uri_for_log( if file_uri.startswith("assistant-"): return "assistant-" if _looks_like_openai_file_id(file_uri): - return "file-" + prefix = file_uri.split("-", 1)[0] + return f"{prefix}-" try: parsed = urlparse(file_uri) except ValueError: