Structured output and tool calls

import json
import os
from openai import OpenAI

with OpenAI(
    base_url="https://api.apione.ai/v1",
    api_key=os.environ["APIONE_API_KEY"],
) as client:
    response = client.chat.completions.create(
        model=os.environ["APIONE_MODEL"],
        messages=[
            {"role": "system", "content": 'Return only JSON with a category string: feature, bug, or other.'},
            {"role": "user", "content": "Add search to the help center."},
        ],
        response_format={"type": "json_object"},
    )

choice = response.choices[0]
if choice.finish_reason != "stop" or not choice.message.content:
    raise ValueError("Incomplete output or missing JSON content")
data = json.loads(choice.message.content)
if (
    not isinstance(data, dict)
    or not isinstance(data.get("category"), str)
    or data["category"] not in {"feature", "bug", "other"}
):
    raise ValueError("category does not meet the application's requirements")
print(data)