CharlesYang030
244e346b83
- Skill optimization framework with training loop analogy - 11 benchmarks, 4 model backends (Azure OpenAI, Claude, Codex, Qwen) - WebUI for browser-based training control - Pluggable architecture for extending benchmarks and backends
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
# Vendored from SkillRL (Apache-2.0 License)
|
|
# Original: agent_system/environments/env_package/alfworld/projection.py
|
|
|
|
from typing import List
|
|
import re
|
|
|
|
|
|
def alfworld_projection(actions: List[str], action_pools: List[List[str]]):
|
|
"""Process raw model outputs into valid ALFWorld actions.
|
|
|
|
Extracts text from ``<action>...</action>`` tags and validates that
|
|
the response also contains ``<think>...</think>`` tags.
|
|
|
|
Parameters
|
|
----------
|
|
actions : list[str]
|
|
Raw model outputs, one per environment.
|
|
action_pools : list[list[str]]
|
|
Admissible action lists per environment (unused but kept for API compat).
|
|
|
|
Returns
|
|
-------
|
|
actions : list[str]
|
|
Cleaned action strings.
|
|
valids : list[int]
|
|
1 if the action was successfully parsed, 0 otherwise.
|
|
"""
|
|
valids = [0] * len(actions)
|
|
|
|
for i in range(len(actions)):
|
|
original_str = actions[i]
|
|
actions[i] = actions[i].lower()
|
|
|
|
start_tag = "<action>"
|
|
end_tag = "</action>"
|
|
start_idx = actions[i].find(start_tag)
|
|
end_idx = actions[i].find(end_tag)
|
|
try:
|
|
if start_idx == -1 or end_idx == -1:
|
|
actions[i] = actions[i][-30:]
|
|
continue
|
|
|
|
extracted_action = actions[i][start_idx + len(start_tag):end_idx].strip().lower()
|
|
actions[i] = extracted_action
|
|
valids[i] = 1
|
|
|
|
except Exception:
|
|
actions[i] = actions[i][-30:]
|
|
|
|
# Require <think>...</think>
|
|
think_start_idx = original_str.find("<think>")
|
|
think_end_idx = original_str.find("</think>")
|
|
if think_start_idx == -1 or think_end_idx == -1:
|
|
valids[i] = 0
|
|
|
|
# Reject responses containing Chinese characters
|
|
if re.search(r'[\u4e00-\u9fff]', original_str):
|
|
valids[i] = 0
|
|
|
|
return actions, valids
|