Skip to content

Commit 368b9ca

Browse files
committed
style(litellm): apply pyink formatting to thought signature changes
1 parent d73961e commit 368b9ca

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
@@ -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

0 commit comments

Comments
 (0)