Function ask_llm

llm/chat.py:70–81

(prompt, temp=None)

def ask_llm(prompt, temp=None):
    key_config = args.key_config
    key_version = args.key_version  # gpt-35-turbo, gpt-4
    api_key = CONFIG[key_config]["api_key"]
    api_base = CONFIG[key_config]["api_base"]
    engine = key_version
    model_name = key_version
    temperature = CONFIG["methods"]["temperature"]
    if temp is not None:
        temperature = temp
    max_tokens = CONFIG["max_tokens"]
    return connect_gpt(engine, api_key, api_base, prompt, max_tokens, temperature)