write_tex.py

# write_tex.py
import argparse
from texs import write_agent_behavior, compare_tokens, compare_behavior_tex
from src.utils.get_model import get_fixed_model, get_variable_model

def main(function_name):
    # List of available functions mapped to their corresponding actions
    functions = {
        'agent_behavior_fixed_model': lambda: write_agent_behavior(get_fixed_model()),
        'agent_behavior_variable_model': lambda: write_agent_behavior(get_variable_model()),
        'compare_tokens': lambda: compare_tokens([get_fixed_model(), get_variable_model()]),
        'compare_behavior': lambda: compare_behavior_tex([get_fixed_model(), get_variable_model()])
    }
    if function_name in functions:
        functions[function_name]()
    else:
        raise ValueError(f"Function '{function_name}' is not available. Choose from: {', '.join(functions.keys())}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Run specific LaTeX generation function.")
    parser.add_argument('function', type=str, help="Name of the function to run (e.g., compare_tokens, agent_behavior_fixed_model, agent_behavior_variable_model)")
    args = parser.parse_args()
    main(args.function)