beam_search.graph_setup

Classes

BeamSearchGraph(k)
NodeType(value)	An enumeration.

class BeamSearchGraph(k: int)

Bases: object

class SingleBeam

Bases: object

copy()

get_parent_node(node: str)

_inc_idx(inc: int = 1)

static _set_color(type: NodeType)

create_nodes_from_beams(nodes: Dict, beam: int, parent: str, k_highlighted: Iterable[str] | None = None)

nodes should be in the form: {

[node name]: (score, NodeType)

}

create_nodes_outside_beams(nodes: Dict, parent: str)

nodes should be in the form: {

[node name]: (score, NodeType)

}

class NodeType(value)

Bases: Enum

An enumeration.

DEFAULT_ACTION = 1

DROP_OFF = 6

GOAL = 5

INTENT = 4

MESSAGE_ACTION = 2

SYSTEM_API = 3