Handle errors

import os
import openai
from openai import OpenAI

client = OpenAI(base_url="https://api.altrum.ai/v1", api_key=os.environ["ALTRUM_VIRTUAL_KEY"], max_retries=4)

def ask(text: str) -> str:
    try:
        reply = client.chat.completions.create(
            model="openai/gpt-4o-mini",
            messages=[{"role": "user", "content": text}],
        )
        return reply.choices[0].message.content
    except openai.APIStatusError as error:
        code = (error.body or {}).get("type") if isinstance(error.body, dict) else None
        request_id = error.response.headers.get("x-request-id")
        if code == "guardrail_blocked":
            return "That message can't be sent. Please remove any personal or sensitive details and try again."
        if code == "response_blocked":
            return "I can't answer that. Try asking in a different way."
        if code == "budget_exceeded":
            return "This assistant has reached its spending limit for now."
        raise RuntimeError(f"AI Gateway refused the request: {code} (reference {request_id})") from error

import OpenAI, { APIError } from "openai";

const client = new OpenAI({ baseURL: "https://api.altrum.ai/v1", apiKey: process.env.ALTRUM_VIRTUAL_KEY, maxRetries: 4 });

export async function ask(text: string): Promise<string> {
  try {
    const reply = await client.chat.completions.create({
      model: "openai/gpt-4o-mini",
      messages: [{ role: "user", content: text }],
    });
    return reply.choices[0].message.content ?? "";
  } catch (error) {
    if (!(error instanceof APIError)) throw error;
    const code = error.type;
    const requestId = error.headers?.["x-request-id"];
    if (code === "guardrail_blocked") return "That message can't be sent. Please remove any personal or sensitive details and try again.";
    if (code === "response_blocked") return "I can't answer that. Try asking in a different way.";
    if (code === "budget_exceeded") return "This assistant has reached its spending limit for now.";
    throw new Error(`AI Gateway refused the request: ${code} (reference ${requestId})`);
  }
}