phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fix unbound var

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-26 20:49:08 -07:00
parent 0eea01dae9
commit 4868a6cf55
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/vertex_ai_and_google_ai_studio/vertex_and_google_ai_studio_gemini.py
View file

@ -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(