@dbx-tools/genie 0.9.187 → 0.9.188

package/README.md

@@ -183,6 +183,35 @@ thrown.
 - `pollIntervalMs` - fallback polling cadence, default `500`.
 - `context` - SDK `Context` or `AbortSignal` for cancellation.
 
+## Download an original chart
+
+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 and
+returns the original bytes; it does not replot data or convert images to SVG.
+This endpoint also accepts visualization attachments from Agent Mode
+conversations.
+
 ## Modules
 
 - `chat` - `genieChat()` raw snapshot stream and `genieEventChat()` typed event
@@ -190,6 +219,7 @@ thrown.
 - `agentMode` - `genieAgentModeChat()`, availability classification, and Agent
   Mode transport options plus Agent Mode conversation history.
 - `space` - `getGenieSpace()` and `genieSampleQuestions()`.
+- `visualization` - `downloadGenieVisualization()` for original 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.188";
 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 { 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.188";
 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 { 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.188";
 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 { downloadGenieVisualization } from "./src/visualization.js";
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSwyQ0FBMkM7QUFDM0MsbURBQW1EO0FBQ25ELHdFQUF3RTtBQUV4RSxNQUFNLENBQUMsTUFBTSxrQkFBa0IsR0FBRyxrQkFBa0IsQ0FBQztBQUNyRCxNQUFNLENBQUMsTUFBTSxlQUFlLEdBQUcsU0FBUyxDQUFDO0FBQ3pDLE9BQU8sS0FBSyxTQUFTLE1BQU0scUJBQXFCLENBQUM7QUFDakQsT0FBTyxLQUFLLElBQUksTUFBTSxlQUFlLENBQUM7QUFDdEMsT0FBTyxLQUFLLFFBQVEsTUFBTSxtQkFBbUIsQ0FBQztBQUM5QyxPQUFPLEtBQUssS0FBSyxNQUFNLGdCQUFnQixDQUFDO0FBQ3hDLE9BQU8sS0FBSyxhQUFhLE1BQU0sd0JBQXdCLENBQUM7QUFDeEQsT0FBTyxFQUFFLGtCQUFrQixFQUFFLHlCQUF5QixFQUFFLHNCQUFzQixFQUFFLE1BQU0scUJBQXFCLENBQUM7QUFFNUcsT0FBTyxFQUFFLFNBQVMsRUFBRSxjQUFjLEVBQUUsTUFBTSxlQUFlLENBQUM7QUFFMUQsT0FBTyxFQUFFLHdCQUF3QixFQUFFLE1BQU0sbUJBQW1CLENBQUM7QUFFN0QsT0FBTyxFQUFFLGdCQUFnQixFQUFFLGFBQWEsRUFBRSxvQkFBb0IsRUFBRSxNQUFNLGdCQUFnQixDQUFDO0FBRXZGLE9BQU8sRUFBRSwwQkFBMEIsRUFBRSxNQUFNLHdCQUF3QixDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLy8gR0VORVJBVEVEIGJ5IHByb2plbiB3YXRjaCAtIERPIE5PVCBFRElULlxuLy8gUmVnZW5lcmF0ZWQgZnJvbSB0aGUgZXhwb3J0aW5nIG1vZHVsZXMgaW4gLi9zcmMuXG4vLyBIYW5kIGVkaXRzIGFyZSBvdmVyd3JpdHRlbiBvbiB0aGUgbmV4dCB3YXRjaDsgdGhpcyBmaWxlIGlzIHJlYWQtb25seS5cblxuZXhwb3J0IGNvbnN0IFBBQ0tBR0VfSURFTlRJRklFUiA9IFwiQGRieC10b29scy9nZW5pZVwiO1xuZXhwb3J0IGNvbnN0IFBBQ0tBR0VfVkVSU0lPTiA9IFwiMC45LjE4OFwiO1xuZXhwb3J0ICogYXMgYWdlbnRNb2RlIGZyb20gXCIuL3NyYy9hZ2VudC1tb2RlLnRzXCI7XG5leHBvcnQgKiBhcyBjaGF0IGZyb20gXCIuL3NyYy9jaGF0LnRzXCI7XG5leHBvcnQgKiBhcyBmZWVkYmFjayBmcm9tIFwiLi9zcmMvZmVlZGJhY2sudHNcIjtcbmV4cG9ydCAqIGFzIHNwYWNlIGZyb20gXCIuL3NyYy9zcGFjZS50c1wiO1xuZXhwb3J0ICogYXMgdmlzdWFsaXphdGlvbiBmcm9tIFwiLi9zcmMvdmlzdWFsaXphdGlvbi50c1wiO1xuZXhwb3J0IHsgZ2VuaWVBZ2VudE1vZGVDaGF0LCBsaXN0QWdlbnRNb2RlQ29udmVyc2F0aW9uLCBpc0FnZW50TW9kZVVuYXZhaWxhYmxlIH0gZnJvbSBcIi4vc3JjL2FnZW50LW1vZGUudHNcIjtcbmV4cG9ydCB0eXBlIHsgR2VuaWVBZ2VudE1vZGVPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2FnZW50LW1vZGUudHNcIjtcbmV4cG9ydCB7IGdlbmllQ2hhdCwgZ2VuaWVFdmVudENoYXQgfSBmcm9tIFwiLi9zcmMvY2hhdC50c1wiO1xuZXhwb3J0IHR5cGUgeyBHZW5pZUNoYXRPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2NoYXQudHNcIjtcbmV4cG9ydCB7IHNlbmRHZW5pZU1lc3NhZ2VGZWVkYmFjayB9IGZyb20gXCIuL3NyYy9mZWVkYmFjay50c1wiO1xuZXhwb3J0IHR5cGUgeyBTZW5kR2VuaWVNZXNzYWdlRmVlZGJhY2tPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2ZlZWRiYWNrLnRzXCI7XG5leHBvcnQgeyBjcmVhdGVHZW5pZVNwYWNlLCBnZXRHZW5pZVNwYWNlLCBnZW5pZVNhbXBsZVF1ZXN0aW9ucyB9IGZyb20gXCIuL3NyYy9zcGFjZS50c1wiO1xuZXhwb3J0IHR5cGUgeyBHZXRHZW5pZVNwYWNlT3B0aW9ucywgQ3JlYXRlR2VuaWVTcGFjZU9wdGlvbnMgfSBmcm9tIFwiLi9zcmMvc3BhY2UudHNcIjtcbmV4cG9ydCB7IGRvd25sb2FkR2VuaWVWaXN1YWxpemF0aW9uIH0gZnJvbSBcIi4vc3JjL3Zpc3VhbGl6YXRpb24udHNcIjtcbmV4cG9ydCB0eXBlIHsgRG93bmxvYWRHZW5pZVZpc3VhbGl6YXRpb25PcHRpb25zIH0gZnJvbSBcIi4vc3JjL3Zpc3VhbGl6YXRpb24udHNcIjtcbiJdfQ==

package/lib/src/visualization.d.ts

@@ -0,0 +1,15 @@
+import { type WorkspaceClient } from "@databricks/appkit";
+import { type ContextLike } from "@dbx-tools/appkit";
+import { type GenieVisualizationRequest } from "@dbx-tools/shared-genie";
+/** 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 original PNG bytes without redrawing or transcoding them.
+ * The SDK's raw transport preserves binary responses and its normal authentication.
+ */
+export declare function downloadGenieVisualization(input: GenieVisualizationRequest, options?: DownloadGenieVisualizationOptions): Promise<Uint8Array>;

package/lib/src/visualization.js

@@ -0,0 +1,44 @@
+import { createWorkspaceClient } from "@databricks/appkit";
+import { toContext } from "@dbx-tools/appkit";
+import { isRecord } from "@dbx-tools/shared-core";
+import { genieVisualizationDownloadPath, } from "@dbx-tools/shared-genie";
+function isByteStream(value) {
+    return (isRecord(value) && typeof value.getReader === "function" && typeof value.cancel === "function");
+}
+/**
+ * Download Genie's original PNG bytes without redrawing or transcoding them.
+ * The SDK's raw transport preserves binary responses and its normal authentication.
+ */
+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 bytes;
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidmlzdWFsaXphdGlvbi5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy92aXN1YWxpemF0aW9uLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxxQkFBcUIsRUFBd0IsTUFBTSxvQkFBb0IsQ0FBQztBQUNqRixPQUFPLEVBQW9CLFNBQVMsRUFBRSxNQUFNLG1CQUFtQixDQUFDO0FBQ2hFLE9BQU8sRUFBRSxRQUFRLEVBQUUsTUFBTSx3QkFBd0IsQ0FBQztBQUNsRCxPQUFPLEVBRUwsOEJBQThCLEdBQy9CLE1BQU0seUJBQXlCLENBQUM7QUFFakMsU0FBUyxZQUFZLENBQUMsS0FBYztJQUNsQyxPQUFPLENBQ0wsUUFBUSxDQUFDLEtBQUssQ0FBQyxJQUFJLE9BQU8sS0FBSyxDQUFDLFNBQVMsS0FBSyxVQUFVLElBQUksT0FBTyxLQUFLLENBQUMsTUFBTSxLQUFLLFVBQVUsQ0FDL0YsQ0FBQztBQUNKLENBQUM7QUFVRDs7O0dBR0c7QUFDSCxNQUFNLENBQUMsS0FBSyxVQUFVLDBCQUEwQixDQUM5QyxLQUFnQyxFQUNoQyxVQUE2QyxFQUFFO0lBRS9DLE1BQU0sTUFBTSxHQUFHLE9BQU8sQ0FBQyxlQUFlLElBQUkscUJBQXFCLEVBQUUsQ0FBQztJQUNsRSxNQUFNLFFBQVEsR0FBRyxNQUFNLE1BQU0sQ0FBQyxTQUFTLENBQUMsT0FBTyxDQUM3QztRQUNFLElBQUksRUFBRSw4QkFBOEIsQ0FBQyxLQUFLLENBQUM7UUFDM0MsTUFBTSxFQUFFLEtBQUs7UUFDYixPQUFPLEVBQUUsSUFBSSxPQUFPLENBQUMsRUFBRSxNQUFNLEVBQUUsV0FBVyxFQUFFLENBQUM7UUFDN0MsZUFBZSxFQUFFLENBQUMsY0FBYyxDQUFDO1FBQ2pDLEdBQUcsRUFBRSxJQUFJO0tBQ1YsRUFDRCxPQUFPLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQyxTQUFTLENBQUMsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsQ0FBQyxTQUFTLENBQ3pELENBQUM7SUFDRixJQUNFLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQztRQUNuQixPQUFPLFFBQVEsQ0FBQyxjQUFjLENBQUMsS0FBSyxRQUFRO1FBQzVDLFFBQVEsQ0FBQyxjQUFjLENBQUMsQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLENBQUMsQ0FBQyxDQUFDLEVBQUUsSUFBSSxFQUFFLENBQUMsV0FBVyxFQUFFLEtBQUssV0FBVztRQUM1RSxDQUFDLFlBQVksQ0FBQyxRQUFRLENBQUMsUUFBUSxDQUFDLEVBQ2hDLENBQUM7UUFDRCxJQUFJLFFBQVEsQ0FBQyxRQUFRLENBQUMsSUFBSSxZQUFZLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQyxFQUFFLENBQUM7WUFDMUQsTUFBTSxRQUFRLENBQUMsUUFBUSxDQUFDLE1BQU0sRUFBRSxDQUFDLEtBQUssQ0FBQyxHQUFHLEVBQUUsQ0FBQyxTQUFTLENBQUMsQ0FBQztRQUMxRCxDQUFDO1FBQ0QsTUFBTSxJQUFJLFNBQVMsQ0FBQywwQ0FBMEMsQ0FBQyxDQUFDO0lBQ2xFLENBQUM7SUFDRCxNQUFNLEtBQUssR0FBRyxJQUFJLFVBQVUsQ0FBQyxNQUFNLElBQUksUUFBUSxDQUFDLFFBQVEsQ0FBQyxRQUFRLENBQUMsQ0FBQyxXQUFXLEVBQUUsQ0FBQyxDQUFDO0lBQ2xGLElBQ0UsS0FBSyxDQUFDLE1BQU0sR0FBRyxDQUFDO1FBQ2hCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJO1FBQ2pCLEtBQUssQ0FBQyxDQUFDLENBQUMsS0FBSyxJQUFJLEVBQ2pCLENBQUM7UUFDRCxNQUFNLElBQUksU0FBUyxDQUFDLDZDQUE2QyxDQUFDLENBQUM7SUFDckUsQ0FBQztJQUNELE9BQU8sS0FBSyxDQUFDO0FBQ2YsQ0FBQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7IGNyZWF0ZVdvcmtzcGFjZUNsaWVudCwgdHlwZSBXb3Jrc3BhY2VDbGllbnQgfSBmcm9tIFwiQGRhdGFicmlja3MvYXBwa2l0XCI7XG5pbXBvcnQgeyB0eXBlIENvbnRleHRMaWtlLCB0b0NvbnRleHQgfSBmcm9tIFwiQGRieC10b29scy9hcHBraXRcIjtcbmltcG9ydCB7IGlzUmVjb3JkIH0gZnJvbSBcIkBkYngtdG9vbHMvc2hhcmVkLWNvcmVcIjtcbmltcG9ydCB7XG4gIHR5cGUgR2VuaWVWaXN1YWxpemF0aW9uUmVxdWVzdCxcbiAgZ2VuaWVWaXN1YWxpemF0aW9uRG93bmxvYWRQYXRoLFxufSBmcm9tIFwiQGRieC10b29scy9zaGFyZWQtZ2VuaWVcIjtcblxuZnVuY3Rpb24gaXNCeXRlU3RyZWFtKHZhbHVlOiB1bmtub3duKTogdmFsdWUgaXMgUmVhZGFibGVTdHJlYW08VWludDhBcnJheT4ge1xuICByZXR1cm4gKFxuICAgIGlzUmVjb3JkKHZhbHVlKSAmJiB0eXBlb2YgdmFsdWUuZ2V0UmVhZGVyID09PSBcImZ1bmN0aW9uXCIgJiYgdHlwZW9mIHZhbHVlLmNhbmNlbCA9PT0gXCJmdW5jdGlvblwiXG4gICk7XG59XG5cbi8qKiBJZGVudGl0eSBhbmQgY2FuY2VsbGF0aW9uIGZvciBhIEdlbmllIHZpc3VhbGl6YXRpb24gZG93bmxvYWQuICovXG5leHBvcnQgaW50ZXJmYWNlIERvd25sb2FkR2VuaWVWaXN1YWxpemF0aW9uT3B0aW9ucyB7XG4gIC8qKiBXb3Jrc3BhY2UgY2xpZW50IHdob3NlIGlkZW50aXR5IG93bnMgdGhlIGNvbnZlcnNhdGlvbi4gKi9cbiAgd29ya3NwYWNlQ2xpZW50PzogV29ya3NwYWNlQ2xpZW50O1xuICAvKiogRXh0ZXJuYWwgY2FuY2VsbGF0aW9uIGZvciB0aGUgcmVxdWVzdCBhbmQgaW1hZ2UgYm9keS4gKi9cbiAgY29udGV4dD86IENvbnRleHRMaWtlO1xufVxuXG4vKipcbiAqIERvd25sb2FkIEdlbmllJ3Mgb3JpZ2luYWwgUE5HIGJ5dGVzIHdpdGhvdXQgcmVkcmF3aW5nIG9yIHRyYW5zY29kaW5nIHRoZW0uXG4gKiBUaGUgU0RLJ3MgcmF3IHRyYW5zcG9ydCBwcmVzZXJ2ZXMgYmluYXJ5IHJlc3BvbnNlcyBhbmQgaXRzIG5vcm1hbCBhdXRoZW50aWNhdGlvbi5cbiAqL1xuZXhwb3J0IGFzeW5jIGZ1bmN0aW9uIGRvd25sb2FkR2VuaWVWaXN1YWxpemF0aW9uKFxuICBpbnB1dDogR2VuaWVWaXN1YWxpemF0aW9uUmVxdWVzdCxcbiAgb3B0aW9uczogRG93bmxvYWRHZW5pZVZpc3VhbGl6YXRpb25PcHRpb25zID0ge30sXG4pOiBQcm9taXNlPFVpbnQ4QXJyYXk+IHtcbiAgY29uc3QgY2xpZW50ID0gb3B0aW9ucy53b3Jrc3BhY2VDbGllbnQgPz8gY3JlYXRlV29ya3NwYWNlQ2xpZW50KCk7XG4gIGNvbnN0IHJlc3BvbnNlID0gYXdhaXQgY2xpZW50LmFwaUNsaWVudC5yZXF1ZXN0KFxuICAgIHtcbiAgICAgIHBhdGg6IGdlbmllVmlzdWFsaXphdGlvbkRvd25sb2FkUGF0aChpbnB1dCksXG4gICAgICBtZXRob2Q6IFwiR0VUXCIsXG4gICAgICBoZWFkZXJzOiBuZXcgSGVhZGVycyh7IGFjY2VwdDogXCJpbWFnZS9wbmdcIiB9KSxcbiAgICAgIHJlc3BvbnNlSGVhZGVyczogW1wiY29udGVudC10eXBlXCJdLFxuICAgICAgcmF3OiB0cnVlLFxuICAgIH0sXG4gICAgb3B0aW9ucy5jb250ZXh0ID8gdG9Db250ZXh0KG9wdGlvbnMuY29udGV4dCkgOiB1bmRlZmluZWQsXG4gICk7XG4gIGlmIChcbiAgICAhaXNSZWNvcmQocmVzcG9uc2UpIHx8XG4gICAgdHlwZW9mIHJlc3BvbnNlW1wiY29udGVudC10eXBlXCJdICE9PSBcInN0cmluZ1wiIHx8XG4gICAgcmVzcG9uc2VbXCJjb250ZW50LXR5cGVcIl0uc3BsaXQoXCI7XCIpWzBdPy50cmltKCkudG9Mb3dlckNhc2UoKSAhPT0gXCJpbWFnZS9wbmdcIiB8fFxuICAgICFpc0J5dGVTdHJlYW0ocmVzcG9uc2UuY29udGVudHMpXG4gICkge1xuICAgIGlmIChpc1JlY29yZChyZXNwb25zZSkgJiYgaXNCeXRlU3RyZWFtKHJlc3BvbnNlLmNvbnRlbnRzKSkge1xuICAgICAgYXdhaXQgcmVzcG9uc2UuY29udGVudHMuY2FuY2VsKCkuY2F0Y2goKCkgPT4gdW5kZWZpbmVkKTtcbiAgICB9XG4gICAgdGhyb3cgbmV3IFR5cGVFcnJvcihcIkdlbmllIGRpZCBub3QgcmV0dXJuIGEgUE5HIHZpc3VhbGl6YXRpb25cIik7XG4gIH1cbiAgY29uc3QgYnl0ZXMgPSBuZXcgVWludDhBcnJheShhd2FpdCBuZXcgUmVzcG9uc2UocmVzcG9uc2UuY29udGVudHMpLmFycmF5QnVmZmVyKCkpO1xuICBpZiAoXG4gICAgYnl0ZXMubGVuZ3RoIDwgOCB8fFxuICAgIGJ5dGVzWzBdICE9PSAweDg5IHx8XG4gICAgYnl0ZXNbMV0gIT09IDB4NTAgfHxcbiAgICBieXRlc1syXSAhPT0gMHg0ZSB8fFxuICAgIGJ5dGVzWzNdICE9PSAweDQ3IHx8XG4gICAgYnl0ZXNbNF0gIT09IDB4MGQgfHxcbiAgICBieXRlc1s1XSAhPT0gMHgwYSB8fFxuICAgIGJ5dGVzWzZdICE9PSAweDFhIHx8XG4gICAgYnl0ZXNbN10gIT09IDB4MGFcbiAgKSB7XG4gICAgdGhyb3cgbmV3IFR5cGVFcnJvcihcIkdlbmllIHJldHVybmVkIGFuIGludmFsaWQgUE5HIHZpc3VhbGl6YXRpb25cIik7XG4gIH1cbiAgcmV0dXJuIGJ5dGVzO1xufVxuIl19

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

@@ -24,9 +24,9 @@
   },
   "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.188",
+    "@dbx-tools/shared-core": "^0.9.188",
+    "@dbx-tools/shared-genie": "^0.9.188"
   },
   "main": "./lib/index.js",
   "license": "Apache-2.0",
@@ -42,7 +42,7 @@
       "./package.json": "./package.json"
     }
   },
-  "version": "0.9.187",
+  "version": "0.9.188",
   "types": "./lib/index.d.ts",
   "dbxToolsConfig": {
     "tags": [

package/src/visualization.ts

@@ -0,0 +1,68 @@
+import { createWorkspaceClient, type WorkspaceClient } from "@databricks/appkit";
+import { type ContextLike, toContext } from "@dbx-tools/appkit";
+import { isRecord } from "@dbx-tools/shared-core";
+import {
+  type GenieVisualizationRequest,
+  genieVisualizationDownloadPath,
+} from "@dbx-tools/shared-genie";
+
+function isByteStream(value: unknown): value is ReadableStream<Uint8Array> {
+  return (
+    isRecord(value) && typeof value.getReader === "function" && typeof value.cancel === "function"
+  );
+}
+
+/** 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 original PNG bytes without redrawing or transcoding them.
+ * The SDK's raw transport preserves binary responses and its normal authentication.
+ */
+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 bytes;
+}