fix: fix unbound var

This commit is contained in:
Krrish Dholakia 2024-08-26 20:49:08 -07:00
parent 5d68f27a57
commit 592d8e933d

View file

@ -1315,6 +1315,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(