Function parse_user_reply

agents/feedback-agent/main.py:234–305

解析用户回复，提取选中的论文编号 支持格式： - "1 2 4 6 7 9 11" （空格分隔） - "1-5 6 9 11" （范围 + 单个编号） - "1,2,4,6,7,9,11" （逗号分隔） - 混合格式："1-3 5 7-9 11" - 快捷命令："all lock"（所有必读）、"all red"（所有高度相关） Args: reply: 用户回复文本 papers: 论文列表（用于快捷命令） Returns: 选中的编号集合

(reply: str, papers: List[Dict] = None)

def parse_user_reply(reply: str, papers: List[Dict] = None) -> Set[int]:
    """
    解析用户回复，提取选中的论文编号

    支持格式：
    - "1 2 4 6 7 9 11" （空格分隔）
    - "1-5 6 9 11" （范围 + 单个编号）
    - "1,2,4,6,7,9,11" （逗号分隔）
    - 混合格式："1-3 5 7-9 11"
    - 快捷命令："all lock"（所有必读）、"all red"（所有高度相关）

    Args:
        reply: 用户回复文本
        papers: 论文列表（用于快捷命令）

    Returns:
        选中的编号集合
    """
    selected: Set[int] = set()
    max_paper_num = len(papers) if papers else 200

    # 快捷命令处理
    reply_lower = reply.lower().strip()

    if reply_lower == "all lock":
        # 选择必读清单论文
        if papers:
            for i, paper in enumerate(papers):
                if paper.get("category") == "must_read":
                    selected.add(i + 1)  # 编号从 1 开始
        return selected

    if reply_lower == "all red":
        # 选择所有高度相关论文
        if papers:
            for i, paper in enumerate(papers):
                if paper.get("category") == "high_relevant":
                    selected.add(i + 1)  # 编号从 1 开始
        return selected

    if reply_lower == "none":
        # 都不选
        return selected

    # 规范化：将逗号、顿号等替换为空格
    normalized = re.sub(r"[,，、;；]", " ", reply)

    # 提取所有 token（由空格分隔的部分）
    tokens = normalized.split()

    for token in tokens:
        token = token.strip()
        if not token:
            continue

        # 检查是否是范围格式（如 "1-5"）
        range_match = re.match(r"^(\d+)-(\d+)$", token)
        if range_match:
            start = int(range_match.group(1))
            end = int(range_match.group(2))