07b615e96e
- Added new environments for reinforcement learning, including `HermesSweEnv` for software engineering tasks and `TerminalTestEnv` for inline testing. - Introduced `ToolContext` for unrestricted access to tools during reward computation. - Updated `.gitignore` to exclude `wandb/` directory. - Enhanced `README.md` with detailed architecture and usage instructions for Atropos environments. - Added configuration files for SWE and terminal test environments to streamline setup. - Removed unnecessary compiled Python files from `__pycache__`.
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""
|
|
GLM 4.7 tool call parser.
|
|
|
|
Same as GLM 4.5 but with slightly different regex patterns.
|
|
The tool_call tags may wrap differently and arg parsing handles
|
|
newlines between key/value pairs.
|
|
|
|
Based on VLLM's Glm47MoeModelToolParser (extends Glm4MoeModelToolParser).
|
|
"""
|
|
|
|
import re
|
|
|
|
from environments.tool_call_parsers import ParseResult, register_parser
|
|
from environments.tool_call_parsers.glm45_parser import Glm45ToolCallParser
|
|
|
|
|
|
@register_parser("glm47")
|
|
class Glm47ToolCallParser(Glm45ToolCallParser):
|
|
"""
|
|
Parser for GLM 4.7 tool calls.
|
|
Extends GLM 4.5 with updated regex patterns.
|
|
"""
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
# GLM 4.7 uses a slightly different detail regex that includes
|
|
# the <tool_call> wrapper and optional arg_key content
|
|
self.FUNC_DETAIL_REGEX = re.compile(
|
|
r"<tool_call>(.*?)(<arg_key>.*?)?</tool_call>", re.DOTALL
|
|
)
|
|
# GLM 4.7 handles newlines between arg_key and arg_value tags
|
|
self.FUNC_ARG_REGEX = re.compile(
|
|
r"<arg_key>(.*?)</arg_key>(?:\\n|\s)*<arg_value>(.*?)</arg_value>",
|
|
re.DOTALL,
|
|
)
|