function_calls.py

dump_flow = {
    'name': 'dump_flow',
    'description': 'dump the newly generated or converted flow to a current folder in local disk for user and return the folder path',
    'parameters': {
        'type': 'object',
        'properties': {
            'flow_yaml': {
                'type': 'string',
                'description': 'flow yaml string'
            },
            'explaination': {
                'type': 'string',
                'description': 'explanation about how the flow works'
            },
            'python_functions': {
                'type': 'array',
                'description': 'python function implementations',
                'items': {
                    'name': {
                        'type': 'string',
                        'description': 'python node name'
                    },
                    'content': {
                        'type': 'string',
                        'description': 'python file content'
                    }
                }
            },
            'prompts': {
                'type': 'array',
                'description': 'prompts',
                'items': {
                    'name': {
                        'type': 'string',
                        'description': 'llm node name'
                    },
                    'content': {
                        'type': 'string',
                        'description': 'prompt content'
                    }
                }
            },
            'flow_inputs_schema': {
                'type': 'array',
                'description': 'flow inputs schemas',
                'items': {
                    'name': {
                        'type': 'string',
                        'description': 'input name'
                    },
                    'schema': {
                        'type': 'string',
                        'description': 'input schema'
                    }
                }
            },
            'flow_outputs_schema': {
                'type': 'array',
                'description': 'flow outputs schemas',
                'items': {
                    'name': {
                        'type': 'string',
                        'description': 'output name'
                    },
                    'schema': {
                        'type': 'string',
                        'description': 'output schema'
                    }
                }
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['flow_yaml', 'explaination', 'python_functions', 'prompts', 'reasoning']
    }
}

read_local_file = {
    'name': 'read_local_file',
    'description': 'read file content from local disk and return the content in string format',
    'parameters': {
        'type': 'object',
        'properties': {
            'path': {
                'type': 'string',
                'description': 'path to local file'
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['path', 'reasoning']
    }
}

read_local_folder = {
    'name': 'read_local_folder',
    'description': 'read all files content from local folder and return a dictionary of file name and file content',
    'parameters': {
        'type': 'object',
        'properties': {
            'path': {
                'type': 'string',
                'description': 'path to local folder'
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['path', 'reasoning']
    }
}

read_flow_from_local_file = {
    'name': 'read_flow_from_local_file',
    'description': 'read existing flow from a local file',
    'parameters': {
        'type': 'object',
        'properties': {
            'path': {
                'type': 'string',
                'description': 'path to local file'
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['path', 'reasoning']
    }
}

read_flow_from_local_folder ={
    'name': 'read_flow_from_local_folder',
    'description': 'read existing flow from local folder',
    'parameters': {
        'type': 'object',
        'properties': {
            'path': {
                'type': 'string',
                'description': 'path to local folder'
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['path', 'reasoning']
    }
}

dump_sample_inputs = {
    'name': 'dump_sample_inputs',
    'description': 'generate sample inputs for the flow and dump the generated sample inputs into local file for user',
    'parameters': {
        'type': 'object',
        'properties': {
            'file_name': {
                'type': 'string',
                'description': 'file name to store the generated sample inputs',
            },
            'total_count': {
                'type': 'number',
                'description': 'sample inputs count to generate',
            },
            'extra_requirements': {
                'type': 'string',
                'description': 'extra requirements on the generated sample input data',
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called',
            },
        },
        'required': ['file_name', 'total_count', 'extra_requirements', 'reasoning']
    }
}

generate_sample_inputs = {
    'name': 'generate_sample_inputs',
    'description': 'generate sample inputs for the flow',
    'parameters': {
        'type': 'object',
        'properties': {
            'sample_inputs': {
                'type': 'array',
                'description': 'generated sample inputs in json string format',
                'items': {
                    'type': 'string'
                },
            }
        },
        'required': ['sample_inputs']
    }
}

dump_evaluation_flow ={
    'name': 'dump_evaluation_flow',
    'description': 'create evaluation flow for the flow and dump the generated evaluation flow into local file for user and return the path',
    'parameters': {
        'type': 'object',
        'properties': {
            'evaluation_flow_folder': {
                'type': 'string',
                'description': 'target folder to dump the evaluation flow, could be empty',
            },
            'target_output': {
                'type': 'string',
                'description': 'specify which flow output you want to evaluate. If not set, evaluate all flow outputs',
            },
            'total_count': {
                'type': 'number',
                'description': 'total count of the generated evaluation data',
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['evaluation_flow_folder', 'target_output', 'total_count', 'reasoning']
    }
}

dump_evaluation_input = {
    'name': 'dump_evaluation_input',
    'description': 'generate evaluation input flow for the flow and dump the generated evaluation input into local file for user',
    'parameters': {
        'type': 'object',
        'properties': {
            'evaluation_inputs': {
                'type': 'array',
                'description': 'generated evaluation inputs in json string format',
                'items': {
                    'type': 'string'
                },
            }
        },
        'required': ['evaluation_inputs']
    }
}

dump_evaluation_functions = {
    'name': 'dump_evaluation_functions',
    'description': 'generate line_process function and aggregate function python code and dump it to local file',
    'parameters': {
        'type': 'object',
        'properties': {
            'line_process': {
                'type': 'string',
                'description': 'line_process function python code'
            },
            'aggregate': {
                'type': 'string',
                'description': 'aggregate function python code'
            }
        },
        'required': ['line_process', 'aggregate']
    }  
}

dump_flow_definition_and_description = {
    'name': 'dump_flow_definition_and_description',
    'description': 'dump flow yaml and description to local disk',
    'parameters': {
        'type': 'object',
        'properties': {
            'flow_yaml': {
                'type': 'string',
                'description': 'flow yaml string'
            },
            'description': {
                'type': 'string',
                'description': 'description about how the flow works'
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['flow_yaml', 'description', 'reasoning']
    }
}

upsert_flow_files = {
    'name': 'upsert_flow_files',
    'description': 'upsert files in user\'s local flow directory, including flow yaml, flow description, python files and jinja files',
    'parameters': {
        'type': 'object',
        'properties': {
            'files_name_content': {
                'type': 'array',
                'description': 'an array of objects describe the upsert file name and file content',
                'items': {
                    'file_name': {
                        'type': 'string',
                        'description': 'upsert file name'
                    },
                    'file_content': {
                        'type': 'string',
                        'description': 'upsert file content'
                    }
                }
            },
            'reasoning': {
                'type': 'string',
                'description': 'reasoning about why this function is called'
            }
        },
        'required': ['files_name_content', 'reasoning']
    }
}