Google スプレッドシートやドキュメントと Azure OpenAI Service を連携してみた

var apikey = "APIキー入力";
var model = "gpt-3.5-turbo";
var temperature = 0;
var url = "https://Azure OpenAI リソースの名前入力.openai.azure.com/openai/deployments/モデルデプロイの名前入力/chat/completions?api-version=2023-05-15";


function GPT(content, maxTokens=2048) {
  if(content){
    const requestBody = {
      "model": model,
      "messages": [{'role': 'user', 'content': content}],
      "temperature": temperature,
      "max_tokens": maxTokens
    };


    const requestOptions = {
      "method": "POST",
      "headers": {
        "Content-Type": "application/json",
        "api-key": apikey
      },
      "payload": JSON.stringify(requestBody)
    };


    var response = JSON.parse(UrlFetchApp.fetch(url, requestOptions).getContentText());


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


  }else{
    return "no result";
  }
}

const apikey = "APIキー入力";
const MAX_TOKENS = 200;


function AI(prompt, temperature = 0, model = "gpt-3.5-turbo") {
  const url = "https://Azure OpenAI リソースの名前入力.openai.azure.com/openai/deployments/モデルデプロイの名前入力/chat/completions?api-version=2023-05-15";
  const requestBody = {
    "model": model,
    "messages": [{ 'role': 'user', 'content': prompt }],
    "temperature": temperature,
    "max_tokens": MAX_TOKENS
  };
  const options = {
    "method": "POST",
    "headers": {
      "Content-Type": "application/json",
      "api-key": apikey
    },
    "payload": JSON.stringify(requestBody)
  };
  const res = JSON.parse(UrlFetchApp.fetch(url, options).getContentText());
  return res.choices[0].message.content;
}




function CATEGORIZE(categories, input, rules = []) {
  const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + ". " + rules.join(". ") + "The category for '" + input + "' is ";
  console.log(prompt);
  const completion = AI(prompt, 0, "gpt-3.5-turbo");
  // Replace "s and .s at the start and end of the string
  return completion.replace(/^"/g, '').replace(/["|.]{0,2}$/, '');
}

function getDocsSelection(e) {
  var text = '';
  var selection = DocumentApp.getActiveDocument().getSelection();
  Logger.log(selection);
  if (selection) {
    var elements = selection.getRangeElements();
    for (var i = 0; i < elements.length; i++) {
      Logger.log(elements[i]);
      var element = elements[i];
      // Only modify elements that can be edited as text; skip images and other non-text elements.
      if (element.getElement().asText() && element.getElement().asText().getText() !== '') {
        text += element.getElement().asText().getText() + '\n';
      }
    }
  }
  addText(text);
}


function addText(question){
  var doc = DocumentApp.getActiveDocument();
  var body = doc.getBody();
  var paragraphs = body.getParagraphs();
  var p1 = paragraphs[paragraphs.length-1]
  p1.appendText("\n\n" + AI(question));
}

function onOpen() {
  var  ui = DocumentApp.getUi()
  
  var  menu = ui.createMenu("文章作成");
  
  menu.addItem("作成","myFunction");
  menu.addToUi();
}


function myFunction(){
  getDocsSelection();
}