notebooklm-sdk 0.3.0 → 0.3.2

package/dist/index.cjs

@@ -21,7 +21,10 @@ __export(enums_exports, {
   ArtifactTypeCode: () => exports.ArtifactTypeCode,
   AudioFormat: () => exports.AudioFormat,
   AudioLength: () => exports.AudioLength,
+  ChatGoal: () => exports.ChatGoal,
   ChatMode: () => exports.ChatMode,
+  ChatResponseLength: () => exports.ChatResponseLength,
+  DriveMimeType: () => exports.DriveMimeType,
   ExportType: () => exports.ExportType,
   InfographicDetail: () => exports.InfographicDetail,
   InfographicOrientation: () => exports.InfographicOrientation,
@@ -64,7 +67,7 @@ function sourceStatusFromCode(code) {
 function chatModeToParams(mode) {
   return CHAT_MODE_PARAMS[mode];
 }
-exports.RPCMethod = void 0; exports.ArtifactTypeCode = void 0; var ArtifactStatusCode, SourceStatusCode; exports.AudioFormat = void 0; exports.AudioLength = void 0; exports.VideoFormat = void 0; exports.VideoStyle = void 0; exports.QuizQuantity = void 0; exports.QuizDifficulty = void 0; exports.InfographicOrientation = void 0; exports.InfographicDetail = void 0; exports.InfographicStyle = void 0; exports.SlideDeckFormat = void 0; exports.SlideDeckLength = void 0; exports.ExportType = void 0; var SOURCE_TYPE_MAP, ARTIFACT_TYPE_MAP, ARTIFACT_STATUS_MAP, SOURCE_STATUS_MAP; exports.ChatMode = void 0; var CHAT_MODE_PARAMS; exports.ShareAccess = void 0; exports.ShareViewLevel = void 0; exports.SharePermission = void 0;
+exports.RPCMethod = void 0; exports.ArtifactTypeCode = void 0; var ArtifactStatusCode, SourceStatusCode; exports.AudioFormat = void 0; exports.AudioLength = void 0; exports.VideoFormat = void 0; exports.VideoStyle = void 0; exports.QuizQuantity = void 0; exports.QuizDifficulty = void 0; exports.InfographicOrientation = void 0; exports.InfographicDetail = void 0; exports.InfographicStyle = void 0; exports.SlideDeckFormat = void 0; exports.SlideDeckLength = void 0; exports.ExportType = void 0; exports.DriveMimeType = void 0; var SOURCE_TYPE_MAP, ARTIFACT_TYPE_MAP, ARTIFACT_STATUS_MAP, SOURCE_STATUS_MAP; exports.ChatMode = void 0; var CHAT_MODE_PARAMS; exports.ChatGoal = void 0; exports.ChatResponseLength = void 0; exports.ShareAccess = void 0; exports.ShareViewLevel = void 0; exports.SharePermission = void 0;
 var init_enums = __esm({
   "src/types/enums.ts"() {
     exports.RPCMethod = {
@@ -213,6 +216,12 @@ var init_enums = __esm({
       DOCS: 1,
       SHEETS: 2
     };
+    exports.DriveMimeType = {
+      GOOGLE_DOC: "application/vnd.google-apps.document",
+      GOOGLE_SLIDES: "application/vnd.google-apps.presentation",
+      GOOGLE_SHEETS: "application/vnd.google-apps.spreadsheet",
+      PDF: "application/pdf"
+    };
     SOURCE_TYPE_MAP = {
       1: "google_docs",
       2: "google_slides",
@@ -264,6 +273,22 @@ var init_enums = __esm({
       concise: [1, 5],
       detailed: [1, 4]
     };
+    exports.ChatGoal = {
+      /** General purpose research and brainstorming. */
+      DEFAULT: 1,
+      /** Custom instructions (up to 10,000 characters). */
+      CUSTOM: 2,
+      /** Educational focus with learning-oriented responses. */
+      LEARNING_GUIDE: 3
+    };
+    exports.ChatResponseLength = {
+      /** Standard response length. */
+      DEFAULT: 1,
+      /** Verbose, detailed responses. */
+      LONGER: 4,
+      /** Concise, brief responses. */
+      SHORTER: 5
+    };
     exports.ShareAccess = {
       /** Only explicitly shared users can access */
       RESTRICTED: 0,
@@ -786,6 +811,30 @@ var ArtifactsAPI = class {
     const artifacts = await this.list(notebookId);
     return artifacts.find((a) => a.id === artifactId) ?? null;
   }
+  async listAudio(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "audio");
+  }
+  async listVideo(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "video");
+  }
+  async listReports(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "report");
+  }
+  async listQuizzes(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "quiz");
+  }
+  async listFlashcards(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "flashcards");
+  }
+  async listInfographics(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "infographic");
+  }
+  async listSlideDecks(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "slide_deck");
+  }
+  async listDataTables(notebookId) {
+    return (await this.list(notebookId)).filter((a) => a.kind === "data_table");
+  }
   async delete(notebookId, artifactId) {
     const params = [[2], notebookId, artifactId];
     await this.rpc.call(exports.RPCMethod.DELETE_ARTIFACT, params, {
@@ -1198,6 +1247,24 @@ ${opts.extraInstructions}` : cfg.prompt;
   }
   // ---------------------------------------------------------------------------
   // Internal
+  /** Export a completed report artifact to Google Docs. Returns the created document URL. */
+  async exportReport(notebookId, artifactId, title) {
+    const params = [null, artifactId, null, title, exports.ExportType.DOCS];
+    const result = await this.rpc.call(exports.RPCMethod.EXPORT_ARTIFACT, params, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
+    });
+    return extractExportUrl(result);
+  }
+  /** Export a completed data table artifact to Google Sheets. Returns the created spreadsheet URL. */
+  async exportDataTable(notebookId, artifactId, title) {
+    const params = [null, artifactId, null, title, exports.ExportType.SHEETS];
+    const result = await this.rpc.call(exports.RPCMethod.EXPORT_ARTIFACT, params, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
+    });
+    return extractExportUrl(result);
+  }
   // ---------------------------------------------------------------------------
   /**
    * Fetch a Google-hosted media URL, manually following redirects to ensure
@@ -1280,6 +1347,21 @@ function parseDataTable(rawData) {
     throw new Error(`Failed to parse data table: ${e}`);
   }
 }
+function extractExportUrl(result) {
+  if (!Array.isArray(result)) return null;
+  function findUrl(data, depth = 5) {
+    if (depth <= 0 || data == null) return null;
+    if (typeof data === "string" && data.startsWith("https://")) return data;
+    if (Array.isArray(data)) {
+      for (const item of data) {
+        const found = findUrl(item, depth - 1);
+        if (found) return found;
+      }
+    }
+    return null;
+  }
+  return findUrl(result);
+}
 function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -1410,6 +1492,23 @@ var ChatAPI = class {
     }
     return null;
   }
+  /**
+   * Low-level chat configuration. Set goal, response length, and optional
+   * custom instructions directly. Persists on the server per notebook.
+   * Use `setMode()` for preset combinations instead.
+   */
+  async configure(notebookId, goal, length, customPrompt) {
+    if (goal === exports.ChatGoal.CUSTOM && !customPrompt) {
+      throw new Error("customPrompt is required when goal is ChatGoal.CUSTOM");
+    }
+    const goalArray = goal === exports.ChatGoal.CUSTOM ? [goal, customPrompt] : [goal];
+    const chatSettings = [goalArray, [length]];
+    const params = [notebookId, [[null, null, null, null, null, null, null, chatSettings]]];
+    await this.rpc.call(exports.RPCMethod.RENAME_NOTEBOOK, params, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
+    });
+  }
   /**
    * Set the chat mode for a notebook. Persists on the server — affects all
    * subsequent `ask()` calls until changed.
@@ -2121,6 +2220,44 @@ var SourcesAPI = class {
       _typeCode: null
     };
   }
+  async addDrive(notebookId, fileId, title, mimeType = exports.DriveMimeType.GOOGLE_DOC, opts = {}) {
+    const sourceData = [
+      [fileId, mimeType, 1, title],
+      null,
+      null,
+      null,
+      null,
+      null,
+      null,
+      null,
+      null,
+      null,
+      1
+    ];
+    const params = [
+      [sourceData],
+      notebookId,
+      [2],
+      [1, null, null, null, null, null, null, null, null, null, [1]]
+    ];
+    const result = await this.rpc.call(exports.RPCMethod.ADD_SOURCE, params, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
+    });
+    const sourceId = extractSourceId(result);
+    if (opts.waitUntilReady) {
+      return this.waitUntilReady(notebookId, sourceId, opts.waitTimeout);
+    }
+    return {
+      id: sourceId,
+      title,
+      url: null,
+      kind: "unknown",
+      createdAt: null,
+      status: "processing",
+      _typeCode: null
+    };
+  }
   async addFile(notebookId, filePath, mimeType, opts = {}) {
     const fileData = fs.readFileSync(filePath);
     const fileName = filePath.split("/").pop() ?? "file";