File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed
vllm/entrypoints/openai/tool_parsers Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -28,6 +28,7 @@ lark == 1.2.2
28
28
xgrammar == 0.1.19; platform_machine == "x86_64" or platform_machine == "aarch64" or platform_machine == "arm64"
29
29
typing_extensions >= 4.10
30
30
filelock >= 3.16.1 # need to contain https://github.com/tox-dev/filelock/pull/317
31
+ json_repair # used for repairing JSON outputs
31
32
partial-json-parser # used for parsing partial JSON outputs
32
33
pyzmq >= 25.0.0
33
34
msgspec
Original file line number Diff line number Diff line change 7
7
8
8
import partial_json_parser
9
9
import regex as re
10
- from partial_json_parser .core .options import Allow
11
10
from json_repair import repair_json
12
11
13
12
from vllm .entrypoints .chat_utils import random_tool_call_id
@@ -171,6 +170,7 @@ def extract_tool_calls_streaming(
171
170
# something with tools with this diff.
172
171
# flags for partial JSON parting. exported constants from
173
172
# "Allow" are handled via BIT MASK
173
+ from partial_json_parser .core .options import Allow
174
174
flags = Allow .ALL if self .current_tool_name_sent \
175
175
else Allow .ALL & ~ Allow .STR
176
176
You can’t perform that action at this time.
0 commit comments