From 4868a6cf55aa62a88b7f9fb146f40ffac48e3681 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 26 Aug 2024 20:49:08 -0700 Subject: [PATCH] fix: fix unbound var --- .../vertex_and_google_ai_studio_gemini.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/llms/vertex_ai_and_google_ai_studio/vertex_and_google_ai_studio_gemini.py b/litellm/llms/vertex_ai_and_google_ai_studio/vertex_and_google_ai_studio_gemini.py index b5e32715b..0a7be19b9 100644 --- a/litellm/llms/vertex_ai_and_google_ai_studio/vertex_and_google_ai_studio_gemini.py +++ b/litellm/llms/vertex_ai_and_google_ai_studio/vertex_and_google_ai_studio_gemini.py @@ -1373,6 +1373,8 @@ class VertexLLM(BaseLLM): cached_content=optional_params.pop("cached_content", None), logging_obj=logging_obj, ) + else: # [TODO] implement context caching for gemini as well + cached_content = optional_params.pop("cached_content", None) # Separate system prompt from rest of message supports_system_message = get_supports_system_message(