Skip to content

Commit fb59487

Browse files
committed
style(litellm): apply pyink formatting to thought signature changes
1 parent 4c0cc66 commit fb59487

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

src/google/adk/models/lite_llm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,7 @@
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",

tests/unittests/models/test_litellm.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)