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__`.
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
"""
|
|
Longcat Flash Chat tool call parser.
|
|
|
|
Same as Hermes but uses <longcat_tool_call> tags instead of <tool_call>.
|
|
Based on VLLM's LongcatFlashToolParser (extends Hermes2ProToolParser).
|
|
"""
|
|
|
|
import json
|
|
import re
|
|
import uuid
|
|
from typing import List, Optional
|
|
|
|
from openai.types.chat.chat_completion_message_tool_call import (
|
|
ChatCompletionMessageToolCall,
|
|
Function,
|
|
)
|
|
|
|
from environments.tool_call_parsers import ParseResult, ToolCallParser, register_parser
|
|
|
|
|
|
@register_parser("longcat")
|
|
class LongcatToolCallParser(ToolCallParser):
|
|
"""
|
|
Parser for Longcat Flash Chat tool calls.
|
|
Identical logic to Hermes, just different tag names.
|
|
"""
|
|
|
|
PATTERN = re.compile(
|
|
r"<longcat_tool_call>\s*(.*?)\s*</longcat_tool_call>|<longcat_tool_call>\s*(.*)",
|
|
re.DOTALL,
|
|
)
|
|
|
|
def parse(self, text: str) -> ParseResult:
|
|
if "<longcat_tool_call>" not in text:
|
|
return text, None
|
|
|
|
try:
|
|
matches = self.PATTERN.findall(text)
|
|
if not matches:
|
|
return text, None
|
|
|
|
tool_calls: List[ChatCompletionMessageToolCall] = []
|
|
for match in matches:
|
|
raw_json = match[0] if match[0] else match[1]
|
|
if not raw_json.strip():
|
|
continue
|
|
|
|
tc_data = json.loads(raw_json)
|
|
tool_calls.append(
|
|
ChatCompletionMessageToolCall(
|
|
id=f"call_{uuid.uuid4().hex[:8]}",
|
|
type="function",
|
|
function=Function(
|
|
name=tc_data["name"],
|
|
arguments=json.dumps(
|
|
tc_data.get("arguments", {}), ensure_ascii=False
|
|
),
|
|
),
|
|
)
|
|
)
|
|
|
|
if not tool_calls:
|
|
return text, None
|
|
|
|
content = text[: text.find("<longcat_tool_call>")].strip()
|
|
return content if content else None, tool_calls
|
|
|
|
except Exception:
|
|
return text, None
|