File tree Expand file tree Collapse file tree 2 files changed +8
-6
lines changed
Expand file tree Collapse file tree 2 files changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -179,10 +179,10 @@ def _encode_litellm_tool_call_id(
179179
180180 encoded_signature = base64 .b64encode (thought_signature ).decode ("utf-8" )
181181 return (
182- f"{ tool_call_id } { _LITELLM_THOUGHT_SIGNATURE_SEPARATOR } "
183- f"{ encoded_signature } "
182+ f"{ tool_call_id } { _LITELLM_THOUGHT_SIGNATURE_SEPARATOR } { encoded_signature } "
184183 )
185184
185+
186186_LITELLM_IMPORTED = False
187187_LITELLM_GLOBAL_SYMBOLS = (
188188 "ChatCompletionAssistantMessage" ,
Original file line number Diff line number Diff line change @@ -2261,10 +2261,12 @@ async def test_content_to_message_param_embeds_thought_signature_in_tool_call():
22612261 tool_calls = message ["tool_calls" ]
22622262 assert tool_calls is not None
22632263 assert len (tool_calls ) == 1
2264- assert (
2265- tool_calls [0 ]["id" ]
2266- == "test_tool_call_id__thought__"
2267- + base64 .b64encode (b"gemini_signature" ).decode ("utf-8" )
2264+ assert tool_calls [0 ][
2265+ "id"
2266+ ] == "test_tool_call_id__thought__" + base64 .b64encode (
2267+ b"gemini_signature"
2268+ ).decode (
2269+ "utf-8"
22682270 )
22692271
22702272
You can’t perform that action at this time.
0 commit comments