granola-toolkit 0.11.0 → 0.12.0

package/README.md

@@ -214,6 +214,7 @@ Supported environment variables:
 - `TIMEOUT`
 - `CACHE_FILE`
 - `TRANSCRIPT_OUTPUT`
+- `GRANOLA_CLIENT_VERSION`
 
 ## Development Checks
 

package/dist/cli.js

@@ -540,13 +540,23 @@ function parseDocument(value) {
 }
 //#endregion
 //#region src/client/granola.ts
-const USER_AGENT = "Granola/5.354.0";
-const CLIENT_VERSION = "5.354.0";
+const DEFAULT_CLIENT_VERSION = "5.354.0";
 const DOCUMENTS_URL = "https://api.granola.ai/v2/get-documents";
+function resolveClientVersion(value) {
+	return value?.trim() || process.env.GRANOLA_CLIENT_VERSION?.trim() || DEFAULT_CLIENT_VERSION;
+}
 var GranolaApiClient = class {
-	constructor(httpClient, documentsUrl = DOCUMENTS_URL) {
+	clientVersion;
+	documentsUrl;
+	constructor(httpClient, options = DOCUMENTS_URL) {
 		this.httpClient = httpClient;
-		this.documentsUrl = documentsUrl;
+		if (typeof options === "string") {
+			this.documentsUrl = options;
+			this.clientVersion = resolveClientVersion();
+			return;
+		}
+		this.documentsUrl = options.documentsUrl ?? DOCUMENTS_URL;
+		this.clientVersion = resolveClientVersion(options.clientVersion);
 	}
 	async listDocuments(options) {
 		const documents = [];
@@ -559,8 +569,8 @@ var GranolaApiClient = class {
 				offset
 			}, {
 				headers: {
-					"User-Agent": USER_AGENT,
-					"X-Client-Version": CLIENT_VERSION
+					"User-Agent": `Granola/${this.clientVersion}`,
+					"X-Client-Version": this.clientVersion
 				},
 				timeoutMs: options.timeoutMs
 			});
@@ -580,35 +590,79 @@ var GranolaApiClient = class {
 };
 //#endregion
 //#region src/client/http.ts
+const RETRYABLE_STATUS_CODES = new Set([
+	429,
+	500,
+	502,
+	503,
+	504
+]);
+function sleep(delayMs) {
+	return new Promise((resolve) => {
+		setTimeout(resolve, delayMs);
+	});
+}
+function parseRetryAfter(headerValue) {
+	if (!headerValue?.trim()) return;
+	if (/^\d+$/.test(headerValue.trim())) return Number(headerValue.trim()) * 1e3;
+	const retryAt = Date.parse(headerValue);
+	if (Number.isNaN(retryAt)) return;
+	return Math.max(0, retryAt - Date.now());
+}
 var AuthenticatedHttpClient = class {
 	fetchImpl;
 	constructor(options) {
 		this.fetchImpl = options.fetchImpl ?? fetch;
 		this.logger = options.logger;
+		this.maxRetries = options.maxRetries ?? 2;
+		this.retryBaseDelayMs = options.retryBaseDelayMs ?? 500;
+		this.retryMaxDelayMs = options.retryMaxDelayMs ?? 5e3;
+		this.sleepImpl = options.sleepImpl ?? sleep;
 		this.tokenProvider = options.tokenProvider;
 	}
 	logger;
+	maxRetries;
+	retryBaseDelayMs;
+	retryMaxDelayMs;
+	sleepImpl;
 	tokenProvider;
-	async request(options) {
+	async retry(options, attempt, reason, response) {
+		const retryAfterMs = parseRetryAfter(response?.headers.get("retry-after") ?? null);
+		const delayMs = Math.min(retryAfterMs ?? this.retryBaseDelayMs * 2 ** attempt, this.retryMaxDelayMs);
+		this.logger?.warn?.(`${reason}; retrying in ${delayMs}ms (${attempt + 1}/${this.maxRetries})`);
+		await this.sleepImpl(delayMs);
+		return this.request(options, attempt + 1);
+	}
+	async request(options, attempt = 0) {
 		const { retryOnUnauthorized = true, timeoutMs, url } = options;
 		const accessToken = await this.tokenProvider.getAccessToken();
-		const response = await this.fetchImpl(url, {
-			body: options.body,
-			headers: {
-				...options.headers,
-				Authorization: `Bearer ${accessToken}`
-			},
-			method: options.method ?? "GET",
-			signal: AbortSignal.timeout(timeoutMs)
-		});
+		let response;
+		try {
+			response = await this.fetchImpl(url, {
+				body: options.body,
+				headers: {
+					...options.headers,
+					Authorization: `Bearer ${accessToken}`
+				},
+				method: options.method ?? "GET",
+				signal: AbortSignal.timeout(timeoutMs)
+			});
+		} catch (error) {
+			if (attempt < this.maxRetries) {
+				const message = error instanceof Error ? error.message : String(error);
+				return this.retry(options, attempt, `request failed: ${message}`);
+			}
+			throw error;
+		}
 		if (response.status === 401 && retryOnUnauthorized) {
 			this.logger?.warn?.("request returned 401; invalidating token provider and retrying once");
 			await this.tokenProvider.invalidate();
 			return this.request({
 				...options,
 				retryOnUnauthorized: false
-			});
+			}, attempt);
 		}
+		if (RETRYABLE_STATUS_CODES.has(response.status) && attempt < this.maxRetries) return this.retry(options, attempt, `request returned ${response.status} ${response.statusText || ""}`.trim(), response);
 		return response;
 	}
 	async postJson(url, body, options = { timeoutMs: 3e4 }) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "granola-toolkit",
-  "version": "0.11.0",
+  "version": "0.12.0",
   "description": "CLI toolkit for exporting and working with Granola notes and transcripts",
   "keywords": [
     "cli",