diff --git a/studio/backend/core/inference/llama_cpp.py b/studio/backend/core/inference/llama_cpp.py index eb548606b9..c2652a683b 100644 --- a/studio/backend/core/inference/llama_cpp.py +++ b/studio/backend/core/inference/llama_cpp.py @@ -5154,7 +5154,7 @@ class LlamaCppBackend: # artifact check. _visible_raw = content_accum.strip() _visible = ( - _strip_tool_markup(content_accum, final=True).strip() + _strip_tool_markup(content_accum, final = True).strip() if _visible_raw else "" ) diff --git a/studio/backend/tests/test_llama_cpp_reprompt_guard.py b/studio/backend/tests/test_llama_cpp_reprompt_guard.py index 8a6087acb0..b0a64bf0bc 100644 --- a/studio/backend/tests/test_llama_cpp_reprompt_guard.py +++ b/studio/backend/tests/test_llama_cpp_reprompt_guard.py @@ -996,11 +996,7 @@ def test_reprompts_on_direct_first_person_read_check_open_plan(): "1. Identify the columns.\n" "2. Return the total." ), - ( - "I will check the docs:\n" - "1. Gather relevant sections.\n" - "2. Answer." - ), + ("I will check the docs:\n" "1. Gather relevant sections.\n" "2. Answer."), ( "First, I'll review the repository:\n" "1. Open the relevant file.\n"