Dump the database to a file.

Load the database from a structured file like JSON, YAML, or TOML.

Load a function from a module. :param path: The path to the function, e.g. "module.submodule.function". :return: The function object.

Get the Gloo group for distributed communication.

Check if the message is a tool call.

Reset the default value of a Megatron argument. :param parser: The argument parser. :param name: The name of the argument to reset. :

Wrap communication functions with memory check.

Returns a string representation of the phone's status bar with basic indicators. Shows network signal, data connection type, wifi sta

Insert a string and its corresponding token IDs, log probabilities, and loss mask into the trie. Args: text: String to in

Register a new Domain implementation

Register a new Domain implementation

Log data to tensorboard Args: data (dict): Dictionary containing metric names and values step (int): Current step/epo

Accept standard chat messages and return a GenerationOutput. If sampling_params is provided, it overrides the defaults set at initial

Simulates the outcome of a cellular network search based on SIM status. This function can be used to update the network_connection_st

Retrieves a bill directly by its unique ID. Args: bill_id: The unique identifier of the bill. Returns:

Get the initial state of the user simulator. Args: message_history: The message history of the conversation. Returns:

Get a game from the database.

Retrieves a line using the customer ID and line ID. Args: customer_id: The unique identifier of the customer.

Get the tools available in the ToolKit. Uses the `as_tool` to convert the functions to Tool objects. Returns: A dictionar

Normalize a final answer to a quantitative reasoning question. Args: final_answer: The answer string to normalize Returns: N

Resolve per-group defaults from model-level then args-level values.

Run the FastAPI server. Args: host: The host to bind to port: The port to bind to

Retrieves a line directly by its phone number. Args: phone_number: The phone number to search for. Returns:

Make a POST request to the specified endpoint with the given payload. Args: endpoint: The API endpoint to call payloa

Calculate the reward for the simulation.

Finds a customer by their primary contact or line phone number. Args: phone_number: The phone number to search for.

Get a registered Domain by name

Remove a string and all nodes with this text as prefix from the trie. Args: text: String to remove (will also remove all

Can be used either as a decorator or a context manager: @timer def func(): ... or with timer("block_name"): ..

Find a booking by ID across all travelers. Returns: (Traveler owning the booking, Booking)

Get the order from the database. Args: order_id: The order id, such as '#W0000000'. Be careful there is a '#' symbol at the begin

Get a service ticket from the database.

Gather tensors across all ranks in the context parallel group. The first dimension of the output tensor will be the `response_length`.

Simple mode: send a text query to the expert. Args: query: The question/context to send expert:

Generate a response from the model. Args: model: The model to use. messages: The messages to send to the model. tool

Retrieves a customer directly by their unique ID. Args: customer_id: The unique identifier of the customer. Ret

Check if the message is a valid agent history message.

Extract the last LaTeX boxed expression from a string. Args: string: Input string containing LaTeX code Returns: The last bo

Remove the LaTeX boxed command from a string. Args: s: String with format "\\boxed{content}" Returns: The content inside the

Get the payment method from the database. Args: payment_method_id: The payment method id, such as 'gift_card_0000000' or 'credit_

Get reservation from database.

Get a sale from the database.

Find a show by ID in a given theater, optionally constrained by date. Returns (show, date)

Do one rollout training

Create a new task for a user. Args: user_id: The ID of the user creating the task title: The title of the ta

Retrieve documents relevant to the query and return a formatted string. Args: query: The retrieval query text R