models/LLM.py

from . import *

model = "openai/gpt-oss-20b"
messages = [
    {
        "role": "system",
        "content": """\
You are a conversational AI, and you must follow the following rules.
1. Keep your answers consice and short.
2. Answer in the same language you were spoke to with.
"""
    }
]

def llm(message:str):
    messages.append(
        {
            "role": "user",
            "content": message
        }
    )

    completion = client.chat.completions.create(
        model=model,
        messages=messages
    )

    response = completion.choices[0].message.content
    messages.append(
        {
            "role": "assistant",
            "content": response
        }
    )
    print("AI Response:", response)

    return response