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 @@ -2266,10 +2266,12 @@ async def test_content_to_message_param_embeds_thought_signature_in_tool_call():
22662266 tool_calls = message ["tool_calls" ]
22672267 assert tool_calls is not None
22682268 assert len (tool_calls ) == 1
2269- assert (
2270- tool_calls [0 ]["id" ]
2271- == "test_tool_call_id__thought__"
2272- + base64 .b64encode (b"gemini_signature" ).decode ("utf-8" )
2269+ assert tool_calls [0 ][
2270+ "id"
2271+ ] == "test_tool_call_id__thought__" + base64 .b64encode (
2272+ b"gemini_signature"
2273+ ).decode (
2274+ "utf-8"
22732275 )
22742276
22752277
You can’t perform that action at this time.
0 commit comments