Add Code Interpreter tooling and generated file download support

src/code.gs

@@ -52,6 +52,10 @@ const GenAIApp = (function () {
       let browsing = false;
       let reasoning_effort = "medium";
       let knowledgeLink = [];
+      this._codeInterpreterEnabled = false;
+      this._codeInterpreterContainerId = null;
+      this._generatedFiles = [];
+      this._lastContainerId = null;
       let compaction_enabled = false;
       let compaction_threshold = 10000;
 
@@ -295,6 +299,28 @@ const GenAIApp = (function () {
         return this;
       };
 
+      /**
+       * Enables OpenAI Code Interpreter support for this chat.
+       * @param {string} [containerId] - OPTIONAL - Explicit container ID to reuse.
+       * @returns {Chat} - The current Chat instance.
+       * @example
+       * const chat = GenAIApp.newChat()
+       *   .addFile(DriveApp.getFileById("YOUR_FILE_ID").getBlob())
+       *   .enableCodeInterpreter()
+       *   .addMessage("Process this file and generate an updated version.");
+       * chat.run({ model: "gpt-5.4" });
+       * const generatedFiles = chat.getGeneratedFiles();
+       * const blob = chat.downloadGeneratedFile(0);
+       * DriveApp.createFile(blob);
+       */
+      this.enableCodeInterpreter = function (containerId) {
+        this._codeInterpreterEnabled = true;
+        if (containerId) {
+          this._codeInterpreterContainerId = containerId;
+        }
+        return this;
+      };
+
       /**
        * Includes the content of a web page in the prompt sent to openAI
        * @param {string} url - the url of the webpage you want to fetch
@@ -425,6 +451,8 @@ const GenAIApp = (function () {
       this.run = function (advancedParametersObject) {
         this._lastUsage = null;
         last_response_id = null;
+        this._generatedFiles = [];
+        this._lastContainerId = null;
 
         model = advancedParametersObject?.model ?? model;
         temperature = advancedParametersObject?.temperature ?? temperature;
@@ -508,6 +536,10 @@ const GenAIApp = (function () {
               console.warn(`[GenAIApp] - Warning: input token usage (${this._lastUsage.input_tokens}) exceeded configured threshold (${this._inputTokenWarningThreshold}) for response ${responseMessage.id}`);
             }
           }
+          this._generatedFiles = this._extractContainerFileCitations(responseMessage);
+          if (this._generatedFiles.length > 0) {
+            this._lastContainerId = this._generatedFiles[0].containerId;
+          }
 
           // OpenAI Responses API returns top-level "id"
           if (!model.includes("gemini")) {
@@ -725,9 +757,108 @@ const GenAIApp = (function () {
           payload.tools.push(fileSearchTool);
           payload.include = ["file_search_call.results"];
         }
+
+        if (this._codeInterpreterEnabled) {
+          payload.parallel_tool_calls = false;
+          if (this._codeInterpreterContainerId) {
+            payload.tools.push({
+              type: "container",
+              container_id: this._codeInterpreterContainerId
+            });
+          }
+          else {
+            payload.tools.push({
+              type: "code_interpreter",
+              container: {
+                type: "auto"
+              }
+            });
+          }
+        }
         return payload;
       }
 
+      this._extractContainerFileCitations = function (response) {
+        if (!Array.isArray(response?.output)) {
+          return [];
+        }
+        const citations = [];
+        response.output.forEach(outputItem => {
+          const contentItems = Array.isArray(outputItem?.content) ? outputItem.content : [];
+          contentItems.forEach(contentItem => {
+            const annotations = Array.isArray(contentItem?.annotations) ? contentItem.annotations : [];
+            annotations.forEach(annotation => {
+              const citation = annotation?.container_file_citation;
+              if (citation?.container_id && citation?.file_id) {
+                citations.push({
+                  containerId: citation.container_id,
+                  fileId: citation.file_id,
+                  filename: citation.filename || null
+                });
+              }
+            });
+          });
+        });
+        return citations;
+      };
+
+      this._downloadContainerFile = function (containerId, fileId, filename) {
+        const endpointUrl = `${apiBaseUrl}/v1/containers/${containerId}/files/${fileId}/content`;
+        const response = _callGenAIApi(endpointUrl, null, "GET", true);
+        const blob = response.getBlob();
+        const contentType = response.getHeaders()["Content-Type"] || blob.getContentType();
+        if (contentType) {
+          blob.setContentType(contentType);
+        }
+        if (filename) {
+          blob.setName(filename);
+        }
+        return blob;
+      };
+
+      /**
+       * Returns generated files from the last run call.
+       * @returns {{containerId: string, fileId: string, filename: string}[]} Generated files metadata.
+       */
+      this.getGeneratedFiles = function () {
+        return this._generatedFiles;
+      };
+
+      /**
+       * Downloads a generated file from the last run.
+       * @param {string|number} fileIdOrIndex - File ID or index from getGeneratedFiles().
+       * @returns {Blob} Downloaded file blob that can be stored with DriveApp.createFile(blob).
+       * @example
+       * const chat = GenAIApp.newChat()
+       *   .addFile(DriveApp.getFileById("YOUR_FILE_ID").getBlob())
+       *   .enableCodeInterpreter()
+       *   .addMessage("Process this file and generate an updated version.");
+       * chat.run({ model: "gpt-5.4" });
+       * const files = chat.getGeneratedFiles();
+       * const blob = chat.downloadGeneratedFile(files[0].fileId);
+       * DriveApp.createFile(blob);
+       */
+      this.downloadGeneratedFile = function (fileIdOrIndex) {
+        let targetFile;
+        if (typeof fileIdOrIndex === "number") {
+          targetFile = this._generatedFiles[fileIdOrIndex];
+        }
+        else {
+          targetFile = this._generatedFiles.find(file => file.fileId === fileIdOrIndex);
+        }
+        if (!targetFile) {
+          throw new Error("[GenAIApp] - Generated file not found. Provide a valid file ID or index from getGeneratedFiles().");
+        }
+        return this._downloadContainerFile(targetFile.containerId, targetFile.fileId, targetFile.filename);
+      };
+
+      this.getContainerId = function () {
+        if (this._lastContainerId) {
+          return this._lastContainerId;
+        }
+        return this._generatedFiles[0]?.containerId || this._codeInterpreterContainerId || null;
+      };
+
       /**
        * Builds and returns a payload for a Gemini API call, configuring content, model parameters, 
        * and tool settings based on advanced options and feature flags such as browsing. 
@@ -1383,7 +1514,7 @@ const GenAIApp = (function () {
 * @returns {object} - The response message from the GenAI API.
 * @throws {Error} If the API call fails after the maximum number of retries.
 */
-  function _callGenAIApi(endpoint, payload) {
+  function _callGenAIApi(endpoint, payload, method = "post", returnRawResponse = false) {
     let authMethod = 'Bearer ' + openAIKey;
     if (endpoint.includes("google")) {
       if (geminiKey) {
@@ -1413,11 +1544,13 @@ const GenAIApp = (function () {
         headers['x-goog-api-key'] = geminiKey;
       }
       const options = {
-        method: 'post',
+        method: method.toLowerCase(),
         headers: headers,
-        payload: JSON.stringify(payload),
         muteHttpExceptions: true
       };
+      if (payload !== null && payload !== undefined) {
+        options.payload = JSON.stringify(payload);
+      }
 
       let response;
       // if the ErrorHandler library is loaded and supports backoff, use it (https://github.com/RomainVialard/ErrorHandler)
@@ -1441,6 +1574,9 @@ const GenAIApp = (function () {
       const responseCode = response.getResponseCode();
 
       if (responseCode === 200) {
+        if (returnRawResponse) {
+          return response;
+        }
         // The request was successful, exit the loop.
         const parsedResponse = JSON.parse(response.getContentText());
         if (endpoint.includes("google")) {

src/testFunctions.gs

@@ -1,6 +1,8 @@
 const GPT_MODEL = "gpt-5.4";
 const REASONING_MODEL = "o4-mini";
 const GEMINI_MODEL = "gemini-2.5-pro";
+const TEST_CODE_INTERPRETER_XLSX_DRIVE_FILE_ID = "";
+const TEST_CODE_INTERPRETER_PDF_DRIVE_FILE_ID = "";
 
 // Run all tests
 function testAll() {
@@ -13,6 +15,13 @@ function testAll() {
   testVision();
   testMaximumAPICalls();
   testInputTokenWarning();
+  // OpenAI-only tests - require valid Drive file IDs.
+  if (TEST_CODE_INTERPRETER_XLSX_DRIVE_FILE_ID) {
+    testCodeInterpreterExcel(TEST_CODE_INTERPRETER_XLSX_DRIVE_FILE_ID);
+  }
+  if (TEST_CODE_INTERPRETER_PDF_DRIVE_FILE_ID) {
+    testCodeInterpreterPDF(TEST_CODE_INTERPRETER_PDF_DRIVE_FILE_ID);
+  }
 }
 
 
@@ -149,6 +158,36 @@ ${highThresholdResponse}`);
   console.log("Input token warning test (high threshold): verify that no warning log was emitted.");
 }
 
+function testCodeInterpreterExcel(driveFileId) {
+  GenAIApp.setOpenAIAPIKey(OPEN_AI_API_KEY);
+  const inputBlob = DriveApp.getFileById(driveFileId).getBlob();
+  const chat = GenAIApp.newChat();
+  chat
+    .addFile(inputBlob)
+    .enableCodeInterpreter()
+    .addMessage("Add a new column at the end that calculates row totals for all numeric columns");
+  const response = chat.run({ model: GPT_MODEL, max_tokens: 4000 });
+  console.log(`Code Interpreter Excel response:\n${response}`);
+  const outputBlob = chat.downloadGeneratedFile(0);
+  const createdFile = DriveApp.createFile(outputBlob);
+  console.log(`Generated Excel file created: ${createdFile.getId()}`);
+}
+
+function testCodeInterpreterPDF(driveFileId) {
+  GenAIApp.setOpenAIAPIKey(OPEN_AI_API_KEY);
+  const inputBlob = DriveApp.getFileById(driveFileId).getBlob();
+  const chat = GenAIApp.newChat();
+  chat
+    .addFile(inputBlob)
+    .enableCodeInterpreter()
+    .addMessage("Add a summary paragraph at the top of the document describing its main contents");
+  const response = chat.run({ model: GPT_MODEL, max_tokens: 4000 });
+  console.log(`Code Interpreter PDF response:\n${response}`);
+  const outputBlob = chat.downloadGeneratedFile(0);
+  const createdFile = DriveApp.createFile(outputBlob);
+  console.log(`Generated PDF file created: ${createdFile.getId()}`);
+}
+
 // Weather function implementation
 function getWeather(cityName) {
   return `The weather in ${cityName} is 19°C today.`;