@sogni-ai/sogni-client 5.65.0 → 5.66.0

package/src/ApiClient/WebSocketClient/BrowserWebSocketClient/ChannelCoordinator.ts

@@ -4,6 +4,12 @@ import { MessageDeliveryUncertainError, SEND_READY_TIMEOUT_MS } from '../request
 
 const PRIMARY_HEARTBEAT_INTERVAL = 2000;
 const PRIMARY_TIMEOUT = 4000;
+// How long a visible tab waits without a heartbeat from the primary before it
+// takes the connection over itself. A primary in a background tab can keep
+// answering elections (messages are not throttled) while its timers, and so its
+// heartbeats, have all but stopped; requests forwarded to it can then wait on a
+// connection that no longer works.
+const PRIMARY_TAKEOVER_TIMEOUT = 12000;
 const ELECTION_SETTLE_TIMEOUT = 600;
 const ACK_TIMEOUT = 5000;
 
@@ -126,6 +132,11 @@ class ChannelCoordinator<M, N> {
 
   // Heartbeat to detect primary tab death
   private lastPrimaryHeartbeat: number = Date.now();
+  // The primary this tab follows, and when it last sent a heartbeat of its own.
+  // Unlike lastPrimaryHeartbeat, the primary re-announcing itself in answer to
+  // an election does not count.
+  private primaryId: string | null = null;
+  private lastHeartbeatFromPrimary: number = Date.now();
   private heartbeatTimer: NodeJS.Timeout | null = null;
   private primaryCheckTimer: NodeJS.Timeout | null = null;
   private readyCallback: () => void | null = () => {};
@@ -160,6 +171,10 @@ class ChannelCoordinator<M, N> {
     return this._isPrimary;
   }
 
+  private tabIsVisible() {
+    return isActiveTab;
+  }
+
   private get currentPriority() {
     return isActiveTab ? this.priority * 10 : this.priority;
   }
@@ -209,8 +224,24 @@ class ChannelCoordinator<M, N> {
     return new Promise((resolve) => setTimeout(resolve, ELECTION_SETTLE_TIMEOUT));
   }
 
+  /**
+   * A visible secondary whose primary has sent no heartbeat for
+   * PRIMARY_TAKEOVER_TIMEOUT becomes primary itself. Announcing demotes the
+   * silent primary, which closes its connection; this tab opens its own.
+   */
+  private takeOverFromSilentPrimary(): boolean {
+    if (this._isPrimary || !this.tabIsVisible()) return false;
+    const silentFor = Date.now() - this.lastHeartbeatFromPrimary;
+    if (silentFor <= PRIMARY_TAKEOVER_TIMEOUT) return false;
+    this.logger.debug(`Primary sent no heartbeat for ${silentFor}ms, taking over as primary`);
+    this.electionInProgress = false;
+    this.electionResponses.clear();
+    this.becomePrimary();
+    return true;
+  }
+
   private async ensureFreshPrimaryBeforeRequest(): Promise<void> {
-    if (this._isPrimary) {
+    if (this._isPrimary || this.takeOverFromSilentPrimary()) {
       return;
     }
 
@@ -260,6 +291,7 @@ class ChannelCoordinator<M, N> {
 
   private becomePrimary() {
     this._isPrimary = true;
+    this.primaryId = this.id;
     this.lastPrimaryHeartbeat = Date.now();
     this.broadcast({
       type: MessageType.PRIMARY_ANNOUNCE,
@@ -300,6 +332,7 @@ class ChannelCoordinator<M, N> {
     }
     this.primaryCheckTimer = setInterval(() => {
       if (!this._isPrimary) {
+        if (this.takeOverFromSilentPrimary()) return;
         const timeSinceLastHeartbeat = Date.now() - this.lastPrimaryHeartbeat;
         if (timeSinceLastHeartbeat > PRIMARY_TIMEOUT) {
           this.startElections();
@@ -325,7 +358,7 @@ class ChannelCoordinator<M, N> {
       case MessageType.PRIMARY_ANNOUNCE:
         return this.handlePrimaryAnnounce(message);
       case MessageType.PRIMARY_HEARTBEAT:
-        return this.handlePrimaryHeartbeat();
+        return this.handlePrimaryHeartbeat(message);
       case MessageType.BROADCAST:
         return this.handleBroadcast(message);
       case MessageType.REQUEST:
@@ -373,6 +406,12 @@ class ChannelCoordinator<M, N> {
     const wasPrimary = this._isPrimary;
     this._isPrimary = false;
     this.lastPrimaryHeartbeat = Date.now();
+    // A new primary starts with a clean record; the same primary answering an
+    // election again proves only that its tab still handles messages.
+    if (message.payload.id !== this.primaryId) {
+      this.primaryId = message.payload.id;
+      this.lastHeartbeatFromPrimary = Date.now();
+    }
     this.electionInProgress = false;
     this.electionResponses.clear();
 
@@ -386,8 +425,10 @@ class ChannelCoordinator<M, N> {
     this.readyCallback();
   }
 
-  private handlePrimaryHeartbeat() {
+  private handlePrimaryHeartbeat(message: PrimaryHeartbeat) {
     this.lastPrimaryHeartbeat = Date.now();
+    this.primaryId = message.payload.id;
+    this.lastHeartbeatFromPrimary = Date.now();
   }
 
   private handleBroadcast(message: BroadcastMessage) {

package/src/Chat/index.ts

@@ -1220,11 +1220,14 @@ class ChatApi extends ApiGroup<ChatApiEvents> {
   }
 
   private handleSocketAuthenticated(data: AuthenticatedData): void {
-    if (!this.jobsAwaitingReconnect.size || !Array.isArray(data?.activeLLMJobIDs)) return;
+    if (!Array.isArray(data?.activeLLMJobIDs)) return;
     const live = new Set(data.activeLLMJobIDs.map((id) => String(id).toUpperCase()));
+    // Every job already sent, not only those a 'connecting' put on watch: a tab
+    // that takes the connection over from another tab gets no 'connecting', and
+    // a job queued on the old connection was cancelled when it closed.
     const gone: string[] = [];
-    for (const jobID of this.jobsAwaitingReconnect) {
-      if (!live.has(jobID.toUpperCase())) gone.push(jobID);
+    for (const jobID of this.activeStreams.keys()) {
+      if (!this.unsentJobs.has(jobID) && !live.has(jobID.toUpperCase())) gone.push(jobID);
     }
     this.jobsAwaitingReconnect.clear();
     this.clearTransportGraceTimer();

package/src/Projects/createJobRequestMessage.ts

@@ -1372,15 +1372,32 @@ function applyImageParams(
   params: ImageProjectParams,
   options: ImageModelOptions
 ) {
-  if (['moss_image_sft_int8_convrot', 'moss_image_sft_dmd8_int8_convrot'].includes(params.modelId) &&
-      (params.startingImage || params.contextImages?.length)) {
-    throw new Error('MOSS-Image SFT generates from text and does not accept guide or reference images.');
+  if (
+    ['moss_image_sft_int8_convrot', 'moss_image_sft_dmd8_int8_convrot'].includes(params.modelId) &&
+    (params.startingImage || params.contextImages?.length)
+  ) {
+    throw new Error(
+      'MOSS-Image SFT generates from text and does not accept guide or reference images.'
+    );
   }
   const keyFrame: Record<string, any> = {
     ...inputKeyframe,
     sizePreset: params.sizePreset
   };
   const contextImages = params.contextImages || [];
+  for (const name of ['mossRopeScaleX', 'mossRopeScaleY'] as const) {
+    const value = params[name];
+    if (value === undefined) continue;
+    if (
+      !['moss_image_sft_int8_convrot', 'moss_image_sft_dmd8_int8_convrot'].includes(params.modelId)
+    ) {
+      throw new Error(`${name} is only supported by MOSS-Image`);
+    }
+    if (typeof value !== 'number' || !Number.isFinite(value) || value < 0 || value > 100) {
+      throw new Error(`${name} must be a finite number between 0 and 100`);
+    }
+    if (value !== 1) keyFrame[name] = value;
+  }
   for (let index = 1; index <= 16; index += 1) {
     keyFrame[`hasContextImage${index}`] = !!contextImages[index - 1];
   }

package/src/Projects/types/index.ts

@@ -931,6 +931,10 @@ export interface Sam3Selection {
 export interface ImageProjectParams
   extends BaseProjectParams, Pixal3dGenerationOptions, Pixal3dMultiViewImages {
   type: 'image';
+  /** MOSS only: horizontal spatial scaling, 0–100; 1 is neutral. Changes composition, not output size. */
+  mossRopeScaleX?: number;
+  /** MOSS only: vertical spatial scaling, 0–100; 1 is neutral. */
+  mossRopeScaleY?: number;
   /**
    * Number of previews to generate. Note that previews affect project cost
    */

package/src/version.ts

@@ -1,3 +1,3 @@
 // AUTO-GENERATED by scripts/generate-version.mjs — do not edit.
 // Source of truth: package.json version.
-export const LIB_VERSION = '5.65.0';
+export const LIB_VERSION = '5.66.0';