From b1b2d019cdd2fd2ffcc324dec7fb6d94ee551b9e Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 22 May 2026 14:30:13 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../core/inference/safetensors_agentic.py | 1 + .../tests/test_safetensors_tool_loop.py | 18 +++++++++--------- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/studio/backend/core/inference/safetensors_agentic.py b/studio/backend/core/inference/safetensors_agentic.py index edde2b0bda..a300e986b3 100644 --- a/studio/backend/core/inference/safetensors_agentic.py +++ b/studio/backend/core/inference/safetensors_agentic.py @@ -144,6 +144,7 @@ def run_safetensors_tool_loop( inject_respond_tool, is_respond_call, ) + tools, _respond_injected = inject_respond_tool(tools) tool_call_history: list[tuple[str, bool]] = [] diff --git a/studio/backend/tests/test_safetensors_tool_loop.py b/studio/backend/tests/test_safetensors_tool_loop.py index 692dfda786..59d67e1e9e 100644 --- a/studio/backend/tests/test_safetensors_tool_loop.py +++ b/studio/backend/tests/test_safetensors_tool_loop.py @@ -537,9 +537,7 @@ class TestRespondToolHelpers: assert out is base def test_is_respond_call(self): - assert is_respond_call( - {"function": {"name": "respond", "arguments": "{}"}} - ) + assert is_respond_call({"function": {"name": "respond", "arguments": "{}"}}) assert not is_respond_call( {"function": {"name": "web_search", "arguments": "{}"}} ) @@ -625,13 +623,15 @@ class TestRespondToolUnwrap: # When the caller supplies a real "respond" tool, the synthetic # one is NOT injected and the call goes through execute_tool # like any other tool, preserving the client's semantics. - turn_iter = iter([ + turn_iter = iter( [ - '{"name":"respond",' - '"arguments":{"message":"hi"}}' - ], - ["thanks"], - ]) + [ + '{"name":"respond",' + '"arguments":{"message":"hi"}}' + ], + ["thanks"], + ] + ) def _gen(_messages): try: