diff --git a/tests/tool_use/test_seed_oss_tool_parser.py b/tests/tool_use/test_seed_oss_tool_parser.py index d85bc9bbf1b30..c276a598aa68c 100644 --- a/tests/tool_use/test_seed_oss_tool_parser.py +++ b/tests/tool_use/test_seed_oss_tool_parser.py @@ -102,9 +102,7 @@ def test_extract_tool_calls_no_tools(seed_oss_tool_parser): ], argnames=["model_output", "expected_tool_calls", "expected_content"], argvalues=[ - ("""\n\n\n""" - """The current thinking budget is 0, so I will directly start answering the question.\n\n""" - """\n\n""" + ("""\n\n""" """Barcelona, Spain\n\n""", [ ToolCall(function=FunctionCall( @@ -114,10 +112,7 @@ def test_extract_tool_calls_no_tools(seed_oss_tool_parser): }, ), ), type='function') - ], - """\n\n\n""" - """The current thinking budget is 0, so I will directly start answering the question.\n\n""" - ), + ], None), ( """The user\'s current thinking budget is 512.\nLet me analyze the """ """question. The user wants to know the weather in Barcelona, Spain. Looking at the functions available, """ diff --git a/vllm/entrypoints/openai/tool_parsers/seed_oss_tool_parser.py b/vllm/entrypoints/openai/tool_parsers/seed_oss_tool_parser.py index 69cf2e68f7c41..95458f07ff2a2 100644 --- a/vllm/entrypoints/openai/tool_parsers/seed_oss_tool_parser.py +++ b/vllm/entrypoints/openai/tool_parsers/seed_oss_tool_parser.py @@ -271,6 +271,9 @@ class SeedOssToolParser(ToolParser): # Extract content after think end token result_content = model_output[think_end_index:] thinking_content = model_output[:think_end_index] + else: + thinking_content = "" + result_content = model_output try: function_calls = self._get_function_calls(result_content)