from nestbox_ai_functions import use_chatbot
from openai import OpenAI

MODEL_NAME = "openai/gpt-oss-20b"

client = OpenAI(
    base_url="http://localhost:8000/v1",
    api_key="sk-xxxxxx"
)

@use_chatbot
async def {{agentName}}(context, events):
    try:
        response = client.chat.completions.create(
            model=MODEL_NAME,
            messages=context.messages,
            stream=False,
        )

        result = (
            response.choices[0].message.content
            if response.choices
            else None
        )

        if not result:
            await events.emitQueryFailed({
                "data": {
                    "error": "No response from model"
                }
            })
        else:
            await events.emitQueryCompleted({
                "data": result
            })

    except Exception as e:
        await events.emitQueryFailed({
            "data": {
                "error": str(e)
            }
        })