config.py

import yaml
from log_writer import logger

with open("config.yaml", "r") as conf:
    config_content = yaml.safe_load(conf)
    for key, value in config_content.items():
        if key == "CODING_MODEL" and value == "gpt-4":
            globals()[key] = "gpt-4-turbo-preview" # Force using gpt-4-turbo-preview if the user set the CODING_MODEL to gpt-4. Because gpt-4 is not longer supports json modes.
        globals()[key] = value
        logger(f"config: {key} -> {value}")

def edit_config(key, value):
    with open("config.yaml", "r") as file:
        lines = file.readlines()

    for i, line in enumerate(lines):
        if f"{key}:" in line:
            lines[i] = line.replace(line.split(":")[1].strip().strip('"'), f"{value}")

    with open("config.yaml", "w") as file:
        file.writelines(lines)

    logger(f"edit_config: {key} -> {value}")