Update frogboss_r2egym_parser.py
Browse files
frogboss_r2egym_parser.py
CHANGED
|
@@ -4,10 +4,10 @@ Custom tool parser for vLLM with R2E-gym XML format.
|
|
| 4 |
Same as frogboss_default_parser but handles XML format instead of JSON.
|
| 5 |
|
| 6 |
Usage:
|
| 7 |
-
vllm serve microsoft/FrogBoss-2510 \
|
| 8 |
--tensor-parallel-size 4 \
|
| 9 |
--enable-auto-tool-choice \
|
| 10 |
-
--tool-parser-plugin frogboss_r2egym_parser.py \
|
| 11 |
--tool-call-parser froggy \
|
| 12 |
--enable-log-requests \
|
| 13 |
--enable-log-outputs \
|
|
@@ -28,8 +28,8 @@ from vllm.entrypoints.openai.protocol import (
|
|
| 28 |
FunctionCall,
|
| 29 |
ToolCall,
|
| 30 |
)
|
| 31 |
-
from vllm.
|
| 32 |
-
from vllm.
|
| 33 |
ExtractedToolCallInformation,
|
| 34 |
)
|
| 35 |
from vllm.transformers_utils.tokenizer import AnyTokenizer
|
|
|
|
| 4 |
Same as frogboss_default_parser but handles XML format instead of JSON.
|
| 5 |
|
| 6 |
Usage:
|
| 7 |
+
vllm serve microsoft/FrogBoss-32B-2510 \
|
| 8 |
--tensor-parallel-size 4 \
|
| 9 |
--enable-auto-tool-choice \
|
| 10 |
+
--tool-parser-plugin ./Froggy-Training/src/vllm/frogboss_r2egym_parser.py \
|
| 11 |
--tool-call-parser froggy \
|
| 12 |
--enable-log-requests \
|
| 13 |
--enable-log-outputs \
|
|
|
|
| 28 |
FunctionCall,
|
| 29 |
ToolCall,
|
| 30 |
)
|
| 31 |
+
from vllm.tool_parsers import ToolParser, ToolParserManager
|
| 32 |
+
from vllm.tool_parsers.abstract_tool_parser import (
|
| 33 |
ExtractedToolCallInformation,
|
| 34 |
)
|
| 35 |
from vllm.transformers_utils.tokenizer import AnyTokenizer
|