llm.py

import os
from groq import Groq

client = Groq(
    api_key=os.environ.get("GROQ_API_KEY")
)

def ask_llm(text):

    response = client.chat.completions.create(
        messages=[
            {"role": "user", "content": text}
        ],
        model="llama-3.3-70b-versatile",
    )

    return response.choices[0].message.content