@dbx-tools/genie 0.1.41 → 0.1.48

package/dist/src/chat.d.ts

@@ -7,10 +7,11 @@
  * the next turn's `options.conversationId`).
  *
  * Two layers serve two kinds of consumer. The low-level layer
- * yields every poll-observed `GenieMessage` verbatim and owns the
- * messy parts - cancellation, conversation seeding,
- * distinct-filtering, and SDK quirks (Waiter stripping); reach for
- * it when you want the raw stream. The high-level layer wraps it
+ * yields every poll-observed `GenieMessage` (validated against
+ * `GenieMessageSchema`, falling back to the raw snapshot on a schema
+ * miss) and owns the messy parts - cancellation, conversation
+ * seeding, distinct-filtering, and SDK quirks (Waiter stripping);
+ * reach for it when you want the raw stream. The high-level layer wraps it
  * and emits semantic, deduplicated `{ type, payload }` events
  * (see {@link GenieChatEvent}), always closing a successful turn
  * with a terminal `result` event carrying the final

package/dist/src/chat.js

@@ -7,10 +7,11 @@
  * the next turn's `options.conversationId`).
  *
  * Two layers serve two kinds of consumer. The low-level layer
- * yields every poll-observed `GenieMessage` verbatim and owns the
- * messy parts - cancellation, conversation seeding,
- * distinct-filtering, and SDK quirks (Waiter stripping); reach for
- * it when you want the raw stream. The high-level layer wraps it
+ * yields every poll-observed `GenieMessage` (validated against
+ * `GenieMessageSchema`, falling back to the raw snapshot on a schema
+ * miss) and owns the messy parts - cancellation, conversation
+ * seeding, distinct-filtering, and SDK quirks (Waiter stripping);
+ * reach for it when you want the raw stream. The high-level layer wraps it
  * and emits semantic, deduplicated `{ type, payload }` events
  * (see {@link GenieChatEvent}), always closing a successful turn
  * with a terminal `result` event carrying the final
@@ -21,8 +22,27 @@
  * event layer.
  */
 import { WorkspaceClient } from "@databricks/sdk-experimental";
-import { eventsFromMessage, isTerminalStatus, } from "@dbx-tools/genie-shared";
-import { apiUtils, commonUtils } from "@dbx-tools/shared";
+import { eventsFromMessage, GenieMessageSchema, isTerminalStatus, } from "@dbx-tools/genie-shared";
+import { apiUtils, commonUtils, logUtils } from "@dbx-tools/shared";
+const log = logUtils.logger("genie/chat");
+/**
+ * Validate a polled wire snapshot against {@link GenieMessageSchema}
+ * and return the schema-normalized message. Genie's wire occasionally
+ * ships a shape the (SDK-derived) schema doesn't model exactly - e.g.
+ * an early poll that omits the SDK-required `message_id` - so a miss
+ * degrades to the raw snapshot rather than throwing, keeping a single
+ * odd poll from aborting the whole turn.
+ */
+function validateMessage(raw) {
+    const result = GenieMessageSchema.safeParse(raw);
+    if (result.success)
+        return result.data;
+    log.debug("wire-message:schema-miss", {
+        message_id: raw.message_id ?? raw.id,
+        issues: result.error.issues.length,
+    });
+    return raw;
+}
 /* ----------------------- low-level: genieChat ----------------------- */
 /**
  * One turn against a Genie space, yielded as a stream of
@@ -108,7 +128,7 @@ export async function* genieChat(space_id, content, options) {
                 message_id: messageId,
             }, context);
         };
-        yield* commonUtils.poll(pollProducer, {
+        yield* commonUtils.poll(async (ctx) => validateMessage(await pollProducer(ctx)), {
             intervalMs: options?.pollIntervalMs ?? 500,
             // Skip yielding identical consecutive snapshots; Genie
             // often returns the exact same payload twice during quiet