@dbx-tools/genie 0.9.187 → 0.9.189

package/README.md

@@ -183,6 +183,42 @@ thrown.
 - `pollIntervalMs` - fallback polling cadence, default `500`.
 - `context` - SDK `Context` or `AbortSignal` for cancellation.
 
+## Download a chart image
+
+After requesting visualizations, select the message attachment with a `viz`
+field and download its PNG using the same conversation and message IDs:
+
+```ts
+import { downloadGenieVisualization } from "@dbx-tools/genie";
+
+const visualization = message.attachments?.find((attachment) => attachment.viz);
+if (visualization?.attachment_id) {
+  const png = await downloadGenieVisualization(
+    {
+      space_id: message.space_id,
+      conversation_id: message.conversation_id,
+      message_id: message.message_id,
+      attachment_id: visualization.attachment_id,
+    },
+    { workspaceClient },
+  );
+}
+```
+
+Use the visualization's attachment ID, not its linked query attachment ID.
+The SDK's `raw: true` transport returns the binary body and keeps normal SDK
+authentication and cancellation. The helper validates the PNG response, then
+uses the optional `sharp` dependency to resize it to four times its original
+width and height with Lanczos3 and apply a mild unsharp mask. If `sharp` is not
+installed or cannot process the image, the helper returns the validated
+original bytes. Processing happens before the bytes return to callers, so a
+downstream cache stores the enhanced image when enhancement succeeds.
+
+This is a raster cleanup fallback. If the source ECharts instance is available,
+prefer `getDataURL({ pixelRatio: 3 })` or SVG so chart text is re-rendered rather
+than enlarged from a screenshot. The endpoint also accepts visualization
+attachments from Agent Mode conversations.
+
 ## Modules
 
 - `chat` - `genieChat()` raw snapshot stream and `genieEventChat()` typed event
@@ -190,6 +226,7 @@ thrown.
 - `agentMode` - `genieAgentModeChat()`, availability classification, and Agent
   Mode transport options plus Agent Mode conversation history.
 - `space` - `getGenieSpace()` and `genieSampleQuestions()`.
+- `visualization` - `downloadGenieVisualization()` for validated and optionally enhanced PNG downloads.
 
 The AppKit-Mastra package builds its Genie tools on top of this driver. See
 [@dbx-tools/appkit-mastra](../appkit-mastra) for the agent-level

package/index.ts

@@ -3,11 +3,12 @@
 // Hand edits are overwritten on the next watch; this file is read-only.
 
 export const PACKAGE_IDENTIFIER = "@dbx-tools/genie";
-export const PACKAGE_VERSION = "0.9.187";
+export const PACKAGE_VERSION = "0.9.189";
 export * as agentMode from "./src/agent-mode.ts";
 export * as chat from "./src/chat.ts";
 export * as feedback from "./src/feedback.ts";
 export * as space from "./src/space.ts";
+export * as visualization from "./src/visualization.ts";
 export { genieAgentModeChat, listAgentModeConversation, isAgentModeUnavailable } from "./src/agent-mode.ts";
 export type { GenieAgentModeOptions } from "./src/agent-mode.ts";
 export { genieChat, genieEventChat } from "./src/chat.ts";
@@ -16,3 +17,5 @@ export { sendGenieMessageFeedback } from "./src/feedback.ts";
 export type { SendGenieMessageFeedbackOptions } from "./src/feedback.ts";
 export { createGenieSpace, getGenieSpace, genieSampleQuestions } from "./src/space.ts";
 export type { GetGenieSpaceOptions, CreateGenieSpaceOptions } from "./src/space.ts";
+export { upscaleGenieVisualizationPng, downloadGenieVisualization } from "./src/visualization.ts";
+export type { DownloadGenieVisualizationOptions } from "./src/visualization.ts";

package/lib/index.d.ts

@@ -1,9 +1,10 @@
 export declare const PACKAGE_IDENTIFIER = "@dbx-tools/genie";
-export declare const PACKAGE_VERSION = "0.9.187";
+export declare const PACKAGE_VERSION = "0.9.189";
 export * as agentMode from "./src/agent-mode.ts";
 export * as chat from "./src/chat.ts";
 export * as feedback from "./src/feedback.ts";
 export * as space from "./src/space.ts";
+export * as visualization from "./src/visualization.ts";
 export { genieAgentModeChat, listAgentModeConversation, isAgentModeUnavailable } from "./src/agent-mode.ts";
 export type { GenieAgentModeOptions } from "./src/agent-mode.ts";
 export { genieChat, genieEventChat } from "./src/chat.ts";
@@ -12,3 +13,5 @@ export { sendGenieMessageFeedback } from "./src/feedback.ts";
 export type { SendGenieMessageFeedbackOptions } from "./src/feedback.ts";
 export { createGenieSpace, getGenieSpace, genieSampleQuestions } from "./src/space.ts";
 export type { GetGenieSpaceOptions, CreateGenieSpaceOptions } from "./src/space.ts";
+export { upscaleGenieVisualizationPng, downloadGenieVisualization } from "./src/visualization.ts";
+export type { DownloadGenieVisualizationOptions } from "./src/visualization.ts";

package/lib/index.js

@@ -2,13 +2,15 @@
 // Regenerated from the exporting modules in ./src.
 // Hand edits are overwritten on the next watch; this file is read-only.
 export const PACKAGE_IDENTIFIER = "@dbx-tools/genie";
-export const PACKAGE_VERSION = "0.9.187";
+export const PACKAGE_VERSION = "0.9.189";
 export * as agentMode from "./src/agent-mode.js";
 export * as chat from "./src/chat.js";
 export * as feedback from "./src/feedback.js";
 export * as space from "./src/space.js";
+export * as visualization from "./src/visualization.js";
 export { genieAgentModeChat, listAgentModeConversation, isAgentModeUnavailable } from "./src/agent-mode.js";
 export { genieChat, genieEventChat } from "./src/chat.js";
 export { sendGenieMessageFeedback } from "./src/feedback.js";
 export { createGenieSpace, getGenieSpace, genieSampleQuestions } from "./src/space.js";
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSwyQ0FBMkM7QUFDM0MsbURBQW1EO0FBQ25ELHdFQUF3RTtBQUV4RSxNQUFNLENBQUMsTUFBTSxrQkFBa0IsR0FBRyxrQkFBa0IsQ0FBQztBQUNyRCxNQUFNLENBQUMsTUFBTSxlQUFlLEdBQUcsU0FBUyxDQUFDO0FBQ3pDLE9BQU8sS0FBSyxTQUFTLE1BQU0scUJBQXFCLENBQUM7QUFDakQsT0FBTyxLQUFLLElBQUksTUFBTSxlQUFlLENBQUM7QUFDdEMsT0FBTyxLQUFLLFFBQVEsTUFBTSxtQkFBbUIsQ0FBQztBQUM5QyxPQUFPLEtBQUssS0FBSyxNQUFNLGdCQUFnQixDQUFDO0FBQ3hDLE9BQU8sRUFBRSxrQkFBa0IsRUFBRSx5QkFBeUIsRUFBRSxzQkFBc0IsRUFBRSxNQUFNLHFCQUFxQixDQUFDO0FBRTVHLE9BQU8sRUFBRSxTQUFTLEVBQUUsY0FBYyxFQUFFLE1BQU0sZUFBZSxDQUFDO0FBRTFELE9BQU8sRUFBRSx3QkFBd0IsRUFBRSxNQUFNLG1CQUFtQixDQUFDO0FBRTdELE9BQU8sRUFBRSxnQkFBZ0IsRUFBRSxhQUFhLEVBQUUsb0JBQW9CLEVBQUUsTUFBTSxnQkFBZ0IsQ0FBQyIsInNvdXJjZXNDb250ZW50IjpbIi8vIEdFTkVSQVRFRCBieSBwcm9qZW4gd2F0Y2ggLSBETyBOT1QgRURJVC5cbi8vIFJlZ2VuZXJhdGVkIGZyb20gdGhlIGV4cG9ydGluZyBtb2R1bGVzIGluIC4vc3JjLlxuLy8gSGFuZCBlZGl0cyBhcmUgb3ZlcndyaXR0ZW4gb24gdGhlIG5leHQgd2F0Y2g7IHRoaXMgZmlsZSBpcyByZWFkLW9ubHkuXG5cbmV4cG9ydCBjb25zdCBQQUNLQUdFX0lERU5USUZJRVIgPSBcIkBkYngtdG9vbHMvZ2VuaWVcIjtcbmV4cG9ydCBjb25zdCBQQUNLQUdFX1ZFUlNJT04gPSBcIjAuOS4xODdcIjtcbmV4cG9ydCAqIGFzIGFnZW50TW9kZSBmcm9tIFwiLi9zcmMvYWdlbnQtbW9kZS50c1wiO1xuZXhwb3J0ICogYXMgY2hhdCBmcm9tIFwiLi9zcmMvY2hhdC50c1wiO1xuZXhwb3J0ICogYXMgZmVlZGJhY2sgZnJvbSBcIi4vc3JjL2ZlZWRiYWNrLnRzXCI7XG5leHBvcnQgKiBhcyBzcGFjZSBmcm9tIFwiLi9zcmMvc3BhY2UudHNcIjtcbmV4cG9ydCB7IGdlbmllQWdlbnRNb2RlQ2hhdCwgbGlzdEFnZW50TW9kZUNvbnZlcnNhdGlvbiwgaXNBZ2VudE1vZGVVbmF2YWlsYWJsZSB9IGZyb20gXCIuL3NyYy9hZ2VudC1tb2RlLnRzXCI7XG5leHBvcnQgdHlwZSB7IEdlbmllQWdlbnRNb2RlT3B0aW9ucyB9IGZyb20gXCIuL3NyYy9hZ2VudC1tb2RlLnRzXCI7XG5leHBvcnQgeyBnZW5pZUNoYXQsIGdlbmllRXZlbnRDaGF0IH0gZnJvbSBcIi4vc3JjL2NoYXQudHNcIjtcbmV4cG9ydCB0eXBlIHsgR2VuaWVDaGF0T3B0aW9ucyB9IGZyb20gXCIuL3NyYy9jaGF0LnRzXCI7XG5leHBvcnQgeyBzZW5kR2VuaWVNZXNzYWdlRmVlZGJhY2sgfSBmcm9tIFwiLi9zcmMvZmVlZGJhY2sudHNcIjtcbmV4cG9ydCB0eXBlIHsgU2VuZEdlbmllTWVzc2FnZUZlZWRiYWNrT3B0aW9ucyB9IGZyb20gXCIuL3NyYy9mZWVkYmFjay50c1wiO1xuZXhwb3J0IHsgY3JlYXRlR2VuaWVTcGFjZSwgZ2V0R2VuaWVTcGFjZSwgZ2VuaWVTYW1wbGVRdWVzdGlvbnMgfSBmcm9tIFwiLi9zcmMvc3BhY2UudHNcIjtcbmV4cG9ydCB0eXBlIHsgR2V0R2VuaWVTcGFjZU9wdGlvbnMsIENyZWF0ZUdlbmllU3BhY2VPcHRpb25zIH0gZnJvbSBcIi4vc3JjL3NwYWNlLnRzXCI7XG4iXX0=
+export { upscaleGenieVisualizationPng, downloadGenieVisualization } from "./src/visualization.js";
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSwyQ0FBMkM7QUFDM0MsbURBQW1EO0FBQ25ELHdFQUF3RTtBQUV4RSxNQUFNLENBQUMsTUFBTSxrQkFBa0IsR0FBRyxrQkFBa0IsQ0FBQztBQUNyRCxNQUFNLENBQUMsTUFBTSxlQUFlLEdBQUcsU0FBUyxDQUFDO0FBQ3pDLE9BQU8sS0FBSyxTQUFTLE1BQU0scUJBQXFCLENBQUM7QUFDakQsT0FBTyxLQUFLLElBQUksTUFBTSxlQUFlLENBQUM7QUFDdEMsT0FBTyxLQUFLLFFBQVEsTUFBTSxtQkFBbUIsQ0FBQztBQUM5QyxPQUFPLEtBQUssS0FBSyxNQUFNLGdCQUFnQixDQUFDO0FBQ3hDLE9BQU8sS0FBSyxhQUFhLE1BQU0sd0JBQXdCLENBQUM7QUFDeEQsT0FBTyxFQUFFLGtCQUFrQixFQUFFLHlCQUF5QixFQUFFLHNCQUFzQixFQUFFLE1BQU0scUJBQXFCLENBQUM7QUFFNUcsT0FBTyxFQUFFLFNBQVMsRUFBRSxjQUFjLEVBQUUsTUFBTSxlQUFlLENBQUM7QUFFMUQsT0FBTyxFQUFFLHdCQUF3QixFQUFFLE1BQU0sbUJBQW1CLENBQUM7QUFFN0QsT0FBTyxFQUFFLGdCQUFnQixFQUFFLGFBQWEsRUFBRSxvQkFBb0IsRUFBRSxNQUFNLGdCQUFnQixDQUFDO0FBRXZGLE9BQU8sRUFBRSw0QkFBNEIsRUFBRSwwQkFBMEIsRUFBRSxNQUFNLHdCQUF3QixDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLy8gR0VORVJBVEVEIGJ5IHByb2plbiB3YXRjaCAtIERPIE5PVCBFRElULlxuLy8gUmVnZW5lcmF0ZWQgZnJvbSB0aGUgZXhwb3J0aW5nIG1vZHVsZXMgaW4gLi9zcmMuXG4vLyBIYW5kIGVkaXRzIGFyZSBvdmVyd3JpdHRlbiBvbiB0aGUgbmV4dCB3YXRjaDsgdGhpcyBmaWxlIGlzIHJlYWQtb25seS5cblxuZXhwb3J0IGNvbnN0IFBBQ0tBR0VfSURFTlRJRklFUiA9IFwiQGRieC10b29scy9nZW5pZVwiO1xuZXhwb3J0IGNvbnN0IFBBQ0tBR0VfVkVSU0lPTiA9IFwiMC45LjE4OVwiO1xuZXhwb3J0ICogYXMgYWdlbnRNb2RlIGZyb20gXCIuL3NyYy9hZ2VudC1tb2RlLnRzXCI7XG5leHBvcnQgKiBhcyBjaGF0IGZyb20gXCIuL3NyYy9jaGF0LnRzXCI7XG5leHBvcnQgKiBhcyBmZWVkYmFjayBmcm9tIFwiLi9zcmMvZmVlZGJhY2sudHNcIjtcbmV4cG9ydCAqIGFzIHNwYWNlIGZyb20gXCIuL3NyYy9zcGFjZS50c1wiO1xuZXhwb3J0ICogYXMgdmlzdWFsaXphdGlvbiBmcm9tIFwiLi9zcmMvdmlzdWFsaXphdGlvbi50c1wiO1xuZXhwb3J0IHsgZ2VuaWVBZ2VudE1vZGVDaGF0LCBsaXN0QWdlbnRNb2RlQ29udmVyc2F0aW9uLCBpc0FnZW50TW9kZVVuYXZhaWxhYmxlIH0gZnJvbSBcIi4vc3JjL2FnZW50LW1vZGUudHNcIjtcbmV4cG9ydCB0eXBlIHsgR2VuaWVBZ2VudE1vZGVPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2FnZW50LW1vZGUudHNcIjtcbmV4cG9ydCB7IGdlbmllQ2hhdCwgZ2VuaWVFdmVudENoYXQgfSBmcm9tIFwiLi9zcmMvY2hhdC50c1wiO1xuZXhwb3J0IHR5cGUgeyBHZW5pZUNoYXRPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2NoYXQudHNcIjtcbmV4cG9ydCB7IHNlbmRHZW5pZU1lc3NhZ2VGZWVkYmFjayB9IGZyb20gXCIuL3NyYy9mZWVkYmFjay50c1wiO1xuZXhwb3J0IHR5cGUgeyBTZW5kR2VuaWVNZXNzYWdlRmVlZGJhY2tPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2ZlZWRiYWNrLnRzXCI7XG5leHBvcnQgeyBjcmVhdGVHZW5pZVNwYWNlLCBnZXRHZW5pZVNwYWNlLCBnZW5pZVNhbXBsZVF1ZXN0aW9ucyB9IGZyb20gXCIuL3NyYy9zcGFjZS50c1wiO1xuZXhwb3J0IHR5cGUgeyBHZXRHZW5pZVNwYWNlT3B0aW9ucywgQ3JlYXRlR2VuaWVTcGFjZU9wdGlvbnMgfSBmcm9tIFwiLi9zcmMvc3BhY2UudHNcIjtcbmV4cG9ydCB7IHVwc2NhbGVHZW5pZVZpc3VhbGl6YXRpb25QbmcsIGRvd25sb2FkR2VuaWVWaXN1YWxpemF0aW9uIH0gZnJvbSBcIi4vc3JjL3Zpc3VhbGl6YXRpb24udHNcIjtcbmV4cG9ydCB0eXBlIHsgRG93bmxvYWRHZW5pZVZpc3VhbGl6YXRpb25PcHRpb25zIH0gZnJvbSBcIi4vc3JjL3Zpc3VhbGl6YXRpb24udHNcIjtcbiJdfQ==

package/lib/src/visualization.d.ts

@@ -0,0 +1,22 @@
+import { type WorkspaceClient } from "@databricks/appkit";
+import { type ContextLike } from "@dbx-tools/appkit";
+import { type GenieVisualizationRequest } from "@dbx-tools/shared-genie";
+/**
+ * Resize a Genie screenshot to four times its source dimensions with Lanczos3,
+ * then apply a mild unsharp mask. Returns the original bytes when the optional
+ * Sharp runtime is unavailable or rejects the image.
+ */
+export declare function upscaleGenieVisualizationPng(inputBytes: Uint8Array): Promise<Uint8Array>;
+/** Identity and cancellation for a Genie visualization download. */
+export interface DownloadGenieVisualizationOptions {
+    /** Workspace client whose identity owns the conversation. */
+    workspaceClient?: WorkspaceClient;
+    /** External cancellation for the request and image body. */
+    context?: ContextLike;
+}
+/**
+ * Download Genie's PNG and apply optional raster cleanup before returning it.
+ * The SDK's raw transport preserves binary responses and its normal authentication;
+ * source-chart exports should still be preferred when they are available.
+ */
+export declare function downloadGenieVisualization(input: GenieVisualizationRequest, options?: DownloadGenieVisualizationOptions): Promise<Uint8Array>;

package/lib/src/visualization.js

@@ -0,0 +1,94 @@
+import { createWorkspaceClient } from "@databricks/appkit";
+import { toContext } from "@dbx-tools/appkit";
+import { isRecord, log, toError } from "@dbx-tools/shared-core";
+import { genieVisualizationDownloadPath, } from "@dbx-tools/shared-genie";
+const logger = log.logger("genie/visualization");
+const GENIE_PNG_SCALE = 4;
+const GENIE_PNG_SHARPEN = { sigma: 1.1, m1: 0.9, m2: 0.02 };
+let sharpFactoryPromise;
+function isByteStream(value) {
+    return (isRecord(value) && typeof value.getReader === "function" && typeof value.cancel === "function");
+}
+async function loadSharp() {
+    sharpFactoryPromise ??= import("sharp")
+        .then((module) => module.default)
+        .catch((error) => {
+        logger.debug("PNG enhancement unavailable; serving the original Genie image", {
+            error: toError(error),
+        });
+        return undefined;
+    });
+    return sharpFactoryPromise;
+}
+/**
+ * Resize a Genie screenshot to four times its source dimensions with Lanczos3,
+ * then apply a mild unsharp mask. Returns the original bytes when the optional
+ * Sharp runtime is unavailable or rejects the image.
+ */
+export async function upscaleGenieVisualizationPng(inputBytes) {
+    const sharp = await loadSharp();
+    if (!sharp)
+        return inputBytes;
+    try {
+        const input = Buffer.from(inputBytes);
+        const metadata = await sharp(input).metadata();
+        if (!metadata.width || !metadata.height) {
+            throw new Error("Could not read Genie visualization dimensions");
+        }
+        const output = await sharp(input)
+            .resize({
+            width: metadata.width * GENIE_PNG_SCALE,
+            height: metadata.height * GENIE_PNG_SCALE,
+            kernel: sharp.kernel.lanczos3,
+            fit: "fill",
+        })
+            .sharpen(GENIE_PNG_SHARPEN)
+            .png()
+            .toBuffer();
+        return new Uint8Array(output);
+    }
+    catch (error) {
+        logger.warn("PNG enhancement failed; serving the original Genie image", {
+            error: toError(error),
+        });
+        return inputBytes;
+    }
+}
+/**
+ * Download Genie's PNG and apply optional raster cleanup before returning it.
+ * The SDK's raw transport preserves binary responses and its normal authentication;
+ * source-chart exports should still be preferred when they are available.
+ */
+export async function downloadGenieVisualization(input, options = {}) {
+    const client = options.workspaceClient ?? createWorkspaceClient();
+    const response = await client.apiClient.request({
+        path: genieVisualizationDownloadPath(input),
+        method: "GET",
+        headers: new Headers({ accept: "image/png" }),
+        responseHeaders: ["content-type"],
+        raw: true,
+    }, options.context ? toContext(options.context) : undefined);
+    if (!isRecord(response) ||
+        typeof response["content-type"] !== "string" ||
+        response["content-type"].split(";")[0]?.trim().toLowerCase() !== "image/png" ||
+        !isByteStream(response.contents)) {
+        if (isRecord(response) && isByteStream(response.contents)) {
+            await response.contents.cancel().catch(() => undefined);
+        }
+        throw new TypeError("Genie did not return a PNG visualization");
+    }
+    const bytes = new Uint8Array(await new Response(response.contents).arrayBuffer());
+    if (bytes.length < 8 ||
+        bytes[0] !== 0x89 ||
+        bytes[1] !== 0x50 ||
+        bytes[2] !== 0x4e ||
+        bytes[3] !== 0x47 ||
+        bytes[4] !== 0x0d ||
+        bytes[5] !== 0x0a ||
+        bytes[6] !== 0x1a ||
+        bytes[7] !== 0x0a) {
+        throw new TypeError("Genie returned an invalid PNG visualization");
+    }
+    return upscaleGenieVisualizationPng(bytes);
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidmlzdWFsaXphdGlvbi5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy92aXN1YWxpemF0aW9uLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxxQkFBcUIsRUFBd0IsTUFBTSxvQkFBb0IsQ0FBQztBQUNqRixPQUFPLEVBQW9CLFNBQVMsRUFBRSxNQUFNLG1CQUFtQixDQUFDO0FBQ2hFLE9BQU8sRUFBRSxRQUFRLEVBQUUsR0FBRyxFQUFFLE9BQU8sRUFBRSxNQUFNLHdCQUF3QixDQUFDO0FBQ2hFLE9BQU8sRUFFTCw4QkFBOEIsR0FDL0IsTUFBTSx5QkFBeUIsQ0FBQztBQUVqQyxNQUFNLE1BQU0sR0FBRyxHQUFHLENBQUMsTUFBTSxDQUFDLHFCQUFxQixDQUFDLENBQUM7QUFDakQsTUFBTSxlQUFlLEdBQUcsQ0FBQyxDQUFDO0FBQzFCLE1BQU0saUJBQWlCLEdBQUcsRUFBRSxLQUFLLEVBQUUsR0FBRyxFQUFFLEVBQUUsRUFBRSxHQUFHLEVBQUUsRUFBRSxFQUFFLElBQUksRUFBVyxDQUFDO0FBSXJFLElBQUksbUJBQWtFLENBQUM7QUFFdkUsU0FBUyxZQUFZLENBQUMsS0FBYztJQUNsQyxPQUFPLENBQ0wsUUFBUSxDQUFDLEtBQUssQ0FBQyxJQUFJLE9BQU8sS0FBSyxDQUFDLFNBQVMsS0FBSyxVQUFVLElBQUksT0FBTyxLQUFLLENBQUMsTUFBTSxLQUFLLFVBQVUsQ0FDL0YsQ0FBQztBQUNKLENBQUM7QUFFRCxLQUFLLFVBQVUsU0FBUztJQUN0QixtQkFBbUIsS0FBSyxNQUFNLENBQUMsT0FBTyxDQUFDO1NBQ3BDLElBQUksQ0FBQyxDQUFDLE1BQU0sRUFBRSxFQUFFLENBQUMsTUFBTSxDQUFDLE9BQU8sQ0FBQztTQUNoQyxLQUFLLENBQUMsQ0FBQyxLQUFjLEVBQUUsRUFBRTtRQUN4QixNQUFNLENBQUMsS0FBSyxDQUFDLCtEQUErRCxFQUFFO1lBQzVFLEtBQUssRUFBRSxPQUFPLENBQUMsS0FBSyxDQUFDO1NBQ3RCLENBQUMsQ0FBQztRQUNILE9BQU8sU0FBUyxDQUFDO0lBQ25CLENBQUMsQ0FBQyxDQUFDO0lBQ0wsT0FBTyxtQkFBbUIsQ0FBQztBQUM3QixDQUFDO0FBRUQ7Ozs7R0FJRztBQUNILE1BQU0sQ0FBQyxLQUFLLFVBQVUsNEJBQTRCLENBQUMsVUFBc0I7SUFDdkUsTUFBTSxLQUFLLEdBQUcsTUFBTSxTQUFTLEVBQUUsQ0FBQztJQUNoQyxJQUFJLENBQUMsS0FBSztRQUFFLE9BQU8sVUFBVSxDQUFDO0lBQzlCLElBQUksQ0FBQztRQUNILE1BQU0sS0FBSyxHQUFHLE1BQU0sQ0FBQyxJQUFJLENBQUMsVUFBVSxDQUFDLENBQUM7UUFDdEMsTUFBTSxRQUFRLEdBQUcsTUFBTSxLQUFLLENBQUMsS0FBSyxDQUFDLENBQUMsUUFBUSxFQUFFLENBQUM7UUFDL0MsSUFBSSxDQUFDLFFBQVEsQ0FBQyxLQUFLLElBQUksQ0FBQyxRQUFRLENBQUMsTUFBTSxFQUFFLENBQUM7WUFDeEMsTUFBTSxJQUFJLEtBQUssQ0FBQywrQ0FBK0MsQ0FBQyxDQUFDO1FBQ25FLENBQUM7UUFDRCxNQUFNLE1BQU0sR0FBRyxNQUFNLEtBQUssQ0FBQyxLQUFLLENBQUM7YUFDOUIsTUFBTSxDQUFDO1lBQ04sS0FBSyxFQUFFLFFBQVEsQ0FBQyxLQUFLLEdBQUcsZUFBZTtZQUN2QyxNQUFNLEVBQUUsUUFBUSxDQUFDLE1BQU0sR0FBRyxlQUFlO1lBQ3pDLE1BQU0sRUFBRSxLQUFLLENBQUMsTUFBTSxDQUFDLFFBQVE7WUFDN0IsR0FBRyxFQUFFLE1BQU07U0FDWixDQUFDO2FBQ0QsT0FBTyxDQUFDLGlCQUFpQixDQUFDO2FBQzFCLEdBQUcsRUFBRTthQUNMLFFBQVEsRUFBRSxDQUFDO1FBQ2QsT0FBTyxJQUFJLFVBQVUsQ0FBQyxNQUFNLENBQUMsQ0FBQztJQUNoQyxDQUFDO0lBQUMsT0FBTyxLQUFLLEVBQUUsQ0FBQztRQUNmLE1BQU0sQ0FBQyxJQUFJLENBQUMsMERBQTBELEVBQUU7WUFDdEUsS0FBSyxFQUFFLE9BQU8sQ0FBQyxLQUFLLENBQUM7U0FDdEIsQ0FBQyxDQUFDO1FBQ0gsT0FBTyxVQUFVLENBQUM7SUFDcEIsQ0FBQztBQUNILENBQUM7QUFVRDs7OztHQUlHO0FBQ0gsTUFBTSxDQUFDLEtBQUssVUFBVSwwQkFBMEIsQ0FDOUMsS0FBZ0MsRUFDaEMsVUFBNkMsRUFBRTtJQUUvQyxNQUFNLE1BQU0sR0FBRyxPQUFPLENBQUMsZUFBZSxJQUFJLHFCQUFxQixFQUFFLENBQUM7SUFDbEUsTUFBTSxRQUFRLEdBQUcsTUFBTSxNQUFNLENBQUMsU0FBUyxDQUFDLE9BQU8sQ0FDN0M7UUFDRSxJQUFJLEVBQUUsOEJBQThCLENBQUMsS0FBSyxDQUFDO1FBQzNDLE1BQU0sRUFBRSxLQUFLO1FBQ2IsT0FBTyxFQUFFLElBQUksT0FBTyxDQUFDLEVBQUUsTUFBTSxFQUFFLFdBQVcsRUFBRSxDQUFDO1FBQzdDLGVBQWUsRUFBRSxDQUFDLGNBQWMsQ0FBQztRQUNqQyxHQUFHLEVBQUUsSUFBSTtLQUNWLEVBQ0QsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsU0FBUyxDQUFDLE9BQU8sQ0FBQyxPQUFPLENBQUMsQ0FBQyxDQUFDLENBQUMsU0FBUyxDQUN6RCxDQUFDO0lBQ0YsSUFDRSxDQUFDLFFBQVEsQ0FBQyxRQUFRLENBQUM7UUFDbkIsT0FBTyxRQUFRLENBQUMsY0FBYyxDQUFDLEtBQUssUUFBUTtRQUM1QyxRQUFRLENBQUMsY0FBYyxDQUFDLENBQUMsS0FBSyxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxFQUFFLElBQUksRUFBRSxDQUFDLFdBQVcsRUFBRSxLQUFLLFdBQVc7UUFDNUUsQ0FBQyxZQUFZLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQyxFQUNoQyxDQUFDO1FBQ0QsSUFBSSxRQUFRLENBQUMsUUFBUSxDQUFDLElBQUksWUFBWSxDQUFDLFFBQVEsQ0FBQyxRQUFRLENBQUMsRUFBRSxDQUFDO1lBQzFELE1BQU0sUUFBUSxDQUFDLFFBQVEsQ0FBQyxNQUFNLEVBQUUsQ0FBQyxLQUFLLENBQUMsR0FBRyxFQUFFLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDMUQsQ0FBQztRQUNELE1BQU0sSUFBSSxTQUFTLENBQUMsMENBQTBDLENBQUMsQ0FBQztJQUNsRSxDQUFDO0lBQ0QsTUFBTSxLQUFLLEdBQUcsSUFBSSxVQUFVLENBQUMsTUFBTSxJQUFJLFFBQVEsQ0FBQyxRQUFRLENBQUMsUUFBUSxDQUFDLENBQUMsV0FBVyxFQUFFLENBQUMsQ0FBQztJQUNsRixJQUNFLEtBQUssQ0FBQyxNQUFNLEdBQUcsQ0FBQztRQUNoQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSTtRQUNqQixLQUFLLENBQUMsQ0FBQyxDQUFDLEtBQUssSUFBSSxFQUNqQixDQUFDO1FBQ0QsTUFBTSxJQUFJLFNBQVMsQ0FBQyw2Q0FBNkMsQ0FBQyxDQUFDO0lBQ3JFLENBQUM7SUFDRCxPQUFPLDRCQUE0QixDQUFDLEtBQUssQ0FBQyxDQUFDO0FBQzdDLENBQUMiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyBjcmVhdGVXb3Jrc3BhY2VDbGllbnQsIHR5cGUgV29ya3NwYWNlQ2xpZW50IH0gZnJvbSBcIkBkYXRhYnJpY2tzL2FwcGtpdFwiO1xuaW1wb3J0IHsgdHlwZSBDb250ZXh0TGlrZSwgdG9Db250ZXh0IH0gZnJvbSBcIkBkYngtdG9vbHMvYXBwa2l0XCI7XG5pbXBvcnQgeyBpc1JlY29yZCwgbG9nLCB0b0Vycm9yIH0gZnJvbSBcIkBkYngtdG9vbHMvc2hhcmVkLWNvcmVcIjtcbmltcG9ydCB7XG4gIHR5cGUgR2VuaWVWaXN1YWxpemF0aW9uUmVxdWVzdCxcbiAgZ2VuaWVWaXN1YWxpemF0aW9uRG93bmxvYWRQYXRoLFxufSBmcm9tIFwiQGRieC10b29scy9zaGFyZWQtZ2VuaWVcIjtcblxuY29uc3QgbG9nZ2VyID0gbG9nLmxvZ2dlcihcImdlbmllL3Zpc3VhbGl6YXRpb25cIik7XG5jb25zdCBHRU5JRV9QTkdfU0NBTEUgPSA0O1xuY29uc3QgR0VOSUVfUE5HX1NIQVJQRU4gPSB7IHNpZ21hOiAxLjEsIG0xOiAwLjksIG0yOiAwLjAyIH0gYXMgY29uc3Q7XG5cbnR5cGUgU2hhcnBGYWN0b3J5ID0gKHR5cGVvZiBpbXBvcnQoXCJzaGFycFwiKSlbXCJkZWZhdWx0XCJdO1xuXG5sZXQgc2hhcnBGYWN0b3J5UHJvbWlzZTogUHJvbWlzZTxTaGFycEZhY3RvcnkgfCB1bmRlZmluZWQ+IHwgdW5kZWZpbmVkO1xuXG5mdW5jdGlvbiBpc0J5dGVTdHJlYW0odmFsdWU6IHVua25vd24pOiB2YWx1ZSBpcyBSZWFkYWJsZVN0cmVhbTxVaW50OEFycmF5PiB7XG4gIHJldHVybiAoXG4gICAgaXNSZWNvcmQodmFsdWUpICYmIHR5cGVvZiB2YWx1ZS5nZXRSZWFkZXIgPT09IFwiZnVuY3Rpb25cIiAmJiB0eXBlb2YgdmFsdWUuY2FuY2VsID09PSBcImZ1bmN0aW9uXCJcbiAgKTtcbn1cblxuYXN5bmMgZnVuY3Rpb24gbG9hZFNoYXJwKCk6IFByb21pc2U8U2hhcnBGYWN0b3J5IHwgdW5kZWZpbmVkPiB7XG4gIHNoYXJwRmFjdG9yeVByb21pc2UgPz89IGltcG9ydChcInNoYXJwXCIpXG4gICAgLnRoZW4oKG1vZHVsZSkgPT4gbW9kdWxlLmRlZmF1bHQpXG4gICAgLmNhdGNoKChlcnJvcjogdW5rbm93bikgPT4ge1xuICAgICAgbG9nZ2VyLmRlYnVnKFwiUE5HIGVuaGFuY2VtZW50IHVuYXZhaWxhYmxlOyBzZXJ2aW5nIHRoZSBvcmlnaW5hbCBHZW5pZSBpbWFnZVwiLCB7XG4gICAgICAgIGVycm9yOiB0b0Vycm9yKGVycm9yKSxcbiAgICAgIH0pO1xuICAgICAgcmV0dXJuIHVuZGVmaW5lZDtcbiAgICB9KTtcbiAgcmV0dXJuIHNoYXJwRmFjdG9yeVByb21pc2U7XG59XG5cbi8qKlxuICogUmVzaXplIGEgR2VuaWUgc2NyZWVuc2hvdCB0byBmb3VyIHRpbWVzIGl0cyBzb3VyY2UgZGltZW5zaW9ucyB3aXRoIExhbmN6b3MzLFxuICogdGhlbiBhcHBseSBhIG1pbGQgdW5zaGFycCBtYXNrLiBSZXR1cm5zIHRoZSBvcmlnaW5hbCBieXRlcyB3aGVuIHRoZSBvcHRpb25hbFxuICogU2hhcnAgcnVudGltZSBpcyB1bmF2YWlsYWJsZSBvciByZWplY3RzIHRoZSBpbWFnZS5cbiAqL1xuZXhwb3J0IGFzeW5jIGZ1bmN0aW9uIHVwc2NhbGVHZW5pZVZpc3VhbGl6YXRpb25QbmcoaW5wdXRCeXRlczogVWludDhBcnJheSk6IFByb21pc2U8VWludDhBcnJheT4ge1xuICBjb25zdCBzaGFycCA9IGF3YWl0IGxvYWRTaGFycCgpO1xuICBpZiAoIXNoYXJwKSByZXR1cm4gaW5wdXRCeXRlcztcbiAgdHJ5IHtcbiAgICBjb25zdCBpbnB1dCA9IEJ1ZmZlci5mcm9tKGlucHV0Qnl0ZXMpO1xuICAgIGNvbnN0IG1ldGFkYXRhID0gYXdhaXQgc2hhcnAoaW5wdXQpLm1ldGFkYXRhKCk7XG4gICAgaWYgKCFtZXRhZGF0YS53aWR0aCB8fCAhbWV0YWRhdGEuaGVpZ2h0KSB7XG4gICAgICB0aHJvdyBuZXcgRXJyb3IoXCJDb3VsZCBub3QgcmVhZCBHZW5pZSB2aXN1YWxpemF0aW9uIGRpbWVuc2lvbnNcIik7XG4gICAgfVxuICAgIGNvbnN0IG91dHB1dCA9IGF3YWl0IHNoYXJwKGlucHV0KVxuICAgICAgLnJlc2l6ZSh7XG4gICAgICAgIHdpZHRoOiBtZXRhZGF0YS53aWR0aCAqIEdFTklFX1BOR19TQ0FMRSxcbiAgICAgICAgaGVpZ2h0OiBtZXRhZGF0YS5oZWlnaHQgKiBHRU5JRV9QTkdfU0NBTEUsXG4gICAgICAgIGtlcm5lbDogc2hhcnAua2VybmVsLmxhbmN6b3MzLFxuICAgICAgICBmaXQ6IFwiZmlsbFwiLFxuICAgICAgfSlcbiAgICAgIC5zaGFycGVuKEdFTklFX1BOR19TSEFSUEVOKVxuICAgICAgLnBuZygpXG4gICAgICAudG9CdWZmZXIoKTtcbiAgICByZXR1cm4gbmV3IFVpbnQ4QXJyYXkob3V0cHV0KTtcbiAgfSBjYXRjaCAoZXJyb3IpIHtcbiAgICBsb2dnZXIud2FybihcIlBORyBlbmhhbmNlbWVudCBmYWlsZWQ7IHNlcnZpbmcgdGhlIG9yaWdpbmFsIEdlbmllIGltYWdlXCIsIHtcbiAgICAgIGVycm9yOiB0b0Vycm9yKGVycm9yKSxcbiAgICB9KTtcbiAgICByZXR1cm4gaW5wdXRCeXRlcztcbiAgfVxufVxuXG4vKiogSWRlbnRpdHkgYW5kIGNhbmNlbGxhdGlvbiBmb3IgYSBHZW5pZSB2aXN1YWxpemF0aW9uIGRvd25sb2FkLiAqL1xuZXhwb3J0IGludGVyZmFjZSBEb3dubG9hZEdlbmllVmlzdWFsaXphdGlvbk9wdGlvbnMge1xuICAvKiogV29ya3NwYWNlIGNsaWVudCB3aG9zZSBpZGVudGl0eSBvd25zIHRoZSBjb252ZXJzYXRpb24uICovXG4gIHdvcmtzcGFjZUNsaWVudD86IFdvcmtzcGFjZUNsaWVudDtcbiAgLyoqIEV4dGVybmFsIGNhbmNlbGxhdGlvbiBmb3IgdGhlIHJlcXVlc3QgYW5kIGltYWdlIGJvZHkuICovXG4gIGNvbnRleHQ/OiBDb250ZXh0TGlrZTtcbn1cblxuLyoqXG4gKiBEb3dubG9hZCBHZW5pZSdzIFBORyBhbmQgYXBwbHkgb3B0aW9uYWwgcmFzdGVyIGNsZWFudXAgYmVmb3JlIHJldHVybmluZyBpdC5cbiAqIFRoZSBTREsncyByYXcgdHJhbnNwb3J0IHByZXNlcnZlcyBiaW5hcnkgcmVzcG9uc2VzIGFuZCBpdHMgbm9ybWFsIGF1dGhlbnRpY2F0aW9uO1xuICogc291cmNlLWNoYXJ0IGV4cG9ydHMgc2hvdWxkIHN0aWxsIGJlIHByZWZlcnJlZCB3aGVuIHRoZXkgYXJlIGF2YWlsYWJsZS5cbiAqL1xuZXhwb3J0IGFzeW5jIGZ1bmN0aW9uIGRvd25sb2FkR2VuaWVWaXN1YWxpemF0aW9uKFxuICBpbnB1dDogR2VuaWVWaXN1YWxpemF0aW9uUmVxdWVzdCxcbiAgb3B0aW9uczogRG93bmxvYWRHZW5pZVZpc3VhbGl6YXRpb25PcHRpb25zID0ge30sXG4pOiBQcm9taXNlPFVpbnQ4QXJyYXk+IHtcbiAgY29uc3QgY2xpZW50ID0gb3B0aW9ucy53b3Jrc3BhY2VDbGllbnQgPz8gY3JlYXRlV29ya3NwYWNlQ2xpZW50KCk7XG4gIGNvbnN0IHJlc3BvbnNlID0gYXdhaXQgY2xpZW50LmFwaUNsaWVudC5yZXF1ZXN0KFxuICAgIHtcbiAgICAgIHBhdGg6IGdlbmllVmlzdWFsaXphdGlvbkRvd25sb2FkUGF0aChpbnB1dCksXG4gICAgICBtZXRob2Q6IFwiR0VUXCIsXG4gICAgICBoZWFkZXJzOiBuZXcgSGVhZGVycyh7IGFjY2VwdDogXCJpbWFnZS9wbmdcIiB9KSxcbiAgICAgIHJlc3BvbnNlSGVhZGVyczogW1wiY29udGVudC10eXBlXCJdLFxuICAgICAgcmF3OiB0cnVlLFxuICAgIH0sXG4gICAgb3B0aW9ucy5jb250ZXh0ID8gdG9Db250ZXh0KG9wdGlvbnMuY29udGV4dCkgOiB1bmRlZmluZWQsXG4gICk7XG4gIGlmIChcbiAgICAhaXNSZWNvcmQocmVzcG9uc2UpIHx8XG4gICAgdHlwZW9mIHJlc3BvbnNlW1wiY29udGVudC10eXBlXCJdICE9PSBcInN0cmluZ1wiIHx8XG4gICAgcmVzcG9uc2VbXCJjb250ZW50LXR5cGVcIl0uc3BsaXQoXCI7XCIpWzBdPy50cmltKCkudG9Mb3dlckNhc2UoKSAhPT0gXCJpbWFnZS9wbmdcIiB8fFxuICAgICFpc0J5dGVTdHJlYW0ocmVzcG9uc2UuY29udGVudHMpXG4gICkge1xuICAgIGlmIChpc1JlY29yZChyZXNwb25zZSkgJiYgaXNCeXRlU3RyZWFtKHJlc3BvbnNlLmNvbnRlbnRzKSkge1xuICAgICAgYXdhaXQgcmVzcG9uc2UuY29udGVudHMuY2FuY2VsKCkuY2F0Y2goKCkgPT4gdW5kZWZpbmVkKTtcbiAgICB9XG4gICAgdGhyb3cgbmV3IFR5cGVFcnJvcihcIkdlbmllIGRpZCBub3QgcmV0dXJuIGEgUE5HIHZpc3VhbGl6YXRpb25cIik7XG4gIH1cbiAgY29uc3QgYnl0ZXMgPSBuZXcgVWludDhBcnJheShhd2FpdCBuZXcgUmVzcG9uc2UocmVzcG9uc2UuY29udGVudHMpLmFycmF5QnVmZmVyKCkpO1xuICBpZiAoXG4gICAgYnl0ZXMubGVuZ3RoIDwgOCB8fFxuICAgIGJ5dGVzWzBdICE9PSAweDg5IHx8XG4gICAgYnl0ZXNbMV0gIT09IDB4NTAgfHxcbiAgICBieXRlc1syXSAhPT0gMHg0ZSB8fFxuICAgIGJ5dGVzWzNdICE9PSAweDQ3IHx8XG4gICAgYnl0ZXNbNF0gIT09IDB4MGQgfHxcbiAgICBieXRlc1s1XSAhPT0gMHgwYSB8fFxuICAgIGJ5dGVzWzZdICE9PSAweDFhIHx8XG4gICAgYnl0ZXNbN10gIT09IDB4MGFcbiAgKSB7XG4gICAgdGhyb3cgbmV3IFR5cGVFcnJvcihcIkdlbmllIHJldHVybmVkIGFuIGludmFsaWQgUE5HIHZpc3VhbGl6YXRpb25cIik7XG4gIH1cbiAgcmV0dXJuIHVwc2NhbGVHZW5pZVZpc3VhbGl6YXRpb25QbmcoYnl0ZXMpO1xufVxuIl19

package/lib/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["../src/agent-mode.ts","../src/chat.ts","../src/feedback.ts","../src/space.ts","../index.ts"],"version":"5.9.3"}
+{"root":["../src/agent-mode.ts","../src/chat.ts","../src/feedback.ts","../src/space.ts","../src/visualization.ts","../index.ts"],"version":"5.9.3"}

package/package.json

@@ -20,13 +20,17 @@
   },
   "devDependencies": {
     "@types/node": "^24.6.0",
+    "sharp": "^0.35.5",
     "typescript": "^5.9.3"
   },
+  "peerDependencies": {
+    "sharp": "^0.35.5"
+  },
   "dependencies": {
     "@databricks/appkit": "0.81.0",
-    "@dbx-tools/appkit": "^0.9.187",
-    "@dbx-tools/shared-core": "^0.9.187",
-    "@dbx-tools/shared-genie": "^0.9.187"
+    "@dbx-tools/appkit": "0.9.189",
+    "@dbx-tools/shared-core": "0.9.189",
+    "@dbx-tools/shared-genie": "0.9.189"
   },
   "main": "./lib/index.js",
   "license": "Apache-2.0",
@@ -42,7 +46,7 @@
       "./package.json": "./package.json"
     }
   },
-  "version": "0.9.187",
+  "version": "0.9.189",
   "types": "./lib/index.d.ts",
   "dbxToolsConfig": {
     "tags": [
@@ -62,5 +66,10 @@
     "src",
     "lib"
   ],
+  "peerDependenciesMeta": {
+    "sharp": {
+      "optional": true
+    }
+  },
   "//": "~~ Generated by projen. To modify, edit .projenrc.js and run \"bunx projen\"."
 }

package/src/visualization.ts

@@ -0,0 +1,122 @@
+import { createWorkspaceClient, type WorkspaceClient } from "@databricks/appkit";
+import { type ContextLike, toContext } from "@dbx-tools/appkit";
+import { isRecord, log, toError } from "@dbx-tools/shared-core";
+import {
+  type GenieVisualizationRequest,
+  genieVisualizationDownloadPath,
+} from "@dbx-tools/shared-genie";
+
+const logger = log.logger("genie/visualization");
+const GENIE_PNG_SCALE = 4;
+const GENIE_PNG_SHARPEN = { sigma: 1.1, m1: 0.9, m2: 0.02 } as const;
+
+type SharpFactory = (typeof import("sharp"))["default"];
+
+let sharpFactoryPromise: Promise<SharpFactory | undefined> | undefined;
+
+function isByteStream(value: unknown): value is ReadableStream<Uint8Array> {
+  return (
+    isRecord(value) && typeof value.getReader === "function" && typeof value.cancel === "function"
+  );
+}
+
+async function loadSharp(): Promise<SharpFactory | undefined> {
+  sharpFactoryPromise ??= import("sharp")
+    .then((module) => module.default)
+    .catch((error: unknown) => {
+      logger.debug("PNG enhancement unavailable; serving the original Genie image", {
+        error: toError(error),
+      });
+      return undefined;
+    });
+  return sharpFactoryPromise;
+}
+
+/**
+ * Resize a Genie screenshot to four times its source dimensions with Lanczos3,
+ * then apply a mild unsharp mask. Returns the original bytes when the optional
+ * Sharp runtime is unavailable or rejects the image.
+ */
+export async function upscaleGenieVisualizationPng(inputBytes: Uint8Array): Promise<Uint8Array> {
+  const sharp = await loadSharp();
+  if (!sharp) return inputBytes;
+  try {
+    const input = Buffer.from(inputBytes);
+    const metadata = await sharp(input).metadata();
+    if (!metadata.width || !metadata.height) {
+      throw new Error("Could not read Genie visualization dimensions");
+    }
+    const output = await sharp(input)
+      .resize({
+        width: metadata.width * GENIE_PNG_SCALE,
+        height: metadata.height * GENIE_PNG_SCALE,
+        kernel: sharp.kernel.lanczos3,
+        fit: "fill",
+      })
+      .sharpen(GENIE_PNG_SHARPEN)
+      .png()
+      .toBuffer();
+    return new Uint8Array(output);
+  } catch (error) {
+    logger.warn("PNG enhancement failed; serving the original Genie image", {
+      error: toError(error),
+    });
+    return inputBytes;
+  }
+}
+
+/** Identity and cancellation for a Genie visualization download. */
+export interface DownloadGenieVisualizationOptions {
+  /** Workspace client whose identity owns the conversation. */
+  workspaceClient?: WorkspaceClient;
+  /** External cancellation for the request and image body. */
+  context?: ContextLike;
+}
+
+/**
+ * Download Genie's PNG and apply optional raster cleanup before returning it.
+ * The SDK's raw transport preserves binary responses and its normal authentication;
+ * source-chart exports should still be preferred when they are available.
+ */
+export async function downloadGenieVisualization(
+  input: GenieVisualizationRequest,
+  options: DownloadGenieVisualizationOptions = {},
+): Promise<Uint8Array> {
+  const client = options.workspaceClient ?? createWorkspaceClient();
+  const response = await client.apiClient.request(
+    {
+      path: genieVisualizationDownloadPath(input),
+      method: "GET",
+      headers: new Headers({ accept: "image/png" }),
+      responseHeaders: ["content-type"],
+      raw: true,
+    },
+    options.context ? toContext(options.context) : undefined,
+  );
+  if (
+    !isRecord(response) ||
+    typeof response["content-type"] !== "string" ||
+    response["content-type"].split(";")[0]?.trim().toLowerCase() !== "image/png" ||
+    !isByteStream(response.contents)
+  ) {
+    if (isRecord(response) && isByteStream(response.contents)) {
+      await response.contents.cancel().catch(() => undefined);
+    }
+    throw new TypeError("Genie did not return a PNG visualization");
+  }
+  const bytes = new Uint8Array(await new Response(response.contents).arrayBuffer());
+  if (
+    bytes.length < 8 ||
+    bytes[0] !== 0x89 ||
+    bytes[1] !== 0x50 ||
+    bytes[2] !== 0x4e ||
+    bytes[3] !== 0x47 ||
+    bytes[4] !== 0x0d ||
+    bytes[5] !== 0x0a ||
+    bytes[6] !== 0x1a ||
+    bytes[7] !== 0x0a
+  ) {
+    throw new TypeError("Genie returned an invalid PNG visualization");
+  }
+  return upscaleGenieVisualizationPng(bytes);
+}