INNER CODE UNIT · Python

get_llm

ikatsov/tensor-house · supply-chain/control_center_llm/utilities.py:21

def get_llm(config, temperature=0.7, streaming=True, stop=""):
    llm = ChatGoogleGenerativeAI(
        google_api_key=config["google_api_key"],
        model=config["text_model"],
        temperature=temperature,
        streaming=streaming,
        stop=stop
    )

    return llm

View source record →