scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix(factory.py): fix linting

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-04 08:21:19 -07:00
parent 3184640291
commit d4577fe476
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/llms/prompt_templates/factory.py
View file

@ -531,7 +531,6 @@ def format_prompt_togetherai(messages, prompt_format, chat_template):
### IBM Granite
def ibm_granite_pt(messages: list):
"""
IBM's Granite models uses the template:
@ -1026,13 +1025,13 @@ def convert_to_gemini_tool_call_invoke(
def convert_to_gemini_tool_call_result(
message: dict,
last_message_with_tool_calls: dict|None,
last_message_with_tool_calls: Optional[dict],
) -> litellm.types.llms.vertex_ai.PartType:
"""
OpenAI message with a tool result looks like:
{
"tool_call_id": "tool_1",
"role": "tool",
"role": "tool",
"content": "function result goes here",
},
@ -1053,7 +1052,11 @@ def convert_to_gemini_tool_call_result(
msg_tool_call_id = message.get("tool_call_id", None)
for tool in tools:
prev_tool_call_id = tool.get("id", None)
if msg_tool_call_id and prev_tool_call_id and msg_tool_call_id == prev_tool_call_id:
if (
msg_tool_call_id
and prev_tool_call_id
and msg_tool_call_id == prev_tool_call_id
):
name = tool.get("function", {}).get("name", "")
if not name: