nat.finetuning.utils.parsers.adk_parser#

Attributes#

logger

_extract_content

_parse_generic_message

Functions#

parse_to_openai_message(→ dict)

Convert IntermediateStep to OpenAI-compatible message dictionary.

_parse_input_message(→ dict | list[dict])

Parse user or system messages from LLM_START event.

_parse_assistant_message(→ dict)

Parse an assistant message from LLM_END event.

_parse_tool_message(→ dict)

Parse a tool/function response message from TOOL_END event.

Module Contents#

logger#
_extract_content#
_parse_generic_message#
parse_to_openai_message(
message: nat.data_models.intermediate_step.IntermediateStep,
) → dict#

Convert IntermediateStep to OpenAI-compatible message dictionary.

Args:

message: An IntermediateStep object representing a single message. previous_message: Previous message for context (reserved for future).

Returns:

A dictionary formatted for OpenAI API consumption.

_parse_input_message(
message: nat.data_models.intermediate_step.IntermediateStep,
) → dict | list[dict]#

Parse user or system messages from LLM_START event.

_parse_assistant_message(
message: nat.data_models.intermediate_step.IntermediateStep,
) → dict#

Parse an assistant message from LLM_END event.

_parse_tool_message(
message: nat.data_models.intermediate_step.IntermediateStep,
) → dict#

Parse a tool/function response message from TOOL_END event.