@databricks/appkit 0.11.2 → 0.12.1

package/dist/connectors/genie/client.js

@@ -0,0 +1,208 @@
+import { createLogger } from "../../logging/logger.js";
+import "../../logging/index.js";
+import { genieConnectorDefaults } from "./defaults.js";
+import { pollWaiter } from "./poll-waiter.js";
+import { Time, TimeUnits } from "@databricks/sdk-experimental";
+
+//#region src/connectors/genie/client.ts
+const logger = createLogger("connectors:genie");
+function mapAttachments(message) {
+	return message.attachments?.map((att) => ({
+		attachmentId: att.attachment_id,
+		query: att.query ? {
+			title: att.query.title,
+			description: att.query.description,
+			query: att.query.query,
+			statementId: att.query.statement_id
+		} : void 0,
+		text: att.text ? { content: att.text.content } : void 0,
+		suggestedQuestions: att.suggested_questions?.questions
+	})) ?? [];
+}
+function toMessageResponse(message) {
+	return {
+		messageId: message.message_id,
+		conversationId: message.conversation_id,
+		spaceId: message.space_id,
+		status: message.status ?? "COMPLETED",
+		content: message.content,
+		attachments: mapAttachments(message),
+		error: message.error?.error
+	};
+}
+var GenieConnector = class {
+	config;
+	constructor(config = {}) {
+		this.config = {
+			timeout: config.timeout ?? genieConnectorDefaults.timeout,
+			maxMessages: config.maxMessages ?? genieConnectorDefaults.maxMessages
+		};
+	}
+	async startMessage(workspaceClient, spaceId, content, conversationId) {
+		if (conversationId) {
+			const waiter = await workspaceClient.genie.createMessage({
+				space_id: spaceId,
+				conversation_id: conversationId,
+				content
+			});
+			return {
+				messageWaiter: waiter,
+				conversationId,
+				messageId: waiter.message_id ?? ""
+			};
+		}
+		const start = await workspaceClient.genie.startConversation({
+			space_id: spaceId,
+			content
+		});
+		return {
+			messageWaiter: start,
+			conversationId: start.conversation_id,
+			messageId: start.message_id
+		};
+	}
+	async waitForMessage(messageWaiter, options) {
+		const timeout = options?.timeout ?? this.config.timeout;
+		const waitOptions = timeout > 0 ? { timeout: new Time(timeout, TimeUnits.milliseconds) } : {};
+		return messageWaiter.wait(waitOptions);
+	}
+	async listConversationMessages(workspaceClient, spaceId, conversationId, options) {
+		const maxMessages = options?.maxMessages ?? this.config.maxMessages;
+		const allMessages = [];
+		let pageToken;
+		do {
+			const response = await workspaceClient.genie.listConversationMessages({
+				space_id: spaceId,
+				conversation_id: conversationId,
+				page_size: genieConnectorDefaults.pageSize,
+				...pageToken ? { page_token: pageToken } : {}
+			});
+			if (response.messages) allMessages.push(...response.messages);
+			pageToken = response.next_page_token;
+		} while (pageToken && allMessages.length < maxMessages);
+		return allMessages.slice(0, maxMessages).reverse().map(toMessageResponse);
+	}
+	async getMessageAttachmentQueryResult(workspaceClient, spaceId, conversationId, messageId, attachmentId, _signal) {
+		return (await workspaceClient.genie.getMessageAttachmentQueryResult({
+			space_id: spaceId,
+			conversation_id: conversationId,
+			message_id: messageId,
+			attachment_id: attachmentId
+		})).statement_response;
+	}
+	async *streamSendMessage(workspaceClient, spaceId, content, conversationId, options) {
+		try {
+			const { messageWaiter, conversationId: resultConversationId, messageId: resultMessageId } = await this.startMessage(workspaceClient, spaceId, content, conversationId);
+			yield {
+				type: "message_start",
+				conversationId: resultConversationId,
+				messageId: resultMessageId,
+				spaceId
+			};
+			const timeout = options?.timeout != null ? options.timeout : this.config.timeout;
+			const waitOptions = timeout > 0 ? { timeout: new Time(timeout, TimeUnits.milliseconds) } : {};
+			let completedMessage;
+			for await (const event of pollWaiter(messageWaiter, waitOptions)) if (event.type === "progress" && event.value.status) yield {
+				type: "status",
+				status: event.value.status
+			};
+			else if (event.type === "completed") completedMessage = event.value;
+			const messageResponse = toMessageResponse(completedMessage);
+			yield {
+				type: "message_result",
+				message: messageResponse
+			};
+			yield* this.emitQueryResults(workspaceClient, spaceId, resultConversationId, messageResponse.messageId, messageResponse);
+		} catch (error) {
+			logger.error("Genie message error: %O", error);
+			yield {
+				type: "error",
+				error: error instanceof Error ? error.message : "Genie request failed"
+			};
+		}
+	}
+	async *emitQueryResults(workspaceClient, spaceId, conversationId, messageId, messageResponse) {
+		const attachments = messageResponse.attachments ?? [];
+		for (const att of attachments) {
+			if (!att.query?.statementId || !att.attachmentId) continue;
+			try {
+				const data = await this.getMessageAttachmentQueryResult(workspaceClient, spaceId, conversationId, messageId, att.attachmentId);
+				yield {
+					type: "query_result",
+					attachmentId: att.attachmentId,
+					statementId: att.query.statementId,
+					data
+				};
+			} catch (error) {
+				logger.error("Failed to fetch query result for attachment %s: %O", att.attachmentId, error);
+				yield {
+					type: "error",
+					error: `Failed to fetch query result for attachment ${att.attachmentId}`
+				};
+			}
+		}
+	}
+	async *streamConversation(workspaceClient, spaceId, conversationId, options) {
+		const includeQueryResults = options?.includeQueryResults !== false;
+		try {
+			const messageResponses = await this.listConversationMessages(workspaceClient, spaceId, conversationId);
+			for (const messageResponse of messageResponses) yield {
+				type: "message_result",
+				message: messageResponse
+			};
+			if (includeQueryResults) {
+				const queryAttachments = [];
+				for (const msg of messageResponses) for (const att of msg.attachments ?? []) if (att.query?.statementId && att.attachmentId) queryAttachments.push({
+					messageId: msg.messageId,
+					attachmentId: att.attachmentId,
+					statementId: att.query.statementId
+				});
+				const results = await Promise.allSettled(queryAttachments.map(async (att) => {
+					const data = await this.getMessageAttachmentQueryResult(workspaceClient, spaceId, conversationId, att.messageId, att.attachmentId);
+					return {
+						attachmentId: att.attachmentId,
+						statementId: att.statementId,
+						data
+					};
+				}));
+				for (const result of results) if (result.status === "fulfilled") yield {
+					type: "query_result",
+					attachmentId: result.value.attachmentId,
+					statementId: result.value.statementId,
+					data: result.value.data
+				};
+				else {
+					logger.error("Failed to fetch query result: %O", result.reason);
+					yield {
+						type: "error",
+						error: result.reason instanceof Error ? result.reason.message : "Failed to fetch query result"
+					};
+				}
+			}
+		} catch (error) {
+			logger.error("Genie getConversation error: %O", error);
+			yield {
+				type: "error",
+				error: error instanceof Error ? error.message : "Failed to fetch conversation"
+			};
+		}
+	}
+	async sendMessage(workspaceClient, spaceId, content, conversationId) {
+		const { messageWaiter, conversationId: resultConversationId } = await this.startMessage(workspaceClient, spaceId, content, conversationId);
+		return {
+			...toMessageResponse(await this.waitForMessage(messageWaiter)),
+			conversationId: resultConversationId
+		};
+	}
+	async getConversation(workspaceClient, spaceId, conversationId) {
+		return {
+			conversationId,
+			spaceId,
+			messages: await this.listConversationMessages(workspaceClient, spaceId, conversationId)
+		};
+	}
+};
+
+//#endregion
+export { GenieConnector };
+//# sourceMappingURL=client.js.map

package/dist/connectors/genie/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/genie/client.ts"],"sourcesContent":["import type { WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport { Time, TimeUnits } from \"@databricks/sdk-experimental\";\nimport type { GenieMessage } from \"@databricks/sdk-experimental/dist/apis/dashboards\";\nimport type { Waiter } from \"@databricks/sdk-experimental/dist/wait\";\nimport { createLogger } from \"../../logging\";\nimport { genieConnectorDefaults } from \"./defaults\";\nimport { pollWaiter } from \"./poll-waiter\";\nimport type {\n  GenieAttachmentResponse,\n  GenieConversationHistoryResponse,\n  GenieMessageResponse,\n  GenieStreamEvent,\n} from \"./types\";\n\nconst logger = createLogger(\"connectors:genie\");\n\ntype CreateMessageWaiter = Waiter<GenieMessage, GenieMessage>;\n\nexport interface GenieConnectorConfig {\n  timeout?: number;\n  maxMessages?: number;\n}\n\nfunction mapAttachments(message: GenieMessage): GenieAttachmentResponse[] {\n  return (\n    message.attachments?.map((att) => ({\n      attachmentId: att.attachment_id,\n      query: att.query\n        ? {\n            title: att.query.title,\n            description: att.query.description,\n            query: att.query.query,\n            statementId: att.query.statement_id,\n          }\n        : undefined,\n      text: att.text ? { content: att.text.content } : undefined,\n      suggestedQuestions: att.suggested_questions?.questions,\n    })) ?? []\n  );\n}\n\nfunction toMessageResponse(message: GenieMessage): GenieMessageResponse {\n  return {\n    messageId: message.message_id,\n    conversationId: message.conversation_id,\n    spaceId: message.space_id,\n    status: message.status ?? \"COMPLETED\",\n    content: message.content,\n    attachments: mapAttachments(message),\n    error: message.error?.error,\n  };\n}\n\nexport class GenieConnector {\n  private readonly config: Required<GenieConnectorConfig>;\n\n  constructor(config: GenieConnectorConfig = {}) {\n    this.config = {\n      timeout: config.timeout ?? genieConnectorDefaults.timeout,\n      maxMessages: config.maxMessages ?? genieConnectorDefaults.maxMessages,\n    };\n  }\n\n  async startMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n  ): Promise<{\n    messageWaiter: CreateMessageWaiter;\n    conversationId: string;\n    messageId: string;\n  }> {\n    if (conversationId) {\n      const waiter = await workspaceClient.genie.createMessage({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        content,\n      });\n      return {\n        messageWaiter: waiter,\n        conversationId,\n        messageId: waiter.message_id ?? \"\",\n      };\n    }\n    const start = await workspaceClient.genie.startConversation({\n      space_id: spaceId,\n      content,\n    });\n    return {\n      messageWaiter: start as unknown as CreateMessageWaiter,\n      conversationId: start.conversation_id,\n      messageId: start.message_id,\n    };\n  }\n\n  async waitForMessage(\n    messageWaiter: CreateMessageWaiter,\n    options?: { timeout?: number },\n  ): Promise<GenieMessage> {\n    const timeout = options?.timeout ?? this.config.timeout;\n    const waitOptions =\n      timeout > 0 ? { timeout: new Time(timeout, TimeUnits.milliseconds) } : {};\n    return messageWaiter.wait(waitOptions);\n  }\n\n  async listConversationMessages(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    options?: { maxMessages?: number },\n  ): Promise<GenieMessageResponse[]> {\n    const maxMessages = options?.maxMessages ?? this.config.maxMessages;\n    const allMessages: GenieMessage[] = [];\n    let pageToken: string | undefined;\n\n    do {\n      const response = await workspaceClient.genie.listConversationMessages({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        page_size: genieConnectorDefaults.pageSize,\n        ...(pageToken ? { page_token: pageToken } : {}),\n      });\n\n      if (response.messages) {\n        allMessages.push(...response.messages);\n      }\n\n      pageToken = response.next_page_token;\n    } while (pageToken && allMessages.length < maxMessages);\n\n    return allMessages.slice(0, maxMessages).reverse().map(toMessageResponse);\n  }\n\n  async getMessageAttachmentQueryResult(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    attachmentId: string,\n    _signal?: AbortSignal,\n  ): Promise<unknown> {\n    const response =\n      await workspaceClient.genie.getMessageAttachmentQueryResult({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        message_id: messageId,\n        attachment_id: attachmentId,\n      });\n    return response.statement_response;\n  }\n\n  async *streamSendMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n    options?: { timeout?: number },\n  ): AsyncGenerator<GenieStreamEvent> {\n    try {\n      const {\n        messageWaiter,\n        conversationId: resultConversationId,\n        messageId: resultMessageId,\n      } = await this.startMessage(\n        workspaceClient,\n        spaceId,\n        content,\n        conversationId,\n      );\n\n      yield {\n        type: \"message_start\",\n        conversationId: resultConversationId,\n        messageId: resultMessageId,\n        spaceId,\n      };\n\n      const timeout =\n        options?.timeout != null ? options.timeout : this.config.timeout;\n      const waitOptions =\n        timeout > 0\n          ? { timeout: new Time(timeout, TimeUnits.milliseconds) }\n          : {};\n\n      let completedMessage!: GenieMessage;\n      for await (const event of pollWaiter(messageWaiter, waitOptions)) {\n        if (event.type === \"progress\" && event.value.status) {\n          yield { type: \"status\", status: event.value.status };\n        } else if (event.type === \"completed\") {\n          completedMessage = event.value;\n        }\n      }\n\n      const messageResponse = toMessageResponse(completedMessage);\n      yield { type: \"message_result\", message: messageResponse };\n\n      yield* this.emitQueryResults(\n        workspaceClient,\n        spaceId,\n        resultConversationId,\n        messageResponse.messageId,\n        messageResponse,\n      );\n    } catch (error) {\n      logger.error(\"Genie message error: %O\", error);\n      yield {\n        type: \"error\",\n        error: error instanceof Error ? error.message : \"Genie request failed\",\n      };\n    }\n  }\n\n  private async *emitQueryResults(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    messageResponse: GenieMessageResponse,\n  ): AsyncGenerator<\n    Extract<GenieStreamEvent, { type: \"query_result\" } | { type: \"error\" }>\n  > {\n    const attachments = messageResponse.attachments ?? [];\n    for (const att of attachments) {\n      if (!att.query?.statementId || !att.attachmentId) continue;\n      try {\n        const data = await this.getMessageAttachmentQueryResult(\n          workspaceClient,\n          spaceId,\n          conversationId,\n          messageId,\n          att.attachmentId,\n        );\n        yield {\n          type: \"query_result\",\n          attachmentId: att.attachmentId,\n          statementId: att.query.statementId,\n          data,\n        };\n      } catch (error) {\n        logger.error(\n          \"Failed to fetch query result for attachment %s: %O\",\n          att.attachmentId,\n          error,\n        );\n        yield {\n          type: \"error\",\n          error: `Failed to fetch query result for attachment ${att.attachmentId}`,\n        };\n      }\n    }\n  }\n\n  async *streamConversation(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    options?: { includeQueryResults?: boolean },\n  ): AsyncGenerator<GenieStreamEvent> {\n    const includeQueryResults = options?.includeQueryResults !== false;\n\n    try {\n      const messageResponses = await this.listConversationMessages(\n        workspaceClient,\n        spaceId,\n        conversationId,\n      );\n\n      for (const messageResponse of messageResponses) {\n        yield { type: \"message_result\", message: messageResponse };\n      }\n\n      if (includeQueryResults) {\n        const queryAttachments: Array<{\n          messageId: string;\n          attachmentId: string;\n          statementId: string;\n        }> = [];\n\n        for (const msg of messageResponses) {\n          for (const att of msg.attachments ?? []) {\n            if (att.query?.statementId && att.attachmentId) {\n              queryAttachments.push({\n                messageId: msg.messageId,\n                attachmentId: att.attachmentId,\n                statementId: att.query.statementId,\n              });\n            }\n          }\n        }\n\n        const results = await Promise.allSettled(\n          queryAttachments.map(async (att) => {\n            const data = await this.getMessageAttachmentQueryResult(\n              workspaceClient,\n              spaceId,\n              conversationId,\n              att.messageId,\n              att.attachmentId,\n            );\n            return {\n              attachmentId: att.attachmentId,\n              statementId: att.statementId,\n              data,\n            };\n          }),\n        );\n\n        for (const result of results) {\n          if (result.status === \"fulfilled\") {\n            yield {\n              type: \"query_result\",\n              attachmentId: result.value.attachmentId,\n              statementId: result.value.statementId,\n              data: result.value.data,\n            };\n          } else {\n            logger.error(\"Failed to fetch query result: %O\", result.reason);\n            yield {\n              type: \"error\",\n              error:\n                result.reason instanceof Error\n                  ? result.reason.message\n                  : \"Failed to fetch query result\",\n            };\n          }\n        }\n      }\n    } catch (error) {\n      logger.error(\"Genie getConversation error: %O\", error);\n      yield {\n        type: \"error\",\n        error:\n          error instanceof Error\n            ? error.message\n            : \"Failed to fetch conversation\",\n      };\n    }\n  }\n\n  async sendMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n  ): Promise<GenieMessageResponse> {\n    const { messageWaiter, conversationId: resultConversationId } =\n      await this.startMessage(\n        workspaceClient,\n        spaceId,\n        content,\n        conversationId,\n      );\n    const completedMessage = await this.waitForMessage(messageWaiter);\n    const messageResponse = toMessageResponse(completedMessage);\n    return {\n      ...messageResponse,\n      conversationId: resultConversationId,\n    };\n  }\n\n  async getConversation(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n  ): Promise<GenieConversationHistoryResponse> {\n    const messages = await this.listConversationMessages(\n      workspaceClient,\n      spaceId,\n      conversationId,\n    );\n    return {\n      conversationId,\n      spaceId,\n      messages,\n    };\n  }\n}\n"],"mappings":";;;;;;;AAcA,MAAM,SAAS,aAAa,mBAAmB;AAS/C,SAAS,eAAe,SAAkD;AACxE,QACE,QAAQ,aAAa,KAAK,SAAS;EACjC,cAAc,IAAI;EAClB,OAAO,IAAI,QACP;GACE,OAAO,IAAI,MAAM;GACjB,aAAa,IAAI,MAAM;GACvB,OAAO,IAAI,MAAM;GACjB,aAAa,IAAI,MAAM;GACxB,GACD;EACJ,MAAM,IAAI,OAAO,EAAE,SAAS,IAAI,KAAK,SAAS,GAAG;EACjD,oBAAoB,IAAI,qBAAqB;EAC9C,EAAE,IAAI,EAAE;;AAIb,SAAS,kBAAkB,SAA6C;AACtE,QAAO;EACL,WAAW,QAAQ;EACnB,gBAAgB,QAAQ;EACxB,SAAS,QAAQ;EACjB,QAAQ,QAAQ,UAAU;EAC1B,SAAS,QAAQ;EACjB,aAAa,eAAe,QAAQ;EACpC,OAAO,QAAQ,OAAO;EACvB;;AAGH,IAAa,iBAAb,MAA4B;CAC1B,AAAiB;CAEjB,YAAY,SAA+B,EAAE,EAAE;AAC7C,OAAK,SAAS;GACZ,SAAS,OAAO,WAAW,uBAAuB;GAClD,aAAa,OAAO,eAAe,uBAAuB;GAC3D;;CAGH,MAAM,aACJ,iBACA,SACA,SACA,gBAKC;AACD,MAAI,gBAAgB;GAClB,MAAM,SAAS,MAAM,gBAAgB,MAAM,cAAc;IACvD,UAAU;IACV,iBAAiB;IACjB;IACD,CAAC;AACF,UAAO;IACL,eAAe;IACf;IACA,WAAW,OAAO,cAAc;IACjC;;EAEH,MAAM,QAAQ,MAAM,gBAAgB,MAAM,kBAAkB;GAC1D,UAAU;GACV;GACD,CAAC;AACF,SAAO;GACL,eAAe;GACf,gBAAgB,MAAM;GACtB,WAAW,MAAM;GAClB;;CAGH,MAAM,eACJ,eACA,SACuB;EACvB,MAAM,UAAU,SAAS,WAAW,KAAK,OAAO;EAChD,MAAM,cACJ,UAAU,IAAI,EAAE,SAAS,IAAI,KAAK,SAAS,UAAU,aAAa,EAAE,GAAG,EAAE;AAC3E,SAAO,cAAc,KAAK,YAAY;;CAGxC,MAAM,yBACJ,iBACA,SACA,gBACA,SACiC;EACjC,MAAM,cAAc,SAAS,eAAe,KAAK,OAAO;EACxD,MAAM,cAA8B,EAAE;EACtC,IAAI;AAEJ,KAAG;GACD,MAAM,WAAW,MAAM,gBAAgB,MAAM,yBAAyB;IACpE,UAAU;IACV,iBAAiB;IACjB,WAAW,uBAAuB;IAClC,GAAI,YAAY,EAAE,YAAY,WAAW,GAAG,EAAE;IAC/C,CAAC;AAEF,OAAI,SAAS,SACX,aAAY,KAAK,GAAG,SAAS,SAAS;AAGxC,eAAY,SAAS;WACd,aAAa,YAAY,SAAS;AAE3C,SAAO,YAAY,MAAM,GAAG,YAAY,CAAC,SAAS,CAAC,IAAI,kBAAkB;;CAG3E,MAAM,gCACJ,iBACA,SACA,gBACA,WACA,cACA,SACkB;AAQlB,UANE,MAAM,gBAAgB,MAAM,gCAAgC;GAC1D,UAAU;GACV,iBAAiB;GACjB,YAAY;GACZ,eAAe;GAChB,CAAC,EACY;;CAGlB,OAAO,kBACL,iBACA,SACA,SACA,gBACA,SACkC;AAClC,MAAI;GACF,MAAM,EACJ,eACA,gBAAgB,sBAChB,WAAW,oBACT,MAAM,KAAK,aACb,iBACA,SACA,SACA,eACD;AAED,SAAM;IACJ,MAAM;IACN,gBAAgB;IAChB,WAAW;IACX;IACD;GAED,MAAM,UACJ,SAAS,WAAW,OAAO,QAAQ,UAAU,KAAK,OAAO;GAC3D,MAAM,cACJ,UAAU,IACN,EAAE,SAAS,IAAI,KAAK,SAAS,UAAU,aAAa,EAAE,GACtD,EAAE;GAER,IAAI;AACJ,cAAW,MAAM,SAAS,WAAW,eAAe,YAAY,CAC9D,KAAI,MAAM,SAAS,cAAc,MAAM,MAAM,OAC3C,OAAM;IAAE,MAAM;IAAU,QAAQ,MAAM,MAAM;IAAQ;YAC3C,MAAM,SAAS,YACxB,oBAAmB,MAAM;GAI7B,MAAM,kBAAkB,kBAAkB,iBAAiB;AAC3D,SAAM;IAAE,MAAM;IAAkB,SAAS;IAAiB;AAE1D,UAAO,KAAK,iBACV,iBACA,SACA,sBACA,gBAAgB,WAChB,gBACD;WACM,OAAO;AACd,UAAO,MAAM,2BAA2B,MAAM;AAC9C,SAAM;IACJ,MAAM;IACN,OAAO,iBAAiB,QAAQ,MAAM,UAAU;IACjD;;;CAIL,OAAe,iBACb,iBACA,SACA,gBACA,WACA,iBAGA;EACA,MAAM,cAAc,gBAAgB,eAAe,EAAE;AACrD,OAAK,MAAM,OAAO,aAAa;AAC7B,OAAI,CAAC,IAAI,OAAO,eAAe,CAAC,IAAI,aAAc;AAClD,OAAI;IACF,MAAM,OAAO,MAAM,KAAK,gCACtB,iBACA,SACA,gBACA,WACA,IAAI,aACL;AACD,UAAM;KACJ,MAAM;KACN,cAAc,IAAI;KAClB,aAAa,IAAI,MAAM;KACvB;KACD;YACM,OAAO;AACd,WAAO,MACL,sDACA,IAAI,cACJ,MACD;AACD,UAAM;KACJ,MAAM;KACN,OAAO,+CAA+C,IAAI;KAC3D;;;;CAKP,OAAO,mBACL,iBACA,SACA,gBACA,SACkC;EAClC,MAAM,sBAAsB,SAAS,wBAAwB;AAE7D,MAAI;GACF,MAAM,mBAAmB,MAAM,KAAK,yBAClC,iBACA,SACA,eACD;AAED,QAAK,MAAM,mBAAmB,iBAC5B,OAAM;IAAE,MAAM;IAAkB,SAAS;IAAiB;AAG5D,OAAI,qBAAqB;IACvB,MAAM,mBAID,EAAE;AAEP,SAAK,MAAM,OAAO,iBAChB,MAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CACrC,KAAI,IAAI,OAAO,eAAe,IAAI,aAChC,kBAAiB,KAAK;KACpB,WAAW,IAAI;KACf,cAAc,IAAI;KAClB,aAAa,IAAI,MAAM;KACxB,CAAC;IAKR,MAAM,UAAU,MAAM,QAAQ,WAC5B,iBAAiB,IAAI,OAAO,QAAQ;KAClC,MAAM,OAAO,MAAM,KAAK,gCACtB,iBACA,SACA,gBACA,IAAI,WACJ,IAAI,aACL;AACD,YAAO;MACL,cAAc,IAAI;MAClB,aAAa,IAAI;MACjB;MACD;MACD,CACH;AAED,SAAK,MAAM,UAAU,QACnB,KAAI,OAAO,WAAW,YACpB,OAAM;KACJ,MAAM;KACN,cAAc,OAAO,MAAM;KAC3B,aAAa,OAAO,MAAM;KAC1B,MAAM,OAAO,MAAM;KACpB;SACI;AACL,YAAO,MAAM,oCAAoC,OAAO,OAAO;AAC/D,WAAM;MACJ,MAAM;MACN,OACE,OAAO,kBAAkB,QACrB,OAAO,OAAO,UACd;MACP;;;WAIA,OAAO;AACd,UAAO,MAAM,mCAAmC,MAAM;AACtD,SAAM;IACJ,MAAM;IACN,OACE,iBAAiB,QACb,MAAM,UACN;IACP;;;CAIL,MAAM,YACJ,iBACA,SACA,SACA,gBAC+B;EAC/B,MAAM,EAAE,eAAe,gBAAgB,yBACrC,MAAM,KAAK,aACT,iBACA,SACA,SACA,eACD;AAGH,SAAO;GACL,GAFsB,kBADC,MAAM,KAAK,eAAe,cAAc,CACN;GAGzD,gBAAgB;GACjB;;CAGH,MAAM,gBACJ,iBACA,SACA,gBAC2C;AAM3C,SAAO;GACL;GACA;GACA,UARe,MAAM,KAAK,yBAC1B,iBACA,SACA,eACD;GAKA"}

package/dist/connectors/genie/defaults.js

@@ -0,0 +1,10 @@
+//#region src/connectors/genie/defaults.ts
+const genieConnectorDefaults = {
+	timeout: 12e4,
+	maxMessages: 200,
+	pageSize: 100
+};
+
+//#endregion
+export { genieConnectorDefaults };
+//# sourceMappingURL=defaults.js.map

package/dist/connectors/genie/defaults.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"defaults.js","names":[],"sources":["../../../src/connectors/genie/defaults.ts"],"sourcesContent":["export const genieConnectorDefaults = {\n  /** Genie waiter timeout in ms. 0 = indefinite. */\n  timeout: 120_000,\n  /** Max messages to fetch when listing a conversation. */\n  maxMessages: 200,\n  /** Default page size for listConversationMessages. */\n  pageSize: 100,\n} as const;\n"],"mappings":";AAAA,MAAa,yBAAyB;CAEpC,SAAS;CAET,aAAa;CAEb,UAAU;CACX"}

package/dist/connectors/genie/index.js

@@ -0,0 +1,5 @@
+import { genieConnectorDefaults } from "./defaults.js";
+import { pollWaiter } from "./poll-waiter.js";
+import { GenieConnector } from "./client.js";
+
+export {  };

package/dist/connectors/genie/poll-waiter.js

@@ -0,0 +1,54 @@
+//#region src/connectors/genie/poll-waiter.ts
+/**
+* Bridges a callback-based waiter into an async generator.
+*
+* The SDK's `waiter.wait({ onProgress })` API uses a callback to report
+* progress and returns a promise that resolves with the final result.
+* This function converts that push-based model into a pull-based async
+* generator so callers can simply `for await (const event of pollWaiter(w))`.
+*
+* Yields `{ type: "progress", value }` for each `onProgress` callback,
+* then `{ type: "completed", value }` for the final result.
+* Throws if the waiter rejects.
+*/
+async function* pollWaiter(waiter, options) {
+	const queue = [];
+	let notify = () => {};
+	let done = false;
+	let result;
+	let error = null;
+	waiter.wait({
+		onProgress: async (p) => {
+			queue.push(p);
+			notify();
+		},
+		...options?.timeout != null ? { timeout: options.timeout } : {}
+	}).then((r) => {
+		result = r;
+		done = true;
+		notify();
+	}).catch((err) => {
+		error = err;
+		done = true;
+		notify();
+	});
+	while (!done || queue.length > 0) {
+		while (queue.length > 0) yield {
+			type: "progress",
+			value: queue.shift()
+		};
+		if (!done) await new Promise((resolve) => {
+			notify = resolve;
+			if (done || queue.length > 0) resolve();
+		});
+	}
+	if (error !== null) throw error;
+	yield {
+		type: "completed",
+		value: result
+	};
+}
+
+//#endregion
+export { pollWaiter };
+//# sourceMappingURL=poll-waiter.js.map

package/dist/connectors/genie/poll-waiter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"poll-waiter.js","names":[],"sources":["../../../src/connectors/genie/poll-waiter.ts"],"sourcesContent":["/**\n * Structural interface matching the SDK's `Waiter.wait()` shape\n * without importing the SDK directly.\n */\nexport interface Pollable<P> {\n  wait(options?: {\n    onProgress?: (p: P) => Promise<void>;\n    timeout?: unknown;\n  }): Promise<P>;\n}\n\nexport type PollEvent<P> =\n  | { type: \"progress\"; value: P }\n  | { type: \"completed\"; value: P };\n\n/**\n * Bridges a callback-based waiter into an async generator.\n *\n * The SDK's `waiter.wait({ onProgress })` API uses a callback to report\n * progress and returns a promise that resolves with the final result.\n * This function converts that push-based model into a pull-based async\n * generator so callers can simply `for await (const event of pollWaiter(w))`.\n *\n * Yields `{ type: \"progress\", value }` for each `onProgress` callback,\n * then `{ type: \"completed\", value }` for the final result.\n * Throws if the waiter rejects.\n */\nexport async function* pollWaiter<P>(\n  waiter: Pollable<P>,\n  options?: { timeout?: unknown },\n): AsyncGenerator<PollEvent<P>> {\n  const queue: P[] = [];\n  let notify: () => void = () => {};\n  let done = false;\n  let result!: P;\n  let error: unknown = null;\n\n  waiter\n    .wait({\n      onProgress: async (p: P) => {\n        queue.push(p);\n        notify();\n      },\n      ...(options?.timeout != null ? { timeout: options.timeout } : {}),\n    })\n    .then((r) => {\n      result = r;\n      done = true;\n      notify();\n    })\n    .catch((err) => {\n      error = err;\n      done = true;\n      notify();\n    });\n\n  while (!done || queue.length > 0) {\n    while (queue.length > 0) {\n      const value = queue.shift() as P;\n      yield { type: \"progress\", value };\n    }\n\n    if (!done) {\n      await new Promise<void>((resolve) => {\n        notify = resolve;\n        if (done || queue.length > 0) resolve();\n      });\n    }\n  }\n\n  if (error !== null) {\n    throw error;\n  }\n\n  yield { type: \"completed\", value: result };\n}\n"],"mappings":";;;;;;;;;;;;;AA2BA,gBAAuB,WACrB,QACA,SAC8B;CAC9B,MAAM,QAAa,EAAE;CACrB,IAAI,eAA2B;CAC/B,IAAI,OAAO;CACX,IAAI;CACJ,IAAI,QAAiB;AAErB,QACG,KAAK;EACJ,YAAY,OAAO,MAAS;AAC1B,SAAM,KAAK,EAAE;AACb,WAAQ;;EAEV,GAAI,SAAS,WAAW,OAAO,EAAE,SAAS,QAAQ,SAAS,GAAG,EAAE;EACjE,CAAC,CACD,MAAM,MAAM;AACX,WAAS;AACT,SAAO;AACP,UAAQ;GACR,CACD,OAAO,QAAQ;AACd,UAAQ;AACR,SAAO;AACP,UAAQ;GACR;AAEJ,QAAO,CAAC,QAAQ,MAAM,SAAS,GAAG;AAChC,SAAO,MAAM,SAAS,EAEpB,OAAM;GAAE,MAAM;GAAY,OADZ,MAAM,OAAO;GACM;AAGnC,MAAI,CAAC,KACH,OAAM,IAAI,SAAe,YAAY;AACnC,YAAS;AACT,OAAI,QAAQ,MAAM,SAAS,EAAG,UAAS;IACvC;;AAIN,KAAI,UAAU,KACZ,OAAM;AAGR,OAAM;EAAE,MAAM;EAAa,OAAO;EAAQ"}

package/dist/connectors/genie/types.d.ts

@@ -0,0 +1,11 @@
+import { GenieAttachmentResponse, GenieMessageResponse, GenieStreamEvent } from "../../shared/src/genie.js";
+
+//#region src/connectors/genie/types.d.ts
+interface GenieConversationHistoryResponse {
+  conversationId: string;
+  spaceId: string;
+  messages: GenieMessageResponse[];
+}
+//#endregion
+export { GenieConversationHistoryResponse };
+//# sourceMappingURL=types.d.ts.map

package/dist/connectors/genie/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/connectors/genie/types.ts"],"sourcesContent":[],"mappings":";;;AAUiB,UAAA,gCAAA,CAAgC;;;YAGrC"}

package/dist/connectors/index.js

@@ -1,4 +1,8 @@
 import { RequestedClaimsPermissionSet, createLakebasePool, generateDatabaseCredential, getLakebaseOrmConfig, getLakebasePgConfig, getUsernameWithApiLookup, getWorkspaceClient } from "./lakebase/index.js";
+import { genieConnectorDefaults } from "./genie/defaults.js";
+import { pollWaiter } from "./genie/poll-waiter.js";
+import { GenieConnector } from "./genie/client.js";
+import "./genie/index.js";
 import "./lakebase-v1/index.js";
 import { SQLWarehouseConnector } from "./sql-warehouse/client.js";
 import "./sql-warehouse/index.js";

package/dist/index.d.ts

@@ -24,7 +24,8 @@ import { ResourcePermission, ResourceType } from "./registry/types.generated.js"
 import { ConfigSchema, PluginManifest, ResourceEntry, ResourceFieldEntry, ResourceRequirement, ValidationResult } from "./registry/types.js";
 import { getPluginManifest, getResourceRequirements } from "./registry/manifest-loader.js";
 import { ResourceRegistry } from "./registry/resource-registry.js";
+import { genie } from "./plugins/genie/genie.js";
 import { lakebase } from "./plugins/lakebase/lakebase.js";
 import { server } from "./plugins/server/index.js";
 import { appKitTypesPlugin } from "./type-generator/vite-plugin.js";
-export { AppKitError, AuthenticationError, type BasePluginConfig, type CacheConfig, CacheManager, type ConfigSchema, ConfigurationError, ConnectionError, type Counter, type DatabaseCredential, ExecutionError, type GenerateDatabaseCredentialRequest, type Histogram, type IAppRouter, type ITelemetry, InitializationError, type LakebasePoolConfig, Plugin, type PluginManifest, type RequestedClaims, RequestedClaimsPermissionSet, type RequestedResource, type ResourceEntry, type ResourceFieldEntry, type ResourcePermission, ResourceRegistry, type ResourceRequirement, ResourceType, ServerError, SeverityNumber, type Span, SpanStatusCode, type StreamExecutionSettings, type TelemetryConfig, type ToPlugin, TunnelError, ValidationError, type ValidationResult, analytics, appKitTypesPlugin, createApp, createLakebasePool, generateDatabaseCredential, getExecutionContext, getLakebaseOrmConfig, getLakebasePgConfig, getPluginManifest, getResourceRequirements, getUsernameWithApiLookup, getWorkspaceClient, isSQLTypeMarker, lakebase, server, sql, toPlugin };
+export { AppKitError, AuthenticationError, type BasePluginConfig, type CacheConfig, CacheManager, type ConfigSchema, ConfigurationError, ConnectionError, type Counter, type DatabaseCredential, ExecutionError, type GenerateDatabaseCredentialRequest, type Histogram, type IAppRouter, type ITelemetry, InitializationError, type LakebasePoolConfig, Plugin, type PluginManifest, type RequestedClaims, RequestedClaimsPermissionSet, type RequestedResource, type ResourceEntry, type ResourceFieldEntry, type ResourcePermission, ResourceRegistry, type ResourceRequirement, ResourceType, ServerError, SeverityNumber, type Span, SpanStatusCode, type StreamExecutionSettings, type TelemetryConfig, type ToPlugin, TunnelError, ValidationError, type ValidationResult, analytics, appKitTypesPlugin, createApp, createLakebasePool, generateDatabaseCredential, genie, getExecutionContext, getLakebaseOrmConfig, getLakebasePgConfig, getPluginManifest, getResourceRequirements, getUsernameWithApiLookup, getWorkspaceClient, isSQLTypeMarker, lakebase, server, sql, toPlugin };

package/dist/index.js

@@ -24,6 +24,7 @@ import { Plugin } from "./plugin/plugin.js";
 import { toPlugin } from "./plugin/to-plugin.js";
 import "./plugin/index.js";
 import { analytics } from "./plugins/analytics/analytics.js";
+import { genie } from "./plugins/genie/genie.js";
 import { lakebase } from "./plugins/lakebase/lakebase.js";
 import { appKitTypesPlugin } from "./type-generator/vite-plugin.js";
 import { server } from "./plugins/server/index.js";
@@ -34,5 +35,5 @@ init_context();
 init_errors();
 
 //#endregion
-export { AppKitError, AuthenticationError, CacheManager, ConfigurationError, ConnectionError, ExecutionError, InitializationError, Plugin, RequestedClaimsPermissionSet, ResourceRegistry, ResourceType, ServerError, SeverityNumber, SpanStatusCode, TunnelError, ValidationError, analytics, appKitTypesPlugin, createApp, createLakebasePool, generateDatabaseCredential, getExecutionContext, getLakebaseOrmConfig, getLakebasePgConfig, getPluginManifest, getResourceRequirements, getUsernameWithApiLookup, getWorkspaceClient, isSQLTypeMarker, lakebase, server, sql, toPlugin };
+export { AppKitError, AuthenticationError, CacheManager, ConfigurationError, ConnectionError, ExecutionError, InitializationError, Plugin, RequestedClaimsPermissionSet, ResourceRegistry, ResourceType, ServerError, SeverityNumber, SpanStatusCode, TunnelError, ValidationError, analytics, appKitTypesPlugin, createApp, createLakebasePool, generateDatabaseCredential, genie, getExecutionContext, getLakebaseOrmConfig, getLakebasePgConfig, getPluginManifest, getResourceRequirements, getUsernameWithApiLookup, getWorkspaceClient, isSQLTypeMarker, lakebase, server, sql, toPlugin };
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["/**\n * @packageDocumentation\n *\n * Core library for building Databricks applications with type-safe SQL queries,\n * plugin architecture, and React integration.\n */\n\n// Types from shared\nexport type {\n  BasePluginConfig,\n  CacheConfig,\n  IAppRouter,\n  StreamExecutionSettings,\n} from \"shared\";\nexport { isSQLTypeMarker, sql } from \"shared\";\nexport { CacheManager } from \"./cache\";\nexport type {\n  DatabaseCredential,\n  GenerateDatabaseCredentialRequest,\n  LakebasePoolConfig,\n  RequestedClaims,\n  RequestedResource,\n} from \"./connectors/lakebase\";\n// Lakebase Autoscaling connector\nexport {\n  createLakebasePool,\n  generateDatabaseCredential,\n  getLakebaseOrmConfig,\n  getLakebasePgConfig,\n  getUsernameWithApiLookup,\n  getWorkspaceClient,\n  RequestedClaimsPermissionSet,\n} from \"./connectors/lakebase\";\nexport { getExecutionContext } from \"./context\";\nexport { createApp } from \"./core\";\n// Errors\nexport {\n  AppKitError,\n  AuthenticationError,\n  ConfigurationError,\n  ConnectionError,\n  ExecutionError,\n  InitializationError,\n  ServerError,\n  TunnelError,\n  ValidationError,\n} from \"./errors\";\n// Plugin authoring\nexport { Plugin, type ToPlugin, toPlugin } from \"./plugin\";\nexport { analytics, lakebase, server } from \"./plugins\";\n// Registry types and utilities for plugin manifests\nexport type {\n  ConfigSchema,\n  PluginManifest,\n  ResourceEntry,\n  ResourceFieldEntry,\n  ResourcePermission,\n  ResourceRequirement,\n  ValidationResult,\n} from \"./registry\";\nexport {\n  getPluginManifest,\n  getResourceRequirements,\n  ResourceRegistry,\n  ResourceType,\n} from \"./registry\";\n// Telemetry (for advanced custom telemetry)\nexport {\n  type Counter,\n  type Histogram,\n  type ITelemetry,\n  SeverityNumber,\n  type Span,\n  SpanStatusCode,\n  type TelemetryConfig,\n} from \"./telemetry\";\n// Vite plugin and type generation\nexport { appKitTypesPlugin } from \"./type-generator/vite-plugin\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAiCgD;aAa9B"}
+{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["/**\n * @packageDocumentation\n *\n * Core library for building Databricks applications with type-safe SQL queries,\n * plugin architecture, and React integration.\n */\n\n// Types from shared\nexport type {\n  BasePluginConfig,\n  CacheConfig,\n  IAppRouter,\n  StreamExecutionSettings,\n} from \"shared\";\nexport { isSQLTypeMarker, sql } from \"shared\";\nexport { CacheManager } from \"./cache\";\nexport type {\n  DatabaseCredential,\n  GenerateDatabaseCredentialRequest,\n  LakebasePoolConfig,\n  RequestedClaims,\n  RequestedResource,\n} from \"./connectors/lakebase\";\n// Lakebase Autoscaling connector\nexport {\n  createLakebasePool,\n  generateDatabaseCredential,\n  getLakebaseOrmConfig,\n  getLakebasePgConfig,\n  getUsernameWithApiLookup,\n  getWorkspaceClient,\n  RequestedClaimsPermissionSet,\n} from \"./connectors/lakebase\";\nexport { getExecutionContext } from \"./context\";\nexport { createApp } from \"./core\";\n// Errors\nexport {\n  AppKitError,\n  AuthenticationError,\n  ConfigurationError,\n  ConnectionError,\n  ExecutionError,\n  InitializationError,\n  ServerError,\n  TunnelError,\n  ValidationError,\n} from \"./errors\";\n// Plugin authoring\nexport { Plugin, type ToPlugin, toPlugin } from \"./plugin\";\nexport { analytics, genie, lakebase, server } from \"./plugins\";\n// Registry types and utilities for plugin manifests\nexport type {\n  ConfigSchema,\n  PluginManifest,\n  ResourceEntry,\n  ResourceFieldEntry,\n  ResourcePermission,\n  ResourceRequirement,\n  ValidationResult,\n} from \"./registry\";\nexport {\n  getPluginManifest,\n  getResourceRequirements,\n  ResourceRegistry,\n  ResourceType,\n} from \"./registry\";\n// Telemetry (for advanced custom telemetry)\nexport {\n  type Counter,\n  type Histogram,\n  type ITelemetry,\n  SeverityNumber,\n  type Span,\n  SpanStatusCode,\n  type TelemetryConfig,\n} from \"./telemetry\";\n// Vite plugin and type generation\nexport { appKitTypesPlugin } from \"./type-generator/vite-plugin\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAiCgD;aAa9B"}

package/dist/logging/index.js

@@ -0,0 +1,16 @@
+import { DEFAULT_SAMPLING_CONFIG, shouldSample } from "./sampling.js";
+import { WideEvent } from "./wide-event.js";
+import { WideEventEmitter } from "./wide-event-emitter.js";
+import { createLogger } from "./logger.js";
+import { AppKitError } from "../errors/base.js";
+import { AuthenticationError } from "../errors/authentication.js";
+import { ConfigurationError } from "../errors/configuration.js";
+import { ConnectionError } from "../errors/connection.js";
+import { ExecutionError } from "../errors/execution.js";
+import { InitializationError } from "../errors/initialization.js";
+import { ServerError } from "../errors/server.js";
+import { TunnelError } from "../errors/tunnel.js";
+import { ValidationError } from "../errors/validation.js";
+import "@opentelemetry/api";
+
+export {  };

package/dist/plugins/genie/defaults.js

@@ -0,0 +1,13 @@
+//#region src/plugins/genie/defaults.ts
+const genieStreamDefaults = {
+	default: {
+		cache: { enabled: false },
+		retry: { enabled: false },
+		timeout: 12e4
+	},
+	stream: { bufferSize: 100 }
+};
+
+//#endregion
+export { genieStreamDefaults };
+//# sourceMappingURL=defaults.js.map

package/dist/plugins/genie/defaults.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"defaults.js","names":[],"sources":["../../../src/plugins/genie/defaults.ts"],"sourcesContent":["import type { StreamExecutionSettings } from \"shared\";\n\nexport const genieStreamDefaults: StreamExecutionSettings = {\n  default: {\n    // Cache disabled: chat messages are conversational and stateful, not repeatable queries.\n    cache: {\n      enabled: false,\n    },\n    // Retry disabled: Genie calls are not idempotent (retries could create duplicate\n    // conversations/messages), and the SDK Waiter already handles transient polling failures.\n    retry: {\n      enabled: false,\n    },\n    timeout: 120_000,\n  },\n  stream: {\n    bufferSize: 100,\n  },\n};\n"],"mappings":";AAEA,MAAa,sBAA+C;CAC1D,SAAS;EAEP,OAAO,EACL,SAAS,OACV;EAGD,OAAO,EACL,SAAS,OACV;EACD,SAAS;EACV;CACD,QAAQ,EACN,YAAY,KACb;CACF"}

package/dist/plugins/genie/genie.d.ts

@@ -0,0 +1,44 @@
+import { IAppRouter, ToPlugin } from "../../shared/src/plugin.js";
+import { GenieStreamEvent } from "../../shared/src/genie.js";
+import { Plugin } from "../../plugin/plugin.js";
+import { PluginManifest } from "../../registry/types.js";
+import { GenieConversationHistoryResponse } from "../../connectors/genie/types.js";
+import { IGenieConfig } from "./types.js";
+import express from "express";
+
+//#region src/plugins/genie/genie.d.ts
+declare class GeniePlugin extends Plugin {
+  name: string;
+  static manifest: PluginManifest;
+  protected static description: string;
+  protected config: IGenieConfig;
+  private readonly genieConnector;
+  constructor(config: IGenieConfig);
+  private defaultSpaces;
+  private resolveSpaceId;
+  injectRoutes(router: IAppRouter): void;
+  _handleSendMessage(req: express.Request, res: express.Response): Promise<void>;
+  _handleGetConversation(req: express.Request, res: express.Response): Promise<void>;
+  getConversation(alias: string, conversationId: string): Promise<GenieConversationHistoryResponse>;
+  /**
+   * Send a message and consume events as a stream (message_start, status,
+   * message_result, query_result, error).
+   */
+  sendMessage(alias: string, content: string, conversationId?: string, options?: {
+    timeout?: number;
+  }): AsyncGenerator<GenieStreamEvent>;
+  shutdown(): Promise<void>;
+  exports(): {
+    sendMessage: (alias: string, content: string, conversationId?: string, options?: {
+      timeout?: number;
+    }) => AsyncGenerator<GenieStreamEvent>;
+    getConversation: (alias: string, conversationId: string) => Promise<GenieConversationHistoryResponse>;
+  };
+}
+/**
+ * @internal
+ */
+declare const genie: ToPlugin<typeof GeniePlugin, IGenieConfig, "genie">;
+//#endregion
+export { genie };
+//# sourceMappingURL=genie.d.ts.map

package/dist/plugins/genie/genie.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"genie.d.ts","names":[],"sources":["../../../src/plugins/genie/genie.ts"],"sourcesContent":[],"mappings":";;;;;;;;;cAkBa,WAAA,SAAoB,MAAA;;mBAAR;;oBAOG;EAPf,iBAAY,cAAA;EAAA,WAAA,CAAA,MAAA,EAWH,YAXG;UAAA,aAAA;UAOG,cAAA;cAIN,CAAA,MAAA,EAqBC,UArBD,CAAA,EAAA,IAAA;oBAqBC,CAAA,GAAA,EAqBd,OAAA,CAAQ,OArBM,EAAA,GAAA,EAsBd,OAAA,CAAQ,QAtBM,CAAA,EAuBlB,OAvBkB,CAAA,IAAA,CAAA;wBAqBN,CAAA,GAAA,EAyDR,OAAA,CAAQ,OAzDA,EAAA,GAAA,EA0DR,OAAA,CAAQ,QA1DA,CAAA,EA2DZ,OA3DY,CAAA,IAAA,CAAA;iBACA,CAAA,KAAA,EAAA,MAAA,EAAA,cAAA,EAAA,MAAA,CAAA,EAwGZ,OAxGY,CAwGJ,gCAxGI,CAAA;;;;;aAwGJ,CAAA,KAAA,EAAA,MAAA,EAAA,OAAA,EAAA,MAAA,EAAA,cAAA,CAAA,EAAA,MAAA,EAAA,QAAA,EAAA;IAAR,OAAA,CAAA,EAAA,MAAA;MAyBA,cAAe,CAAA,gBAAA,CAAA;UAAf,CAAA,CAAA,EAgBe,OAhBf,CAAA,IAAA,CAAA;SAgBe,CAAA,CAAA,EAAA;IAhBA,WAAA,EAAA,CAAA,KAAA,EAAA,MAAA,EAAA,OAAA,EAAA,MAAA,EAAA,cAAA,CAAA,EAAA,MAAA,EAAA,OAzBP,CAyBO,EAAA;MAAf,OAAA,CAAA,EAAA,MAAA;IAzBQ,CAAA,EAAA,GAyBR,cAzBQ,CAyBO,gBAzBP,CAAA;IAAR,eAAA,EAAA,CAAA,KAAA,EAAA,MAAA,EAAA,cAAA,EAAA,MAAA,EAAA,GAAA,OAAA,CAAQ,gCAAR,CAAA;;;AAwDL;;;AAAkB,cAAL,KAAK,EAAA,QAAA,CAAA,OAAA,WAAA,EAAA,YAAA,EAAA,OAAA,CAAA"}

package/dist/plugins/genie/genie.js

@@ -0,0 +1,141 @@
+import { createLogger } from "../../logging/logger.js";
+import { getWorkspaceClient } from "../../context/execution-context.js";
+import { init_context } from "../../context/index.js";
+import { Plugin } from "../../plugin/plugin.js";
+import { toPlugin } from "../../plugin/to-plugin.js";
+import "../../plugin/index.js";
+import "../../logging/index.js";
+import { GenieConnector } from "../../connectors/genie/client.js";
+import "../../connectors/index.js";
+import { genieStreamDefaults } from "./defaults.js";
+import { genieManifest } from "./manifest.js";
+import { randomUUID } from "node:crypto";
+
+//#region src/plugins/genie/genie.ts
+init_context();
+const logger = createLogger("genie");
+var GeniePlugin = class extends Plugin {
+	name = "genie";
+	static manifest = genieManifest;
+	static description = "AI/BI Genie space integration for natural language data queries";
+	genieConnector;
+	constructor(config) {
+		super(config);
+		this.config = {
+			...config,
+			spaces: config.spaces ?? this.defaultSpaces()
+		};
+		this.genieConnector = new GenieConnector({
+			timeout: this.config.timeout,
+			maxMessages: 200
+		});
+	}
+	defaultSpaces() {
+		const spaceId = process.env.DATABRICKS_GENIE_SPACE_ID;
+		return spaceId ? { default: spaceId } : {};
+	}
+	resolveSpaceId(alias) {
+		return this.config.spaces?.[alias] ?? null;
+	}
+	injectRoutes(router) {
+		this.route(router, {
+			name: "sendMessage",
+			method: "post",
+			path: "/:alias/messages",
+			handler: async (req, res) => {
+				await this.asUser(req)._handleSendMessage(req, res);
+			}
+		});
+		this.route(router, {
+			name: "getConversation",
+			method: "get",
+			path: "/:alias/conversations/:conversationId",
+			handler: async (req, res) => {
+				await this.asUser(req)._handleGetConversation(req, res);
+			}
+		});
+	}
+	async _handleSendMessage(req, res) {
+		const { alias } = req.params;
+		const spaceId = this.resolveSpaceId(alias);
+		if (!spaceId) {
+			res.status(404).json({ error: `Unknown space alias: ${alias}` });
+			return;
+		}
+		const { content, conversationId } = req.body;
+		if (!content) {
+			res.status(400).json({ error: "content is required" });
+			return;
+		}
+		logger.debug("Sending message to space %s (alias=%s, conversationId=%s)", spaceId, alias, conversationId ?? "new");
+		const timeout = this.config.timeout ?? 12e4;
+		const requestId = req.query.requestId || randomUUID();
+		const streamSettings = {
+			...genieStreamDefaults,
+			default: {
+				...genieStreamDefaults.default,
+				timeout
+			},
+			stream: {
+				...genieStreamDefaults.stream,
+				streamId: requestId
+			}
+		};
+		const workspaceClient = getWorkspaceClient();
+		await this.executeStream(res, () => this.genieConnector.streamSendMessage(workspaceClient, spaceId, content, conversationId, { timeout }), streamSettings);
+	}
+	async _handleGetConversation(req, res) {
+		const { alias, conversationId } = req.params;
+		const spaceId = this.resolveSpaceId(alias);
+		if (!spaceId) {
+			res.status(404).json({ error: `Unknown space alias: ${alias}` });
+			return;
+		}
+		const includeQueryResults = req.query.includeQueryResults !== "false";
+		const requestId = req.query.requestId || randomUUID();
+		logger.debug("Fetching conversation %s from space %s (alias=%s, includeQueryResults=%s)", conversationId, spaceId, alias, includeQueryResults);
+		const streamSettings = {
+			...genieStreamDefaults,
+			stream: {
+				...genieStreamDefaults.stream,
+				streamId: requestId
+			}
+		};
+		const workspaceClient = getWorkspaceClient();
+		await this.executeStream(res, () => this.genieConnector.streamConversation(workspaceClient, spaceId, conversationId, { includeQueryResults }), streamSettings);
+	}
+	async getConversation(alias, conversationId) {
+		const spaceId = this.resolveSpaceId(alias);
+		if (!spaceId) throw new Error(`Unknown space alias: ${alias}`);
+		const workspaceClient = getWorkspaceClient();
+		return this.genieConnector.getConversation(workspaceClient, spaceId, conversationId);
+	}
+	/**
+	* Send a message and consume events as a stream (message_start, status,
+	* message_result, query_result, error).
+	*/
+	async *sendMessage(alias, content, conversationId, options) {
+		const spaceId = this.resolveSpaceId(alias);
+		if (!spaceId) throw new Error(`Unknown space alias: ${alias}`);
+		const workspaceClient = getWorkspaceClient();
+		const timeout = options?.timeout ?? this.config.timeout ?? 12e4;
+		yield* this.genieConnector.streamSendMessage(workspaceClient, spaceId, content, conversationId, { timeout });
+	}
+	async shutdown() {
+		this.streamManager.abortAll();
+	}
+	exports() {
+		return {
+			sendMessage: this.sendMessage,
+			getConversation: this.getConversation
+		};
+	}
+};
+/**
+* @internal
+*/
+const genie = toPlugin(GeniePlugin, "genie");
+
+//#endregion
+export { GeniePlugin, genie };
+//# sourceMappingURL=genie.js.map

package/dist/plugins/genie/genie.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"genie.js","names":[],"sources":["../../../src/plugins/genie/genie.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type express from \"express\";\nimport type { IAppRouter, StreamExecutionSettings } from \"shared\";\nimport { GenieConnector } from \"../../connectors\";\nimport { getWorkspaceClient } from \"../../context\";\nimport { createLogger } from \"../../logging\";\nimport { Plugin, toPlugin } from \"../../plugin\";\nimport { genieStreamDefaults } from \"./defaults\";\nimport { genieManifest } from \"./manifest\";\nimport type {\n  GenieConversationHistoryResponse,\n  GenieSendMessageRequest,\n  GenieStreamEvent,\n  IGenieConfig,\n} from \"./types\";\n\nconst logger = createLogger(\"genie\");\n\nexport class GeniePlugin extends Plugin {\n  name = \"genie\";\n\n  static manifest = genieManifest;\n\n  protected static description =\n    \"AI/BI Genie space integration for natural language data queries\";\n  protected declare config: IGenieConfig;\n\n  private readonly genieConnector: GenieConnector;\n\n  constructor(config: IGenieConfig) {\n    super(config);\n    this.config = {\n      ...config,\n      spaces: config.spaces ?? this.defaultSpaces(),\n    };\n    this.genieConnector = new GenieConnector({\n      timeout: this.config.timeout,\n      maxMessages: 200,\n    });\n  }\n\n  private defaultSpaces(): Record<string, string> {\n    const spaceId = process.env.DATABRICKS_GENIE_SPACE_ID;\n    return spaceId ? { default: spaceId } : {};\n  }\n\n  private resolveSpaceId(alias: string): string | null {\n    return this.config.spaces?.[alias] ?? null;\n  }\n\n  injectRoutes(router: IAppRouter) {\n    this.route(router, {\n      name: \"sendMessage\",\n      method: \"post\",\n      path: \"/:alias/messages\",\n      handler: async (req: express.Request, res: express.Response) => {\n        await this.asUser(req)._handleSendMessage(req, res);\n      },\n    });\n\n    this.route(router, {\n      name: \"getConversation\",\n      method: \"get\",\n      path: \"/:alias/conversations/:conversationId\",\n      handler: async (req: express.Request, res: express.Response) => {\n        await this.asUser(req)._handleGetConversation(req, res);\n      },\n    });\n  }\n\n  async _handleSendMessage(\n    req: express.Request,\n    res: express.Response,\n  ): Promise<void> {\n    const { alias } = req.params;\n    const spaceId = this.resolveSpaceId(alias);\n\n    if (!spaceId) {\n      res.status(404).json({ error: `Unknown space alias: ${alias}` });\n      return;\n    }\n\n    const { content, conversationId } = req.body as GenieSendMessageRequest;\n\n    if (!content) {\n      res.status(400).json({ error: \"content is required\" });\n      return;\n    }\n\n    logger.debug(\n      \"Sending message to space %s (alias=%s, conversationId=%s)\",\n      spaceId,\n      alias,\n      conversationId ?? \"new\",\n    );\n\n    const timeout = this.config.timeout ?? 120_000;\n    const requestId = (req.query.requestId as string) || randomUUID();\n\n    const streamSettings: StreamExecutionSettings = {\n      ...genieStreamDefaults,\n      default: {\n        ...genieStreamDefaults.default,\n        timeout,\n      },\n      stream: {\n        ...genieStreamDefaults.stream,\n        streamId: requestId,\n      },\n    };\n\n    const workspaceClient = getWorkspaceClient();\n\n    await this.executeStream<GenieStreamEvent>(\n      res,\n      () =>\n        this.genieConnector.streamSendMessage(\n          workspaceClient,\n          spaceId,\n          content,\n          conversationId,\n          { timeout },\n        ),\n      streamSettings,\n    );\n  }\n\n  async _handleGetConversation(\n    req: express.Request,\n    res: express.Response,\n  ): Promise<void> {\n    const { alias, conversationId } = req.params;\n    const spaceId = this.resolveSpaceId(alias);\n\n    if (!spaceId) {\n      res.status(404).json({ error: `Unknown space alias: ${alias}` });\n      return;\n    }\n\n    const includeQueryResults = req.query.includeQueryResults !== \"false\";\n    const requestId = (req.query.requestId as string) || randomUUID();\n\n    logger.debug(\n      \"Fetching conversation %s from space %s (alias=%s, includeQueryResults=%s)\",\n      conversationId,\n      spaceId,\n      alias,\n      includeQueryResults,\n    );\n\n    const streamSettings: StreamExecutionSettings = {\n      ...genieStreamDefaults,\n      stream: {\n        ...genieStreamDefaults.stream,\n        streamId: requestId,\n      },\n    };\n\n    const workspaceClient = getWorkspaceClient();\n\n    await this.executeStream<GenieStreamEvent>(\n      res,\n      () =>\n        this.genieConnector.streamConversation(\n          workspaceClient,\n          spaceId,\n          conversationId,\n          { includeQueryResults },\n        ),\n      streamSettings,\n    );\n  }\n\n  async getConversation(\n    alias: string,\n    conversationId: string,\n  ): Promise<GenieConversationHistoryResponse> {\n    const spaceId = this.resolveSpaceId(alias);\n\n    if (!spaceId) {\n      throw new Error(`Unknown space alias: ${alias}`);\n    }\n\n    const workspaceClient = getWorkspaceClient();\n\n    return this.genieConnector.getConversation(\n      workspaceClient,\n      spaceId,\n      conversationId,\n    );\n  }\n\n  /**\n   * Send a message and consume events as a stream (message_start, status,\n   * message_result, query_result, error).\n   */\n  async *sendMessage(\n    alias: string,\n    content: string,\n    conversationId?: string,\n    options?: { timeout?: number },\n  ): AsyncGenerator<GenieStreamEvent> {\n    const spaceId = this.resolveSpaceId(alias);\n    if (!spaceId) {\n      throw new Error(`Unknown space alias: ${alias}`);\n    }\n    const workspaceClient = getWorkspaceClient();\n    const timeout = options?.timeout ?? this.config.timeout ?? 120_000;\n    yield* this.genieConnector.streamSendMessage(\n      workspaceClient,\n      spaceId,\n      content,\n      conversationId,\n      { timeout },\n    );\n  }\n\n  async shutdown(): Promise<void> {\n    this.streamManager.abortAll();\n  }\n\n  exports() {\n    return {\n      sendMessage: this.sendMessage,\n      getConversation: this.getConversation,\n    };\n  }\n}\n\n/**\n * @internal\n */\nexport const genie = toPlugin<typeof GeniePlugin, IGenieConfig, \"genie\">(\n  GeniePlugin,\n  \"genie\",\n);\n"],"mappings":";;;;;;;;;;;;;;cAImD;AAYnD,MAAM,SAAS,aAAa,QAAQ;AAEpC,IAAa,cAAb,cAAiC,OAAO;CACtC,OAAO;CAEP,OAAO,WAAW;CAElB,OAAiB,cACf;CAGF,AAAiB;CAEjB,YAAY,QAAsB;AAChC,QAAM,OAAO;AACb,OAAK,SAAS;GACZ,GAAG;GACH,QAAQ,OAAO,UAAU,KAAK,eAAe;GAC9C;AACD,OAAK,iBAAiB,IAAI,eAAe;GACvC,SAAS,KAAK,OAAO;GACrB,aAAa;GACd,CAAC;;CAGJ,AAAQ,gBAAwC;EAC9C,MAAM,UAAU,QAAQ,IAAI;AAC5B,SAAO,UAAU,EAAE,SAAS,SAAS,GAAG,EAAE;;CAG5C,AAAQ,eAAe,OAA8B;AACnD,SAAO,KAAK,OAAO,SAAS,UAAU;;CAGxC,aAAa,QAAoB;AAC/B,OAAK,MAAM,QAAQ;GACjB,MAAM;GACN,QAAQ;GACR,MAAM;GACN,SAAS,OAAO,KAAsB,QAA0B;AAC9D,UAAM,KAAK,OAAO,IAAI,CAAC,mBAAmB,KAAK,IAAI;;GAEtD,CAAC;AAEF,OAAK,MAAM,QAAQ;GACjB,MAAM;GACN,QAAQ;GACR,MAAM;GACN,SAAS,OAAO,KAAsB,QAA0B;AAC9D,UAAM,KAAK,OAAO,IAAI,CAAC,uBAAuB,KAAK,IAAI;;GAE1D,CAAC;;CAGJ,MAAM,mBACJ,KACA,KACe;EACf,MAAM,EAAE,UAAU,IAAI;EACtB,MAAM,UAAU,KAAK,eAAe,MAAM;AAE1C,MAAI,CAAC,SAAS;AACZ,OAAI,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,wBAAwB,SAAS,CAAC;AAChE;;EAGF,MAAM,EAAE,SAAS,mBAAmB,IAAI;AAExC,MAAI,CAAC,SAAS;AACZ,OAAI,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,uBAAuB,CAAC;AACtD;;AAGF,SAAO,MACL,6DACA,SACA,OACA,kBAAkB,MACnB;EAED,MAAM,UAAU,KAAK,OAAO,WAAW;EACvC,MAAM,YAAa,IAAI,MAAM,aAAwB,YAAY;EAEjE,MAAM,iBAA0C;GAC9C,GAAG;GACH,SAAS;IACP,GAAG,oBAAoB;IACvB;IACD;GACD,QAAQ;IACN,GAAG,oBAAoB;IACvB,UAAU;IACX;GACF;EAED,MAAM,kBAAkB,oBAAoB;AAE5C,QAAM,KAAK,cACT,WAEE,KAAK,eAAe,kBAClB,iBACA,SACA,SACA,gBACA,EAAE,SAAS,CACZ,EACH,eACD;;CAGH,MAAM,uBACJ,KACA,KACe;EACf,MAAM,EAAE,OAAO,mBAAmB,IAAI;EACtC,MAAM,UAAU,KAAK,eAAe,MAAM;AAE1C,MAAI,CAAC,SAAS;AACZ,OAAI,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,wBAAwB,SAAS,CAAC;AAChE;;EAGF,MAAM,sBAAsB,IAAI,MAAM,wBAAwB;EAC9D,MAAM,YAAa,IAAI,MAAM,aAAwB,YAAY;AAEjE,SAAO,MACL,6EACA,gBACA,SACA,OACA,oBACD;EAED,MAAM,iBAA0C;GAC9C,GAAG;GACH,QAAQ;IACN,GAAG,oBAAoB;IACvB,UAAU;IACX;GACF;EAED,MAAM,kBAAkB,oBAAoB;AAE5C,QAAM,KAAK,cACT,WAEE,KAAK,eAAe,mBAClB,iBACA,SACA,gBACA,EAAE,qBAAqB,CACxB,EACH,eACD;;CAGH,MAAM,gBACJ,OACA,gBAC2C;EAC3C,MAAM,UAAU,KAAK,eAAe,MAAM;AAE1C,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,wBAAwB,QAAQ;EAGlD,MAAM,kBAAkB,oBAAoB;AAE5C,SAAO,KAAK,eAAe,gBACzB,iBACA,SACA,eACD;;;;;;CAOH,OAAO,YACL,OACA,SACA,gBACA,SACkC;EAClC,MAAM,UAAU,KAAK,eAAe,MAAM;AAC1C,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,wBAAwB,QAAQ;EAElD,MAAM,kBAAkB,oBAAoB;EAC5C,MAAM,UAAU,SAAS,WAAW,KAAK,OAAO,WAAW;AAC3D,SAAO,KAAK,eAAe,kBACzB,iBACA,SACA,SACA,gBACA,EAAE,SAAS,CACZ;;CAGH,MAAM,WAA0B;AAC9B,OAAK,cAAc,UAAU;;CAG/B,UAAU;AACR,SAAO;GACL,aAAa,KAAK;GAClB,iBAAiB,KAAK;GACvB;;;;;;AAOL,MAAa,QAAQ,SACnB,aACA,QACD"}