Google Sheets Integration

const ANTBASE_API_KEY = 'your-ant-api-key';
const ANTBASE_URL = 'https://antbase.ai/v1/chat/completions';

/**
 * Call Antbase AI from a cell.
 * @param {string} prompt The prompt to send.
 * @param {string} model The model to use (default: ant:auto).
 * @return {string} The AI response.
 * @customfunction
 */
function AI(prompt, model) {
  model = model || 'ant:auto';

  const payload = {
    model: model,
    messages: [{ role: 'user', content: prompt }],
    max_tokens: 512,
    temperature: 0,
  };

  const options = {
    method: 'post',
    contentType: 'application/json',
    headers: {
      Authorization: 'Bearer ' + ANTBASE_API_KEY,
    },
    payload: JSON.stringify(payload),
    muteHttpExceptions: true,
  };

  const response = UrlFetchApp.fetch(ANTBASE_URL, options);
  const json = JSON.parse(response.getContentText());

  if (json.error) {
    return 'Error: ' + json.error.message;
  }

  return json.choices[0].message.content.trim();
}

/**
 * Classify text into a category.
 * @param {string} text The text to classify.
 * @param {string} categories Comma-separated categories.
 * @return {string} The category.
 * @customfunction
 */
function AI_CLASSIFY(text, categories) {
  const prompt = 'Classify the following text into exactly one of these categories: '
    + categories + '. Respond with only the category name.\n\nText: ' + text;
  return AI(prompt, 'ant:fast');
}

/**
 * Analyze sentiment of text.
 * @param {string} text The text to analyze.
 * @return {string} positive, negative, or neutral.
 * @customfunction
 */
function AI_SENTIMENT(text) {
  const prompt = 'Analyze the sentiment of this text. Respond with exactly one word: positive, negative, or neutral.\n\n' + text;
  return AI(prompt, 'ant:fast');
}

=AI("Summarize: " & A1)
=AI("Translate to French: " & A1, "ant:best")
=AI_CLASSIFY(A1, "support, sales, billing, spam")
=AI_SENTIMENT(A1)