mrkstu

import json
import os
import ollama


def query_ollama(prompt, model='openhermes:7b-mistral-v2.5-q6_K', context=''):
    response = ollama.generate(
        model=model,
        prompt=context + prompt)
    return response['response'].strip()