ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-21 23:18:45 +00:00
commit 18673a562a
682 changed files with 25903 additions and 8390 deletions
Download patch file Download diff file Expand all files Collapse all files

47
studio/backend/tests/test_tool_loop_controller.py
View file

@ -23,7 +23,10 @@ from core.inference.tool_loop_controller import (
def test_append_deferred_nudges_merges_deduped_into_one_message():
conversation = [{"role": "assistant", "tool_calls": [1]}, {"role": "tool", "content": "r"}]
conversation = [
{"role": "assistant", "tool_calls": [1]},
{"role": "tool", "content": "r"},
]
nudges = [
{"role": "user", "content": "duplicate"},
{"role": "user", "content": "duplicate"}, # dropped: same content
@ -31,7 +34,9 @@ def test_append_deferred_nudges_merges_deduped_into_one_message():
]
append_deferred_nudges(conversation, nudges)
# One user message, after the results, with distinct contents joined.
assert conversation[2:] == [{"role": "user", "content": "duplicate\n\ndisabled foo"}]
assert conversation[2:] == [
{"role": "user", "content": "duplicate\n\ndisabled foo"}
]
# Empty is a no-op.
before = list(conversation)
append_deferred_nudges(conversation, [])
@ -86,7 +91,10 @@ def test_status_and_provenance_match_local_event_conventions():
status_for_tool("web_search", {"url": "https://www.example.com/a"})
== "Reading: example.com"
)
assert status_for_tool("python", {"code": "print(1)\nprint(2)"}) == "Running Python: print(1)"
assert (
status_for_tool("python", {"code": "print(1)\nprint(2)"})
== "Running Python: print(1)"
)
assert tool_event_provenance(healed = True, forced = False, provisional = None) == {
"source": "local",
"healed": True,
@ -102,7 +110,10 @@ def test_prepare_execute_builds_visible_events_and_model_tool_message():
assert decision.status_text == "Searching: gpu prices"
assert decision.tool_start_payload()["arguments"] == {"query": "gpu prices"}
assert decision.tool_start_event()["type"] == "tool_start"
assert decision.as_assistant_tool_call()["function"]["arguments"] == '{"query":"gpu prices"}'
assert (
decision.as_assistant_tool_call()["function"]["arguments"]
== '{"query":"gpu prices"}'
)
completion = controller.record_result(decision, "Search result\n__IMAGES__:{...}")
@ -118,10 +129,14 @@ def test_prepare_execute_builds_visible_events_and_model_tool_message():
def test_successful_duplicate_is_internal_noop_and_keeps_remaining_tools():
controller = ToolLoopController(tools = [_tool("web_search"), _tool("python")])
first = controller.prepare_call(_call("web_search", {"query": "gpu prices"}, "call_a"))
first = controller.prepare_call(
_call("web_search", {"query": "gpu prices"}, "call_a")
)
controller.record_result(first, "ok")
duplicate = controller.prepare_call(_call("web_search", {"query": "gpu prices"}, "call_b"))
duplicate = controller.prepare_call(
_call("web_search", {"query": "gpu prices"}, "call_b")
)
completion = controller.record_noop(duplicate)
assert duplicate.action == "duplicate"
@ -144,10 +159,14 @@ def test_successful_duplicate_is_internal_noop_and_keeps_remaining_tools():
def test_repeated_successful_duplicate_becomes_terminal_after_one_recovery_nudge():
controller = ToolLoopController(tools = [_tool("web_search"), _tool("python")])
first = controller.prepare_call(_call("web_search", {"query": "gpu prices"}, "call_a"))
first = controller.prepare_call(
_call("web_search", {"query": "gpu prices"}, "call_a")
)
controller.record_result(first, "ok")
duplicate_one = controller.prepare_call(_call("web_search", {"query": "gpu prices"}, "call_b"))
duplicate_one = controller.prepare_call(
_call("web_search", {"query": "gpu prices"}, "call_b")
)
completion_one = controller.record_noop(duplicate_one)
assert duplicate_one.action == "duplicate"
@ -158,7 +177,9 @@ def test_repeated_successful_duplicate_becomes_terminal_after_one_recovery_nudge
"python",
]
duplicate_two = controller.prepare_call(_call("web_search", {"query": "gpu prices"}, "call_c"))
duplicate_two = controller.prepare_call(
_call("web_search", {"query": "gpu prices"}, "call_c")
)
completion_two = controller.record_noop(duplicate_two)
assert duplicate_two.action == "duplicate"
@ -216,12 +237,16 @@ def test_render_html_success_filters_active_tools_and_repeat_is_internal():
"web_search",
]
first = controller.prepare_call(_call("render_html", {"code": "<html></html>"}, "call_html_1"))
first = controller.prepare_call(
_call("render_html", {"code": "<html></html>"}, "call_html_1")
)
controller.record_result(first, "Rendered HTML canvas: Demo")
assert [t["function"]["name"] for t in controller.active_tools()] == ["web_search"]
repeat = controller.prepare_call(_call("render_html", {"code": "<html></html>"}, "call_html_2"))
repeat = controller.prepare_call(
_call("render_html", {"code": "<html></html>"}, "call_html_2")
)
completion = controller.record_noop(repeat)
assert repeat.action == "render_html_repeat"