scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 131

fix gemini test

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-10 17:20:01 -07:00
parent 96fa9d46f5
commit 24474789d7
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/vertex_ai_and_google_ai_studio/vertex_llm_base.py
View file

@ -166,7 +166,7 @@ class VertexBase(BaseLLM):
def _get_token_and_url(
self,
model: str,
auth_header: str,
auth_header: Optional[str],
gemini_api_key: Optional[str],
vertex_project: Optional[str],
vertex_location: Optional[str],
@ -192,6 +192,7 @@ class VertexBase(BaseLLM):
stream=stream,
gemini_api_key=gemini_api_key,
)
auth_header = None # this field is not used for gemin
else:
vertex_location = self.get_vertex_region(vertex_region=vertex_location)