climateqa/engine/llm/ollama.py

from langchain_community.llms import Ollama

def get_llm(model="llama3", **kwargs):
    return Ollama(model=model, **kwargs)