Tachi67's picture
add docs and comments
cce4677 verified

Table of Contents

run

CodeGeneratorAtomicFlow

CodeGeneratorAtomicFlow Objects

class CodeGeneratorAtomicFlow(ChatAtomicFlow)

This class wraps around the Chat API to generate code from a goal. One thing worth noting is that we need to make sure the code generator does not write repetitive code that is present in the library, so we need to inject the function signatures in the library to the system prompts.

Input Interface Non Initialized:

  • goal
  • code_library
  • memory_files

Input Interface Initialized:

  • goal
  • code_library
  • memory_files

Output Interface:

  • code
  • language_of_code

Configuration Parameters:

  • Also refer to ChatAtomicFlow (https://huggingface.co/aiflows/ChatFlowModule/blob/main/ChatAtomicFlow.py)
  • input_interface_non_initialized: The input interface when the conversation is not initialized.
  • input_interface_initialized: The input interface when the conversation is initialized.
  • output_interface: The output interface.
  • backend: The backend to use for the Chat API.
  • system_message_prompt_template: The template for the system message prompt.
  • human_message_prompt_template: The template for the human message prompt.
  • init_human_message_prompt_template: The initial human message prompt.

__init__

def __init__(**kwargs)

Initialize the CodeGeneratorAtomicFlow.

Arguments:

  • kwargs (Any): Keyword arguments.

instantiate_from_config

@classmethod
def instantiate_from_config(cls, config)

Instantiate a CodeGeneratorAtomicFlow from a configuration.

Arguments:

  • config (Dict[str, Any]): Configuration dictionary.

Returns:

CodeGeneratorAtomicFlow: Instantiated CodeGeneratorAtomicFlow.

run

def run(input_data: Dict[str, Any]) -> Dict[str, Any]

Run the flow.

Arguments:

  • input_data (Dict[str, Any]): Input data.

Returns:

Dict[str, Any]: Output data.

__init__