xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-16 10:45:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Frontend] Improve error message in tool_choice validation (#19239)

Browse Source 
Signed-off-by: 22quinn <33176974+22quinn@users.noreply.github.com>
This commit is contained in:
22quinn 2025-06-11 23:13:00 -06:00 committed by GitHub
parent 3f6341bf7f
commit 7e3e74c97c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
vllm/entrypoints/openai/protocol.py
View File

@ -700,22 +700,26 @@ class ChatCompletionRequest(OpenAIBaseModel):
# ensure that if "tool_choice" is specified as an object,
# it matches a valid tool
correct_usage_message = 'Correct usage: `{"type": "function",' \
' "function": {"name": "my_function"}}`'
if isinstance(data["tool_choice"], dict):
valid_tool = False
specified_function = data["tool_choice"].get("function")
if not specified_function:
function = data["tool_choice"].get("function")
if not isinstance(function, dict):
raise ValueError(
"Expected field `function` in `tool_choice`."
" Correct usage: `{\"type\": \"function\","
" \"function\": {\"name\": \"my_function\"}}`")
specified_function_name = specified_function.get("name")
if not specified_function_name:
f"Invalid value for `function`: `{function}` in "
f"`tool_choice`! {correct_usage_message}")
if "name" not in function:
raise ValueError(f"Expected field `name` in `function` in "
f"`tool_choice`! {correct_usage_message}")
function_name = function["name"]
if not isinstance(function_name,
str) or len(function_name) == 0:
raise ValueError(
"Expected field `name` in `function` in `tool_choice`."
"Correct usage: `{\"type\": \"function\", "
"\"function\": {\"name\": \"my_function\"}}`")
f"Invalid `name` in `function`: `{function_name}`"
f" in `tool_choice`! {correct_usage_message}")
for tool in data["tools"]:
if tool["function"]["name"] == specified_function_name:
if tool["function"]["name"] == function_name:
valid_tool = True
break
if not valid_tool: