nat.finetuning.utils.parsers.llama_index_parser

Attributes

_extract_content
_parse_generic_message

Functions

parse_to_openai_message(→ dict)	Convert IntermediateStep to OpenAI-compatible message dictionary.
_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.
_parse_input_message(→ dict)	Parse user or system messages from LLM_START event.
_parse_dict_message(→ dict)	Parse a dictionary-based message.

Module Contents

_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_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.

_parse_input_message(

message: nat.data_models.intermediate_step.IntermediateStep,

) → dict

Parse user or system messages from LLM_START event.

_parse_dict_message(msg_dict: dict) → dict

Parse a dictionary-based message.