from langchain_community.chat_models import ChatOllama
from langchain.schema.messages import HumanMessage

llm = ChatOllama(model="llama3", temperature=0)

def call_llm(prompt: str) -> str:
    response = llm([HumanMessage(content=prompt)])
    return response.content