import requests

MODEL = "finstrom-ai-v1.5"


def main() -> None:
    response = requests.post(
        "http://localhost:11434/api/generate",
        json={
            "model": MODEL,
            "prompt": "Hello FINSTROM, introduce yourself.",
            "stream": False,
        },
        timeout=120,
    )
    response.raise_for_status()
    print(response.json()["response"])


if __name__ == "__main__":
    main()