INNER CODE UNIT · Python

load_llm

Niketkumardheeryan/ML-CaPsule · AI_Based__Resume_Gap_Detector/app.py:58

def load_llm(p_provider, p_model, p_temp, p_tokens, p_api_key, p_base_url):
    kwargs = {
        "model": p_model,
        "model_provider": p_provider,
        "temperature": p_temp,
        "max_tokens": p_tokens,
        "api_key": p_api_key
    }
    # Only add base_url if it's provided so we don't overwrite provider defaults unnecessarily
    if p_base_url:
        kwargs["base_url"] = p_base_url
        
    return init_chat_model(**kwargs)

# Instantiate the model using the sidebar state
model = load_llm(provider, model_name, temperature, max_tokens, api_key, base_url)

# Main app logic

View source record →