base_handler

GalileoBaseHandler

Callback handler for logging traces to the Galileo platform.

Arguments

• _galileo_logger (GalileoLogger): The Galileo logger instance.

• _nodes (dict[UUID, Node]): A dictionary of nodes, where the key is the run_id and the value is the node.

• _start_new_trace (bool): Whether to start a new trace when a chain starts. Set this to False to continue using the current trace.

• _flush_on_chain_end (bool): Whether to flush the trace when a chain ends.

• _root_node (Optional[Node]): The root node of the trace, if any.

• _nodes (dict[str, Node]): A dictionary of nodes, where the key is the run_id as a string and the value is the Node object.

• _integration (INTEGRATION): The integration type, e.g., “langchain”. This is used to identify the source of the trace.

commit

def commit(self) -> None

Commit the nodes to the trace using the Galileo Logger. Optionally flush the trace.

end_node

def end_node(self, run_id: UUID, **kwargs: Any) -> None

End a node in the chain. Commit the nodes to a trace if the run_id matches the root node.

Arguments

• run_id (UUID): The run ID.
• **kwargs (Any): Additional parameters to update the span with.

get_node

def get_node(self, run_id: UUID) -> Node | None

Get a node by its run ID.

Arguments

• run_id (UUID): The run ID of the node to retrieve.

Returns

• Optional[Node]: The node if found, otherwise None.

get_nodes

def get_nodes(self) -> dict[str, Node]

Get all nodes.

Returns

• dict[str, Node]: A dictionary of all nodes.

log_node_tree

def log_node_tree(self, node: Node) -> None

Log a node and its children recursively.

Arguments

• node (Node): The node to log.

start_node

def start_node(self,
               node_type: NODE_TYPE,
               parent_run_id: UUID | None,
               run_id: UUID,
               **kwargs: Any) -> Node

Start a new node in the chain.

Arguments

• node_type (NODE_TYPE): The type of node.
• parent_run_id (Optional[UUID]): The parent run ID.
• run_id (UUID): The run ID.
• **kwargs (Any): Additional parameters for the span.

Returns

• Node: The created node.