def get_llm()

in experiments/legacy/backend/utils.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

def get_llm(project=config.PROJECT, location=config.LOCATION):
    vertexai.init(project=config.PROJECT, location=config.LOCATION)
    return vertexai.language_models.TextGenerationModel.from_pretrained("text-bison")