20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Misc] Replace print with logger (#15923)

Browse Source 
Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
This commit is contained in:
Chauncey 2025-04-02 16:37:38 +08:00 committed by GitHub
parent aa557e6422
commit cdb57015a7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vllm/entrypoints/openai/tool_parsers/phi4mini_tool_parser.py
View File

@ -47,13 +47,13 @@ class Phi4MiniJsonToolParser(ToolParser):
""" """
Extract the tool calls from a complete model response. Extract the tool calls from a complete model response.
""" """
print(f"Model output: {model_output}") logger.debug("Model output: %s", model_output)
pattern = r'functools\[(.*?)\]' pattern = r'functools\[(.*?)\]'
matches = re.search(pattern, model_output, re.DOTALL) matches = re.search(pattern, model_output, re.DOTALL)
if not matches: if not matches:
print("No function calls found") logger.debug("No function calls found")
return ExtractedToolCallInformation(tools_called=False, return ExtractedToolCallInformation(tools_called=False,
tool_calls=[], tool_calls=[],
content=model_output) content=model_output)
@ -64,10 +64,12 @@ class Phi4MiniJsonToolParser(ToolParser):
json_content = '[' + matches.group(1) + ']' json_content = '[' + matches.group(1) + ']'
function_call_arr = json.loads(json_content) function_call_arr = json.loads(json_content)
print(f"Successfully extracted {len(function_call_arr)} " logger.debug("Successfully extracted %d function calls",
"function calls") len(function_call_arr))
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
print(f"Error parsing JSON: {e}") logger.error(
"Failed to parse function calls from model output: %s. "
"Error: %s", model_output, str(e))
tool_calls: list[ToolCall] = [ tool_calls: list[ToolCall] = [
ToolCall( ToolCall(