Function vuln_select_node

workflow/nodes.py:114–150

(
    state: WorkflowState,
    *,
    context: WorkflowContext,
    sender_name: str = "Vuln_select",
)

def vuln_select_node(
    state: WorkflowState,
    *,
    context: WorkflowContext,
    sender_name: str = "Vuln_select",
) -> dict[str, Any]:
    check_count = state["check_count"]

    if check_count == 0:
        scan_res = message_content(state["message"][-1])
        vulns = parse_vulnerabilities(scan_res)
        if vulns:
            selected = vulns[0]
            vuln_select_message = (
                "I think we can try this vulnerability. "
                f"The vulnerability information is as follows {selected}"
            )
        else:
            vuln_select_message = "continue to select vulnerability"
    else:
        vulns = state["vulns"]
        if len(vulns) > 1:
            vulns.pop(0)
        selected = vulns[0] if vulns else {}
        vuln_select_message = (
            "I think we can try this vulnerability. "
            f"The vulnerability information is as follows {selected}"
        )

    context.history.append(f"{sender_name}{check_count}_{vuln_select_message}")

    return {
        "message": [_human_message(vuln_select_message)],
        "sender": sender_name,
        "vulns": vulns if check_count == 0 else state["vulns"],
        "check_count": check_count,
    }


def check_node(