phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(vertex_httpx.py): fix supported vertex params

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-04 21:17:35 -07:00
parent 6d2b429176
commit d0862697b8
4 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/vertex_ai.py
View file

@ -155,6 +155,7 @@ class VertexAIConfig:
"response_format",
"n",
"stop",
"extra_headers",
]
def map_openai_params(self, non_default_params: dict, optional_params: dict):
@ -400,7 +401,9 @@ def _gemini_convert_messages_with_history(messages: list) -> List[ContentType]:
## APPEND TOOL CALL MESSAGES ##
if msg_i < len(messages) and messages[msg_i]["role"] == "tool":
_part = convert_to_gemini_tool_call_result(messages[msg_i], last_message_with_tool_calls)
_part = convert_to_gemini_tool_call_result(
messages[msg_i], last_message_with_tool_calls
)
contents.append(ContentType(parts=[_part])) # type: ignore
msg_i += 1
if msg_i == init_msg_i: # prevent infinite loops