Fix linter errors

Signed-off-by: xiyuan lee <[email protected]>
vllm-project · Nov 19, 2024 · fec6efb · fec6efb
1 parent ef7173d
commit fec6efb
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/vllm/entrypoints/openai/tool_parsers/hermes_tool_parser.py b/vllm/entrypoints/openai/tool_parsers/hermes_tool_parser.py
@@ -187,9 +187,12 @@ def extract_tool_calls_streaming(
                   and cur_tool_end_count > prev_tool_end_count):
                 diff = self.prev_tool_call_arr[self.current_tool_id].get(
                     "arguments")
-                if diff:
-                    diff = diff.encode('utf-8').decode('unicode_escape') if diff is str else diff
-                    diff = json.dumps(diff, ensure_ascii=False)[len(self.streamed_args_for_tool[self.current_tool_id]):]
+                if diff:                    
+                     diff = diff.encode('utf-8').decode(
+                         'unicode_escape') if diff is str else diff
+                     diff = json.dumps(
+                         diff, ensure_ascii=False
+                     )[len(self.streamed_args_for_tool[self.current_tool_id]):]
                     logger.debug(
                         "Finishing tool and found diff that had not "
                         "been streamed yet: %s", diff)
@@ -305,7 +308,8 @@ def extract_tool_calls_streaming(
 
             # last case -- we have an update to existing arguments.
             elif cur_arguments and prev_arguments:
-                if isinstance(delta_text, str) and len(delta_text.rstrip())>=1 and delta_text.rstrip()[-1] == '}':
+                if isinstance(delta_text, str) and len(delta_text.rstrip(
+                )) >= 1 and delta_text.rstrip()[-1] == '}':
                     delta_text = delta_text.rstrip()[:-1]
 
                 logger.debug("got diff %s", delta_text)