feat(genie): use client-provided requestId query param as SSE streamId

Replace conversation-derived stream IDs with a client-provided
?requestId=<uuid> query param, enabling reliable SSE reconnection.
Falls back to crypto.randomUUID() when not provided.

Signed-off-by: Jorge Calvar <jorge.calvar@databricks.com>

Author: calvarjorge

packages/appkit/src/plugins/genie/genie.ts

@@ -1,3 +1,4 @@
+import { randomUUID } from "node:crypto";
 import { Time, TimeUnits } from "@databricks/sdk-experimental";
 import type {
   GenieMessage,
@@ -134,6 +135,7 @@ export class GeniePlugin extends Plugin {
     );
 
     const timeout = this.config.timeout ?? 120_000;
+    const requestId = (req.query.requestId as string) || randomUUID();
 
     const streamSettings: StreamExecutionSettings = {
       ...genieStreamDefaults,
@@ -144,9 +146,7 @@ export class GeniePlugin extends Plugin {
       },
       stream: {
         ...genieStreamDefaults.stream,
-        streamId: conversationId
-          ? `genie:send:${spaceId}:${conversationId}`
-          : `genie:send:${spaceId}:new-${Date.now()}`,
+        streamId: requestId,
       },
     };
 
@@ -294,6 +294,7 @@ export class GeniePlugin extends Plugin {
     }
 
     const includeQueryResults = req.query.includeQueryResults !== "false";
+    const requestId = (req.query.requestId as string) || randomUUID();
 
     logger.debug(
       "Fetching conversation %s from space %s (alias=%s, includeQueryResults=%s)",
@@ -309,7 +310,7 @@ export class GeniePlugin extends Plugin {
       ...genieStreamDefaults,
       stream: {
         ...genieStreamDefaults.stream,
-        streamId: `genie:conversation:${spaceId}:${conversationId}`,
+        streamId: requestId,
       },
     };
 

packages/appkit/src/plugins/genie/tests/genie.test.ts

@@ -994,7 +994,7 @@ describe("Genie Plugin", () => {
       executeStreamSpy.mockRestore();
     });
 
-    test("sendMessage with conversationId should use deterministic streamId", async () => {
+    test("sendMessage should use requestId query param as streamId", async () => {
       const plugin = new GeniePlugin(config);
       const { router, getHandler } = createMockRouter();
 
@@ -1003,6 +1003,7 @@ describe("Genie Plugin", () => {
       const handler = getHandler("POST", "/:alias/messages");
       const mockReq = createMockRequest({
         params: { alias: "myspace" },
+        query: { requestId: "req-uuid-123" },
         body: {
           content: "follow-up question",
           conversationId: "conv-42",
@@ -1021,14 +1022,14 @@ describe("Genie Plugin", () => {
         expect.any(Function),
         expect.objectContaining({
           stream: expect.objectContaining({
-            streamId: "genie:send:test-space-id:conv-42",
+            streamId: "req-uuid-123",
             bufferSize: 100,
           }),
         }),
       );
     });
 
-    test("sendMessage without conversationId should use timestamped streamId", async () => {
+    test("sendMessage without requestId should generate a random streamId", async () => {
       const plugin = new GeniePlugin(config);
       const { router, getHandler } = createMockRouter();
 
@@ -1053,15 +1054,15 @@ describe("Genie Plugin", () => {
         expect.objectContaining({
           stream: expect.objectContaining({
             streamId: expect.stringMatching(
-              /^genie:send:test-space-id:new-\d+$/,
+              /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/,
             ),
             bufferSize: 100,
           }),
         }),
       );
     });
 
-    test("getConversation should use deterministic streamId", async () => {
+    test("getConversation should use requestId query param as streamId", async () => {
       const plugin = new GeniePlugin(config);
       const { router, getHandler } = createMockRouter();
 
@@ -1073,7 +1074,7 @@ describe("Genie Plugin", () => {
       );
       const mockReq = createMockRequest({
         params: { alias: "myspace", conversationId: "conv-99" },
-        query: {},
+        query: { requestId: "req-uuid-456" },
         headers: {
           "x-forwarded-access-token": "user-token",
           "x-forwarded-user": "user-1",
@@ -1088,7 +1089,7 @@ describe("Genie Plugin", () => {
         expect.any(Function),
         expect.objectContaining({
           stream: expect.objectContaining({
-            streamId: "genie:conversation:test-space-id:conv-99",
+            streamId: "req-uuid-456",
             bufferSize: 100,
           }),
         }),