@neta-art/cohub 1.33.0 → 1.33.1

package/dist/chunks/http.js

@@ -1,3 +1,4 @@
+import { r as getRealtimeSpaceRoom } from "./types.js";
 import { a as resolveApiBaseUrl } from "./environment.js";
 //#region src/apis/channels.ts
 var ChannelsApi = class {
@@ -928,6 +929,7 @@ var SessionGenerationStreamClient = class SessionGenerationStreamClient {
 		let recoveryAborted = false;
 		let bufferedEvents = [];
 		let bufferedEventsReplayed = false;
+		const releaseRoom = this.websocketClient.retainRooms([getRealtimeSpaceRoom(this.spaceId)]);
 		const unsubscribe = this.websocketClient.on("event", (event) => {
 			if (event.spaceId !== this.spaceId || event.sessionId !== this.sessionId) return;
 			if (recovering && isGenerationRealtimeEvent(event)) {
@@ -952,6 +954,7 @@ var SessionGenerationStreamClient = class SessionGenerationStreamClient {
 		return () => {
 			disposed = true;
 			unsubscribe();
+			releaseRoom();
 		};
 	}
 	async recoverFromSnapshot(handlers, options, shouldContinue) {
@@ -1529,6 +1532,7 @@ var SessionRealtimeClient = class {
 	subscribe(handlers) {
 		if (!this.websocketClient) throw new Error("realtime transport is not configured for this client");
 		ensureRealtimeConnected(this.websocketClient);
+		const releaseRoom = this.websocketClient.retainRooms([getRealtimeSpaceRoom(this.spaceId)]);
 		const unsubscribe = this.websocketClient.on("event", (event) => {
 			if (event.spaceId !== this.spaceId || event.sessionId !== this.sessionId) return;
 			handlers.event?.(event);
@@ -1577,7 +1581,10 @@ var SessionRealtimeClient = class {
 				turnId: getPersistedMessageTurnId(event)
 			});
 		});
-		return () => unsubscribe();
+		return () => {
+			unsubscribe();
+			releaseRoom();
+		};
 	}
 	on(type, handler) {
 		return this.subscribe({ event: (event) => {
@@ -1691,10 +1698,15 @@ var SpaceEventsApi = class {
 	subscribe(handler) {
 		if (!this.websocketClient) throw new Error("realtime transport is not configured for this client");
 		ensureRealtimeConnected(this.websocketClient);
-		return this.websocketClient.on("event", (event) => {
+		const releaseRoom = this.websocketClient.retainRooms([getRealtimeSpaceRoom(this.spaceId)]);
+		const offEvent = this.websocketClient.on("event", (event) => {
 			if (event.spaceId !== this.spaceId) return;
 			handler(event);
 		});
+		return () => {
+			offEvent();
+			releaseRoom();
+		};
 	}
 	on(type, handler) {
 		return this.subscribe((event) => {

package/dist/chunks/types.js

@@ -0,0 +1,37 @@
+//#region ../protocol/src/realtime/types.ts
+const WS_COMPACT_STREAM_CAPABILITY = "session.compact_stream.v1";
+const WS_ROOM_SUBSCRIPTION_CAPABILITY = "realtime.rooms.v1";
+const getRealtimeSpaceRoom = (spaceId) => `space:${spaceId}`;
+const parseRealtimeRoom = (room) => {
+	const trimmed = room.trim();
+	const separatorIndex = trimmed.indexOf(":");
+	if (separatorIndex <= 0) return null;
+	const kind = trimmed.slice(0, separatorIndex);
+	const id = trimmed.slice(separatorIndex + 1).trim();
+	if (!id) return null;
+	if (kind !== "space" && kind !== "user") return null;
+	return {
+		kind,
+		id
+	};
+};
+const normalizeRealtimeRooms = (rooms) => {
+	const normalized = /* @__PURE__ */ new Set();
+	for (const room of rooms) {
+		const parsed = typeof room === "string" ? parseRealtimeRoom(room) : null;
+		if (!parsed) continue;
+		normalized.add(`${parsed.kind}:${parsed.id}`);
+	}
+	return [...normalized];
+};
+const getNonEmptyString = (value) => typeof value === "string" && value.trim() ? value : null;
+const getSessionTurnPatchStreamKey = (input, options = {}) => {
+	const turnId = getNonEmptyString(input.turnId);
+	const messageKey = getNonEmptyString(input.messageId) ?? getNonEmptyString(input.sourceMessageId) ?? getNonEmptyString(input.anchorUserMessageId) ?? (typeof input.messageOrdinal === "number" && Number.isFinite(input.messageOrdinal) ? `ordinal:${input.messageOrdinal}` : null);
+	if (turnId && messageKey) return `${turnId}:${messageKey}`;
+	const streamKey = messageKey ?? turnId;
+	if (streamKey) return streamKey;
+	return options.includeSessionFallback ? getNonEmptyString(input.sessionId) : null;
+};
+//#endregion
+export { normalizeRealtimeRooms as a, getSessionTurnPatchStreamKey as i, WS_ROOM_SUBSCRIPTION_CAPABILITY as n, getRealtimeSpaceRoom as r, WS_COMPACT_STREAM_CAPABILITY as t };

package/dist/chunks/websocket.d.ts

@@ -275,6 +275,7 @@ type SpacePublicEndpoint = {
 type SpacePublicEndpoints = Record<string, SpacePublicEndpoint>;
 //#endregion
 //#region ../protocol/src/realtime/types.d.ts
+type RealtimeRoom = `space:${string}` | `user:${string}`;
 type RealtimeEnvelope = {
   id: string;
   timestamp: number;
@@ -283,6 +284,7 @@ type RealtimeEnvelope = {
   requestId?: string | null;
   spaceId?: string | null;
   sessionId?: string | null;
+  rooms?: RealtimeRoom[];
   payload: Record<string, unknown>;
 };
 type ChannelEnvelope = RealtimeEnvelope;
@@ -346,6 +348,34 @@ type SystemAckOkEvent = {
     eventId?: string;
   };
 };
+type SystemSubscribeOkEvent = {
+  id: string;
+  timestamp: number;
+  domain: "system";
+  type: "system.subscribe.ok";
+  requestId?: string | null;
+  spaceId?: string | null;
+  sessionId?: string | null;
+  payload: {
+    rooms: RealtimeRoom[];
+  };
+};
+type SystemSubscribeErrorEvent = {
+  id: string;
+  timestamp: number;
+  domain: "system";
+  type: "system.subscribe.error";
+  requestId?: string | null;
+  spaceId?: string | null;
+  sessionId?: string | null;
+  payload: {
+    rejected: Array<{
+      room: string;
+      code: "BAD_ROOM" | "FORBIDDEN";
+      message: string;
+    }>;
+  };
+};
 type RealtimeSessionRecord = Pick<SessionRecord, "id" | "spaceId" | "userUuid" | "title" | "source" | "status" | "externalSessionId" | "latestMessageText" | "lastMessageAt" | "lastMessageId" | "createdAt" | "updatedAt">;
 type SessionCreatedEvent = {
   id: string;
@@ -651,7 +681,7 @@ type LabelAssignmentsUpdatedEvent = {
     affectedLabelIds: string[];
   };
 };
-type RealtimeServerEvent = SystemReadyEvent | SystemAuthOkEvent | SystemRequestErrorEvent | SystemPongEvent | SystemAckOkEvent | SessionCreatedEvent | SessionUpdatedEvent | SessionRequestAcceptedEvent | SessionRequestErrorEvent | SessionTurnCreatedEvent | SessionTurnPatchEvent | SessionTurnErrorEvent | SessionTurnLifecycleEvent | SessionTurnUpdatedEvent | SessionTurnFinalizedEvent | SessionMessagePersistedEvent | SpaceFsChangedEvent | SpacePortsChangedEvent | CanvasTransactionAppliedEvent | CanvasTransactionAckEvent | CanvasTransactionErrorEvent | TaskCreatedEvent | TaskUpdatedEvent | LabelAssignmentsUpdatedEvent;
+type RealtimeServerEvent = SystemReadyEvent | SystemAuthOkEvent | SystemRequestErrorEvent | SystemPongEvent | SystemAckOkEvent | SystemSubscribeOkEvent | SystemSubscribeErrorEvent | SessionCreatedEvent | SessionUpdatedEvent | SessionRequestAcceptedEvent | SessionRequestErrorEvent | SessionTurnCreatedEvent | SessionTurnPatchEvent | SessionTurnErrorEvent | SessionTurnLifecycleEvent | SessionTurnUpdatedEvent | SessionTurnFinalizedEvent | SessionMessagePersistedEvent | SpaceFsChangedEvent | SpacePortsChangedEvent | CanvasTransactionAppliedEvent | CanvasTransactionAckEvent | CanvasTransactionErrorEvent | TaskCreatedEvent | TaskUpdatedEvent | LabelAssignmentsUpdatedEvent;
 //#endregion
 //#region src/websocket.d.ts
 type WebsocketEventPayload = ChannelEnvelope;
@@ -726,6 +756,18 @@ type WebsocketClientEvents = {
     spaceId?: string | null;
     clientMessageId?: string | null;
   };
+  subscribed: {
+    rooms: RealtimeRoom[];
+    requestId?: string | null;
+  };
+  subscribeError: {
+    rejected: Array<{
+      room: string;
+      code: string;
+      message: string;
+    }>;
+    requestId?: string | null;
+  };
   pong: {
     requestId?: string | null;
   };
@@ -754,6 +796,7 @@ declare class WebsocketClient {
   private pongDeadlineAt;
   private readonly compactStreamContexts;
   private readonly patchStreamBuffers;
+  private readonly roomSubscriptions;
   state: WebsocketClientState;
   connectionId: string | null;
   private readonly listeners;
@@ -783,11 +826,16 @@ declare class WebsocketClient {
     model?: string;
     provider?: string;
   }): Promise<void>;
+  retainRooms(rooms: readonly string[]): () => void;
+  subscribeRooms(rooms: readonly string[]): () => void;
+  subscribeSpace(spaceId: string): () => void;
   ack(eventId?: string, requestId?: string): void;
   ping(requestId?: string): void;
   private ensureOpen;
   private send;
   private authenticate;
+  private flushRoomSubscriptions;
+  private restoreRoomSubscriptions;
   private createAuthWaiter;
   private resolveAuthWaiter;
   private rejectAuthWaiter;

package/dist/index.js

@@ -1,6 +1,6 @@
 import { C as ChannelsApi, S as CronJobsApi, _ as SearchApi, a as TasksApi, b as ModelsApi, c as PublicInviteApi, d as parseAssistantMessageCommit, f as SessionPatchReducer, g as SessionAccessApi, h as HttpTransport, i as UserApi, l as SessionGenerationStreamClient, m as HttpError, n as createHttpClient, o as SpaceClient, p as createSessionPatchReducer, r as WorksApi, s as SpacesApi, t as CohubHttpClient, u as createSessionGenerationStreamClient, v as PublicAssetsApi, x as GenerationsApi, y as PromptsApi } from "./chunks/http.js";
 import { a as resolveApiBaseUrl, c as resolveWebsocketUrl, i as normalizeWebsocketUrl, n as normalizeBaseUrl, o as resolveCohubEnvironment, r as normalizeVoiceInputWebsocketUrl, s as resolveVoiceInputWebsocketUrl, t as COHUB_ENVIRONMENTS } from "./chunks/environment.js";
-import { n as createWebsocketClient, t as WebsocketClient } from "./chunks/websocket.js";
+import { WebsocketClient, createWebsocketClient } from "./websocket.js";
 import { VoiceApi, VoiceInputClient, createVoiceInputClient } from "./voice-input.js";
 //#region src/apis/billing.ts
 var BillingApi = class {