File tree Expand file tree Collapse file tree 2 files changed +7
-4
lines changed
Expand file tree Collapse file tree 2 files changed +7
-4
lines changed Original file line number Diff line number Diff line change 149149# https://ai.google.dev/gemini-api/docs/thought-signatures
150150_THOUGHT_SIGNATURE_SEPARATOR = "__thought__"
151151
152+
152153_LITELLM_IMPORTED = False
153154_LITELLM_GLOBAL_SYMBOLS = (
154155 "ChatCompletionAssistantMessage" ,
Original file line number Diff line number Diff line change @@ -2269,10 +2269,12 @@ async def test_content_to_message_param_embeds_thought_signature_in_tool_call():
22692269 tool_calls = message ["tool_calls" ]
22702270 assert tool_calls is not None
22712271 assert len (tool_calls ) == 1
2272- assert (
2273- tool_calls [0 ]["id" ]
2274- == "test_tool_call_id__thought__"
2275- + base64 .b64encode (b"gemini_signature" ).decode ("utf-8" )
2272+ assert tool_calls [0 ][
2273+ "id"
2274+ ] == "test_tool_call_id__thought__" + base64 .b64encode (
2275+ b"gemini_signature"
2276+ ).decode (
2277+ "utf-8"
22762278 )
22772279
22782280
You can’t perform that action at this time.
0 commit comments