import argparse
import yaml
import os


parser = argparse.ArgumentParser()
parser.add_argument("--config", type=str, default="config.yaml.dev")
parser.add_argument("--mode", type=str, default="cli")
args = parser.parse_args()

if __name__ != "__main__":
    args.config = "config.yaml"

config = yaml.load(open(args.config, "r"), Loader=yaml.FullLoader)

if not config["dev"]:
    GROQ_API_KEY = os.getenv('GROQ_API_KEY')
else:
    GROQ_API_KEY = config.get("GROQ", {}).get("key")