feat(func_call): Add more check in BaseFormatDetector.parse_streaming_increment (#7479)
This commit is contained in:
@@ -142,6 +142,10 @@ class BaseFormatDetector(ABC):
|
||||
try:
|
||||
if current_text.startswith(self.bot_token):
|
||||
start_idx = len(self.bot_token)
|
||||
elif self.current_tool_id > 0 and current_text.startswith(
|
||||
self.tool_call_separator + self.bot_token
|
||||
):
|
||||
start_idx = len(self.tool_call_separator + self.bot_token)
|
||||
elif self.current_tool_id > 0 and current_text.startswith(
|
||||
self.tool_call_separator
|
||||
):
|
||||
|
||||
@@ -468,10 +468,7 @@ class TestToolChoiceQwen25(TestToolChoiceLlama32):
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
cls.flaky_tests = {
|
||||
"test_multi_tool_scenario_auto",
|
||||
"test_multi_tool_scenario_required",
|
||||
}
|
||||
cls.flaky_tests = {}
|
||||
|
||||
cls.model = "Qwen/Qwen2.5-7B-Instruct"
|
||||
cls.base_url = DEFAULT_URL_FOR_TEST
|
||||
|
||||
Reference in New Issue
Block a user