@sogni-ai/sogni-client 4.2.0-alpha.2 → 4.2.0-alpha.21

package/dist/ApiClient/index.js

@@ -23,11 +23,12 @@ class ApiError extends Error {
 }
 exports.ApiError = ApiError;
 class ApiClient extends TypedEventEmitter_1.default {
-    constructor({ baseUrl, socketUrl, appId, networkType, authType, logger, disableSocket = false, multiInstance = false }) {
+    constructor({ baseUrl, socketUrl, appId, appSource, socketEventSubscriptions, networkType, authType, logger, disableSocket = false, multiInstance = false }) {
         super();
         this._reconnectAttempts = WS_RECONNECT_ATTEMPTS;
         this._disableSocket = false;
         this.appId = appId;
+        this.appSource = (appSource === null || appSource === void 0 ? void 0 : appSource.trim()) || undefined;
         this.logger = logger;
         if (authType === 'apiKey') {
             this._auth = new ApiKeyAuthManager_1.default(logger);
@@ -42,10 +43,10 @@ class ApiClient extends TypedEventEmitter_1.default {
         const supportMultiInstance = !isNodejs_1.default && this._auth instanceof CookieAuthManager_1.default;
         if (supportMultiInstance && multiInstance) {
             // Use coordinated WebSocket client to share single connection between tabs
-            this._socket = new BrowserWebSocketClient_1.default(socketUrl, this._auth, appId, networkType, logger);
+            this._socket = new BrowserWebSocketClient_1.default(socketUrl, this._auth, appId, networkType, logger, this.appSource, socketEventSubscriptions);
         }
         else {
-            this._socket = new WebSocketClient_1.default(socketUrl, this._auth, appId, networkType, logger);
+            this._socket = new WebSocketClient_1.default(socketUrl, this._auth, appId, networkType, logger, this.appSource, socketEventSubscriptions);
         }
         this._disableSocket = disableSocket;
         this._auth.on('updated', this.handleAuthUpdated.bind(this));
@@ -67,6 +68,9 @@ class ApiClient extends TypedEventEmitter_1.default {
     get socketEnabled() {
         return !this._disableSocket;
     }
+    setSocketEventSubscriptions(update) {
+        return this.socket.setSocketEventSubscriptions(update);
+    }
     handleSocketConnect({ network }) {
         this._reconnectAttempts = WS_RECONNECT_ATTEMPTS;
         this.emit('connected', { network });

package/dist/ApiClient/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ApiClient/index.ts"],"names":[],"mappings":";;;;;;AAAA,mEAA2C;AAC3C,wEAAgD;AAChD,iFAAyD;AAGzD,2DAA0E;AAI1E,6FAAqE;AACrE,6FAAqE;AACrE,oDAAmE;AACnE,+DAAuC;AACvC,sGAA8E;AAE9E,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAchC,MAAa,QAAS,SAAQ,KAAK;IAGjC,YAAY,MAAc,EAAE,OAAyB;QACnD,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF;AARD,4BAQC;AAaD,MAAM,SAAU,SAAQ,2BAAkC;IASxD,YAAY,EACV,OAAO,EACP,SAAS,EACT,KAAK,EACL,WAAW,EACX,QAAQ,EACR,MAAM,EACN,aAAa,GAAG,KAAK,EACrB,aAAa,GAAG,KAAK,EACJ;QACjB,KAAK,EAAE,CAAC;QAbF,uBAAkB,GAAG,qBAAqB,CAAC;QAC3C,mBAAc,GAAY,KAAK,CAAC;QAatC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,2BAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,GAAG,IAAI,8BAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,GAAG,IAAI,2BAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,oBAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACzD,MAAM,oBAAoB,GAAG,CAAC,kBAAQ,IAAI,IAAI,CAAC,KAAK,YAAY,2BAAiB,CAAC;QAClF,IAAI,oBAAoB,IAAI,aAAa,EAAE,CAAC;YAC1C,2EAA2E;YAC3E,IAAI,CAAC,OAAO,GAAG,IAAI,gCAAsB,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;QAC/F,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAClE,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,cAAc,EAAE,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC;IACnC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;IAC9B,CAAC;IAED,mBAAmB,CAAC,EAAE,OAAO,EAAqB;QAChD,IAAI,CAAC,kBAAkB,GAAG,qBAAqB,CAAC;QAChD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IACtC,CAAC;IAED,sBAAsB,CAAC,IAA0B;QAC/C,2DAA2D;QAC3D,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAChC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAA,4BAAgB,EAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,qGAAqG;YACrG,IACE,IAAI,CAAC,OAAO,YAAY,gCAAsB;gBAC9C,IAAI,CAAC,IAAI,KAAK,qBAAS,CAAC,iBAAiB,EACzC,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;gBACpE,OAAO;YACT,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,IAAI,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QACD,IAAI,IAAI,CAAC,kBAAkB,IAAI,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,kBAAkB,GAAG,qBAAqB,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;IAChD,CAAC;IAED,iBAAiB,CAAC,eAAwB;QACxC,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5D,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC1B,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAClC,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;QAChC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED,kBAAe,SAAS,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ApiClient/index.ts"],"names":[],"mappings":";;;;;;AAAA,mEAA2C;AAC3C,wEAAgD;AAChD,iFAAyD;AAGzD,2DAA0E;AAQ1E,6FAAqE;AACrE,6FAAqE;AACrE,oDAAmE;AACnE,+DAAuC;AACvC,sGAA8E;AAE9E,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAchC,MAAa,QAAS,SAAQ,KAAK;IAGjC,YAAY,MAAc,EAAE,OAAyB;QACnD,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF;AARD,4BAQC;AAeD,MAAM,SAAU,SAAQ,2BAAkC;IAUxD,YAAY,EACV,OAAO,EACP,SAAS,EACT,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,WAAW,EACX,QAAQ,EACR,MAAM,EACN,aAAa,GAAG,KAAK,EACrB,aAAa,GAAG,KAAK,EACJ;QACjB,KAAK,EAAE,CAAC;QAfF,uBAAkB,GAAG,qBAAqB,CAAC;QAC3C,mBAAc,GAAY,KAAK,CAAC;QAetC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,IAAI,EAAE,KAAI,SAAS,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,2BAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,GAAG,IAAI,8BAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,GAAG,IAAI,2BAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,oBAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACzD,MAAM,oBAAoB,GAAG,CAAC,kBAAQ,IAAI,IAAI,CAAC,KAAK,YAAY,2BAAiB,CAAC;QAClF,IAAI,oBAAoB,IAAI,aAAa,EAAE,CAAC;YAC1C,2EAA2E;YAC3E,IAAI,CAAC,OAAO,GAAG,IAAI,gCAAsB,CACvC,SAAS,EACT,IAAI,CAAC,KAAK,EACV,KAAK,EACL,WAAW,EACX,MAAM,EACN,IAAI,CAAC,SAAS,EACd,wBAAwB,CACzB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,GAAG,IAAI,yBAAe,CAChC,SAAS,EACT,IAAI,CAAC,KAAK,EACV,KAAK,EACL,WAAW,EACX,MAAM,EACN,IAAI,CAAC,SAAS,EACd,wBAAwB,CACzB,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAClE,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,cAAc,EAAE,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC;IACnC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;IAC9B,CAAC;IAED,2BAA2B,CAAC,MAAoC;QAC9D,OAAO,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;IAED,mBAAmB,CAAC,EAAE,OAAO,EAAqB;QAChD,IAAI,CAAC,kBAAkB,GAAG,qBAAqB,CAAC;QAChD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IACtC,CAAC;IAED,sBAAsB,CAAC,IAA0B;QAC/C,2DAA2D;QAC3D,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAChC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAA,4BAAgB,EAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,qGAAqG;YACrG,IACE,IAAI,CAAC,OAAO,YAAY,gCAAsB;gBAC9C,IAAI,CAAC,IAAI,KAAK,qBAAS,CAAC,iBAAiB,EACzC,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;gBACpE,OAAO;YACT,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,IAAI,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QACD,IAAI,IAAI,CAAC,kBAAkB,IAAI,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,kBAAkB,GAAG,qBAAqB,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;IAChD,CAAC;IAED,iBAAiB,CAAC,eAAwB;QACxC,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5D,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC1B,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAClC,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;QAChC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED,kBAAe,SAAS,CAAC"}

package/dist/Chat/ChatTools.d.ts

@@ -1,65 +1,21 @@
 import type ProjectsApi from '../Projects';
 import { ToolCall, ToolExecutionOptions, ToolExecutionProgress, ToolExecutionResult } from './types';
-/**
- * API for executing Sogni platform tool calls (image, video, music generation).
- *
- * Accessed via `sogni.chat.tools`. Provides methods to execute tool calls returned
- * by the LLM, mapping them to `sogni.projects.create()` calls automatically.
- *
- * @example
- * ```typescript
- * // Execute a single tool call
- * const result = await sogni.chat.tools.execute(toolCall, {
- *   tokenType: 'sogni',
- *   onProgress: (p) => console.log(`${p.status}: ${p.percent}%`),
- * });
- *
- * // Execute all tool calls from a completion
- * const results = await sogni.chat.tools.executeAll(result.tool_calls, {
- *   onToolCall: async (tc) => myCustomHandler(tc), // for non-Sogni tools
- * });
- * ```
- */
 declare class ChatToolsApi {
     private projects;
     constructor(projects: ProjectsApi);
-    /**
-     * Execute a single Sogni platform tool call.
-     *
-     * Maps tool call arguments to `sogni.projects.create()`, waits for the media
-     * generation to complete, and returns the result URLs.
-     *
-     * @throws Error if the tool call is not a Sogni tool (use `isSogniToolCall()` to check first)
-     */
     execute(toolCall: ToolCall, options?: ToolExecutionOptions): Promise<ToolExecutionResult>;
-    /**
-     * Execute multiple tool calls from a single LLM response.
-     *
-     * Sogni tool calls (prefixed with `sogni_`) are executed automatically via
-     * `projects.create()`. Non-Sogni tool calls are delegated to the `onToolCall`
-     * callback if provided, or returned as errors.
-     *
-     * @param toolCalls - Array of tool calls from `result.tool_calls`
-     * @param options - Execution options plus optional handler for non-Sogni tools
-     */
     executeAll(toolCalls: ToolCall[], options?: ToolExecutionOptions & {
-        /** Handler for non-Sogni tool calls. Must return the tool result content string. */
         onToolCall?: (toolCall: ToolCall) => Promise<string>;
-        /** Per-tool progress callback (wraps the per-tool onProgress with tool identity). */
         onToolProgress?: (toolCall: ToolCall, progress: ToolExecutionProgress) => void;
     }): Promise<ToolExecutionResult[]>;
-    /**
-     * Get the default model for a media type. For video, prefers LTX-2.3 t2v models.
-     * Falls back to the model with the most available workers.
-     */
-    private getDefaultModel;
-    /**
-     * Create a project, wait for completion with timeout, track per-job progress,
-     * and clean up on failure or timeout.
-     */
+    private getAvailableModels;
+    private selectModel;
     private executeProject;
     private executeImageGeneration;
+    private executeImageEdit;
     private executeVideoGeneration;
+    private executeSoundToVideo;
+    private executeVideoToVideo;
     private executeMusicGeneration;
     private makeErrorResult;
 }

package/dist/Chat/ChatTools.js

@@ -9,41 +9,47 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+const validation_1 = require("../lib/validation");
+const mediaValidation_1 = require("../lib/mediaValidation");
+const modelRouting_1 = require("./modelRouting");
 const tools_1 = require("./tools");
-/** Default timeout for media generation: 10 minutes. */
-const DEFAULT_TIMEOUT = 10 * 60 * 1000;
-/**
- * API for executing Sogni platform tool calls (image, video, music generation).
- *
- * Accessed via `sogni.chat.tools`. Provides methods to execute tool calls returned
- * by the LLM, mapping them to `sogni.projects.create()` calls automatically.
- *
- * @example
- * ```typescript
- * // Execute a single tool call
- * const result = await sogni.chat.tools.execute(toolCall, {
- *   tokenType: 'sogni',
- *   onProgress: (p) => console.log(`${p.status}: ${p.percent}%`),
- * });
- *
- * // Execute all tool calls from a completion
- * const results = await sogni.chat.tools.executeAll(result.tool_calls, {
- *   onToolCall: async (tc) => myCustomHandler(tc), // for non-Sogni tools
- * });
- * ```
- */
+const DEFAULT_TIMEOUT = 30 * 60 * 1000;
+const MAX_SOGNI_TOOL_CALLS_PER_ROUND = 8;
+const MAX_INPUT_MEDIA_BYTES = {
+    image: 20 * 1024 * 1024,
+    audio: 50 * 1024 * 1024,
+    video: 100 * 1024 * 1024
+};
+function getVariationCount(args, options) {
+    return (0, modelRouting_1.getHostedVariationCount)(args, options === null || options === void 0 ? void 0 : options.numberOfMedia);
+}
+function getStringArg(value) {
+    return typeof value === 'string' && value.trim().length > 0 ? value.trim() : undefined;
+}
+function normalizeImageOutputFormat(value) {
+    var _a;
+    const outputFormat = (_a = getStringArg(value)) === null || _a === void 0 ? void 0 : _a.toLowerCase();
+    if (!outputFormat)
+        return undefined;
+    if (outputFormat === 'jpeg')
+        return 'jpg';
+    return outputFormat === 'png' || outputFormat === 'jpg' || outputFormat === 'webp'
+        ? outputFormat
+        : undefined;
+}
+function applyHostedImageOptions(projectParams, args) {
+    var _a, _b;
+    const gptImageQuality = getStringArg((_a = args.gpt_image_quality) !== null && _a !== void 0 ? _a : args.gptImageQuality);
+    if (gptImageQuality)
+        projectParams.gptImageQuality = gptImageQuality.toLowerCase();
+    const outputFormat = normalizeImageOutputFormat((_b = args.output_format) !== null && _b !== void 0 ? _b : args.outputFormat);
+    if (outputFormat)
+        projectParams.outputFormat = outputFormat;
+}
 class ChatToolsApi {
     constructor(projects) {
         this.projects = projects;
     }
-    /**
-     * Execute a single Sogni platform tool call.
-     *
-     * Maps tool call arguments to `sogni.projects.create()`, waits for the media
-     * generation to complete, and returns the result URLs.
-     *
-     * @throws Error if the tool call is not a Sogni tool (use `isSogniToolCall()` to check first)
-     */
     execute(toolCall, options) {
         return __awaiter(this, void 0, void 0, function* () {
             if (!this.projects) {
@@ -55,11 +61,18 @@ class ChatToolsApi {
             const args = (0, tools_1.parseToolCallArguments)(toolCall);
             const name = toolCall.function.name;
             try {
+                (0, modelRouting_1.assertHostedToolArguments)(tools_1.SogniTools.all, name, args);
                 switch (name) {
                     case 'sogni_generate_image':
                         return yield this.executeImageGeneration(toolCall, args, options);
+                    case 'sogni_edit_image':
+                        return yield this.executeImageEdit(toolCall, args, options);
                     case 'sogni_generate_video':
                         return yield this.executeVideoGeneration(toolCall, args, options);
+                    case 'sogni_sound_to_video':
+                        return yield this.executeSoundToVideo(toolCall, args, options);
+                    case 'sogni_video_to_video':
+                        return yield this.executeVideoToVideo(toolCall, args, options);
                     case 'sogni_generate_music':
                         return yield this.executeMusicGeneration(toolCall, args, options);
                     default:
@@ -67,23 +80,17 @@ class ChatToolsApi {
                 }
             }
             catch (err) {
-                const error = err instanceof Error ? err.message : String(err);
+                const error = (0, modelRouting_1.serializeUnknownError)(err);
                 return this.makeErrorResult(toolCall, error);
             }
         });
     }
-    /**
-     * Execute multiple tool calls from a single LLM response.
-     *
-     * Sogni tool calls (prefixed with `sogni_`) are executed automatically via
-     * `projects.create()`. Non-Sogni tool calls are delegated to the `onToolCall`
-     * callback if provided, or returned as errors.
-     *
-     * @param toolCalls - Array of tool calls from `result.tool_calls`
-     * @param options - Execution options plus optional handler for non-Sogni tools
-     */
     executeAll(toolCalls, options) {
         return __awaiter(this, void 0, void 0, function* () {
+            const sogniToolCallCount = toolCalls.filter((toolCall) => (0, tools_1.isSogniToolCall)(toolCall)).length;
+            if (sogniToolCallCount > MAX_SOGNI_TOOL_CALLS_PER_ROUND) {
+                throw new Error(`Too many Sogni tool calls in a single round (${sogniToolCallCount}); maximum is ${MAX_SOGNI_TOOL_CALLS_PER_ROUND}`);
+            }
             const results = [];
             for (const toolCall of toolCalls) {
                 if ((0, tools_1.isSogniToolCall)(toolCall)) {
@@ -110,7 +117,7 @@ class ChatToolsApi {
                         });
                     }
                     catch (err) {
-                        const error = err instanceof Error ? err.message : String(err);
+                        const error = (0, modelRouting_1.serializeUnknownError)(err);
                         results.push(this.makeErrorResult(toolCall, error));
                     }
                 }
@@ -121,34 +128,17 @@ class ChatToolsApi {
             return results;
         });
     }
-    /**
-     * Get the default model for a media type. For video, prefers LTX-2.3 t2v models.
-     * Falls back to the model with the most available workers.
-     */
-    getDefaultModel(mediaType) {
+    getAvailableModels() {
         return __awaiter(this, void 0, void 0, function* () {
-            const models = yield this.projects.waitForModels(10000);
-            const candidates = models.filter((m) => m.media === mediaType);
-            if (candidates.length === 0) {
-                throw new Error(`No ${mediaType} models currently available on the network`);
-            }
-            // For video, prefer LTX-2.3 text-to-video models
-            if (mediaType === 'video') {
-                const ltx2 = candidates.filter((m) => m.id.includes('ltx2') && m.id.includes('t2v'));
-                if (ltx2.length > 0) {
-                    ltx2.sort((a, b) => b.workerCount - a.workerCount);
-                    return ltx2[0].id;
-                }
-            }
-            // Default: pick the model with most workers (highest availability)
-            candidates.sort((a, b) => b.workerCount - a.workerCount);
-            return candidates[0].id;
+            return this.projects.waitForModels(10000);
+        });
+    }
+    selectModel(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const models = yield this.getAvailableModels();
+            return (0, modelRouting_1.selectBackboneModel)(models, options).modelId;
         });
     }
-    /**
-     * Create a project, wait for completion with timeout, track per-job progress,
-     * and clean up on failure or timeout.
-     */
     executeProject(toolCall, mediaType, modelId, projectParams, prompt, options) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a, _b, _c, _d, _e;
@@ -159,7 +149,6 @@ class ChatToolsApi {
             let jobsCompleted = 0;
             let jobsFailed = 0;
             const totalJobs = projectParams.numberOfMedia || 1;
-            // Track per-job progress via project events
             const onJobCompleted = () => {
                 var _a;
                 jobsCompleted++;
@@ -169,7 +158,6 @@ class ChatToolsApi {
             const onJobFailed = () => {
                 jobsFailed++;
             };
-            // Track step-level progress via project progress event
             const onProgress = (percent) => {
                 var _a;
                 (_a = options === null || options === void 0 ? void 0 : options.onProgress) === null || _a === void 0 ? void 0 : _a.call(options, {
@@ -208,12 +196,11 @@ class ChatToolsApi {
                 };
             }
             catch (err) {
-                // Attempt to cancel the project on timeout to free worker resources
                 try {
                     yield project.cancel();
                 }
                 catch (_f) {
-                    /* best-effort */
+                    // best-effort cleanup
                 }
                 (_e = options === null || options === void 0 ? void 0 : options.onProgress) === null || _e === void 0 ? void 0 : _e.call(options, { status: 'failed', percent: 0 });
                 throw err;
@@ -229,12 +216,15 @@ class ChatToolsApi {
     }
     executeImageGeneration(toolCall, args, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const modelId = args.model || (yield this.getDefaultModel('image'));
+            const modelId = yield this.selectModel({
+                mediaType: 'image',
+                requestedModel: (0, modelRouting_1.resolveHostedToolModelSelector)('sogni_generate_image', args)
+            });
             const projectParams = {
                 type: 'image',
                 modelId,
                 positivePrompt: args.prompt,
-                numberOfMedia: (options === null || options === void 0 ? void 0 : options.numberOfMedia) || 1
+                numberOfMedia: getVariationCount(args, options)
             };
             if (args.negative_prompt)
                 projectParams.negativePrompt = args.negative_prompt;
@@ -247,6 +237,48 @@ class ChatToolsApi {
                 projectParams.steps = args.steps;
             if (args.seed !== undefined)
                 projectParams.seed = args.seed;
+            applyHostedImageOptions(projectParams, args);
+            if (options === null || options === void 0 ? void 0 : options.tokenType)
+                projectParams.tokenType = options.tokenType;
+            if (options === null || options === void 0 ? void 0 : options.network)
+                projectParams.network = options.network;
+            return this.executeProject(toolCall, 'image', modelId, projectParams, args.prompt, options);
+        });
+    }
+    executeImageEdit(toolCall, args, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const sourceImageUrl = (0, modelRouting_1.isNonEmptyString)(args.source_image_url) ? args.source_image_url : null;
+            const referenceImageUrls = (0, modelRouting_1.asStringArray)(args.reference_image_urls);
+            const inputUrls = [...(sourceImageUrl ? [sourceImageUrl] : []), ...referenceImageUrls];
+            if (inputUrls.length === 0) {
+                throw new Error('sogni_edit_image requires source_image_url or reference_image_urls');
+            }
+            const modelId = yield this.selectModel({
+                mediaType: 'image',
+                requestedModel: (0, modelRouting_1.resolveHostedToolModelSelector)('sogni_edit_image', args),
+                filter: modelRouting_1.isEditImageModel
+            });
+            const maxContextImages = (0, validation_1.getMaxContextImages)(modelId);
+            const contextImages = yield Promise.all(inputUrls.slice(0, maxContextImages).map((url) => (0, mediaValidation_1.parseInlineMediaDataUri)(url, 'image', {
+                maxBytes: MAX_INPUT_MEDIA_BYTES.image
+            }).blob));
+            const projectParams = {
+                type: 'image',
+                modelId,
+                positivePrompt: args.prompt,
+                numberOfMedia: getVariationCount(args, options),
+                contextImages
+            };
+            if (args.negative_prompt)
+                projectParams.negativePrompt = args.negative_prompt;
+            if (args.width && args.height) {
+                projectParams.width = args.width;
+                projectParams.height = args.height;
+                projectParams.sizePreset = 'custom';
+            }
+            if (args.seed !== undefined)
+                projectParams.seed = args.seed;
+            applyHostedImageOptions(projectParams, args);
             if (options === null || options === void 0 ? void 0 : options.tokenType)
                 projectParams.tokenType = options.tokenType;
             if (options === null || options === void 0 ? void 0 : options.network)
@@ -256,27 +288,197 @@ class ChatToolsApi {
     }
     executeVideoGeneration(toolCall, args, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const modelId = args.model || (yield this.getDefaultModel('video'));
-            // LTX-2.3 defaults: 1920x1088 landscape, 24fps
-            const isLtx2 = modelId.includes('ltx2');
-            const defaultWidth = isLtx2 ? 1920 : 848;
-            const defaultHeight = isLtx2 ? 1088 : 480;
-            const defaultFps = isLtx2 ? 24 : 16;
+            const hasReferenceImages = (0, modelRouting_1.isNonEmptyString)(args.reference_image_url) || (0, modelRouting_1.isNonEmptyString)(args.reference_image_end_url);
+            const workflowPreference = hasReferenceImages ? ['i2v'] : ['t2v'];
+            const preferredModelIds = hasReferenceImages
+                ? [modelRouting_1.PREFERRED_MODEL_IDS.video.i2v]
+                : [modelRouting_1.PREFERRED_MODEL_IDS.video.t2v];
+            const modelId = yield this.selectModel({
+                mediaType: 'video',
+                requestedModel: (0, modelRouting_1.resolveHostedToolModelSelector)('sogni_generate_video', args),
+                workflows: workflowPreference,
+                preferredModelIds
+            });
+            const defaults = (0, modelRouting_1.getVideoDefaults)(modelId);
+            const isSeedanceModel = modelId.startsWith('seedance-2-0');
             const projectParams = {
                 type: 'video',
                 modelId,
                 positivePrompt: args.prompt,
-                numberOfMedia: (options === null || options === void 0 ? void 0 : options.numberOfMedia) || 1,
-                width: args.width || defaultWidth,
-                height: args.height || defaultHeight,
-                fps: args.fps || defaultFps
+                numberOfMedia: getVariationCount(args, options),
+                width: args.width || defaults.width,
+                height: args.height || defaults.height,
+                fps: args.fps || defaults.fps
             };
-            if (args.negative_prompt)
+            if (args.negative_prompt && !isSeedanceModel) {
                 projectParams.negativePrompt = args.negative_prompt;
+            }
             if (args.duration !== undefined)
                 projectParams.duration = args.duration;
             if (args.seed !== undefined)
                 projectParams.seed = args.seed;
+            if ((0, modelRouting_1.isNonEmptyString)(args.reference_image_url)) {
+                projectParams.referenceImage = (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_image_url, 'image', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.image
+                }).blob;
+            }
+            if ((0, modelRouting_1.isNonEmptyString)(args.reference_image_end_url)) {
+                projectParams.referenceImageEnd = (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_image_end_url, 'image', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.image
+                }).blob;
+            }
+            if ((0, modelRouting_1.isNonEmptyString)(args.reference_audio_identity_url)) {
+                projectParams.referenceAudioIdentity = (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_audio_identity_url, 'audio', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.audio
+                }).blob;
+            }
+            if (args.audio_identity_strength !== undefined) {
+                projectParams.audioIdentityStrength = args.audio_identity_strength;
+            }
+            if (args.first_frame_strength !== undefined) {
+                projectParams.firstFrameStrength = args.first_frame_strength;
+            }
+            if (args.last_frame_strength !== undefined) {
+                projectParams.lastFrameStrength = args.last_frame_strength;
+            }
+            if (args.generate_audio !== undefined) {
+                projectParams.generateAudio = Boolean(args.generate_audio);
+            }
+            if (options === null || options === void 0 ? void 0 : options.tokenType)
+                projectParams.tokenType = options.tokenType;
+            if (options === null || options === void 0 ? void 0 : options.network)
+                projectParams.network = options.network;
+            return this.executeProject(toolCall, 'video', modelId, projectParams, args.prompt, options);
+        });
+    }
+    executeSoundToVideo(toolCall, args, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            if (!(0, modelRouting_1.isNonEmptyString)(args.reference_audio_url)) {
+                throw new Error('sogni_sound_to_video requires reference_audio_url');
+            }
+            const hasReferenceImage = (0, modelRouting_1.isNonEmptyString)(args.reference_image_url);
+            const workflows = hasReferenceImage ? ['ia2v', 's2v'] : ['a2v'];
+            const preferredModelIds = hasReferenceImage
+                ? [modelRouting_1.PREFERRED_MODEL_IDS.video.ia2v, modelRouting_1.PREFERRED_MODEL_IDS.video.s2v]
+                : [modelRouting_1.PREFERRED_MODEL_IDS.video.a2v];
+            const modelId = yield this.selectModel({
+                mediaType: 'video',
+                requestedModel: (0, modelRouting_1.resolveHostedToolModelSelector)('sogni_sound_to_video', args),
+                workflows,
+                preferredModelIds
+            });
+            const defaults = (0, modelRouting_1.getVideoDefaults)(modelId);
+            const duration = (_a = (0, modelRouting_1.asFiniteNumber)(args.duration)) !== null && _a !== void 0 ? _a : 5;
+            const projectParams = {
+                type: 'video',
+                modelId,
+                positivePrompt: args.prompt,
+                numberOfMedia: getVariationCount(args, options),
+                referenceAudio: (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_audio_url, 'audio', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.audio
+                }).blob,
+                width: args.width || defaults.width,
+                height: args.height || defaults.height,
+                fps: defaults.fps,
+                duration,
+                audioDuration: duration
+            };
+            if ((0, modelRouting_1.isNonEmptyString)(args.reference_image_url)) {
+                projectParams.referenceImage = (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_image_url, 'image', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.image
+                }).blob;
+            }
+            if (args.audio_start !== undefined)
+                projectParams.audioStart = args.audio_start;
+            if (args.generate_audio !== undefined) {
+                projectParams.generateAudio = Boolean(args.generate_audio);
+            }
+            if (args.seed !== undefined)
+                projectParams.seed = args.seed;
+            if (options === null || options === void 0 ? void 0 : options.tokenType)
+                projectParams.tokenType = options.tokenType;
+            if (options === null || options === void 0 ? void 0 : options.network)
+                projectParams.network = options.network;
+            return this.executeProject(toolCall, 'video', modelId, projectParams, args.prompt, options);
+        });
+    }
+    executeVideoToVideo(toolCall, args, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            if (!(0, modelRouting_1.isNonEmptyString)(args.reference_video_url)) {
+                throw new Error('sogni_video_to_video requires reference_video_url');
+            }
+            const controlMode = (0, modelRouting_1.normalizeVideoControlMode)(args.control_mode);
+            const isAnimateMode = controlMode === 'animate-move' || controlMode === 'animate-replace';
+            const isSeedanceMode = controlMode === 'seedance-v2v';
+            const workflows = isAnimateMode ? [controlMode] : ['v2v'];
+            const preferredModelIds = isAnimateMode
+                ? [
+                    controlMode === 'animate-move'
+                        ? modelRouting_1.PREFERRED_MODEL_IDS.video.animateMove
+                        : modelRouting_1.PREFERRED_MODEL_IDS.video.animateReplace
+                ]
+                : isSeedanceMode
+                    ? [modelRouting_1.PREFERRED_MODEL_IDS.video.seedanceV2v, modelRouting_1.PREFERRED_MODEL_IDS.video.v2v]
+                    : [modelRouting_1.PREFERRED_MODEL_IDS.video.v2v];
+            const modelId = yield this.selectModel({
+                mediaType: 'video',
+                requestedModel: (0, modelRouting_1.resolveHostedToolModelSelector)('sogni_video_to_video', args),
+                workflows,
+                preferredModelIds
+            });
+            const defaults = (0, modelRouting_1.getVideoDefaults)(modelId);
+            const isSeedanceModel = modelId.startsWith('seedance-2-0');
+            if (isAnimateMode && !(0, modelRouting_1.isNonEmptyString)(args.reference_image_url)) {
+                throw new Error(`${controlMode} requires reference_image_url`);
+            }
+            const projectParams = {
+                type: 'video',
+                modelId,
+                positivePrompt: args.prompt,
+                numberOfMedia: getVariationCount(args, options),
+                referenceVideo: (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_video_url, 'video', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.video
+                }).blob,
+                width: args.width || defaults.width,
+                height: args.height || defaults.height,
+                fps: defaults.fps,
+                duration: (_a = (0, modelRouting_1.asFiniteNumber)(args.duration)) !== null && _a !== void 0 ? _a : 5
+            };
+            if (args.negative_prompt && !isSeedanceModel) {
+                projectParams.negativePrompt = args.negative_prompt;
+            }
+            if (args.seed !== undefined)
+                projectParams.seed = args.seed;
+            if ((0, modelRouting_1.isNonEmptyString)(args.reference_image_url)) {
+                projectParams.referenceImage = (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_image_url, 'image', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.image
+                }).blob;
+            }
+            if ((0, modelRouting_1.isNonEmptyString)(args.reference_audio_identity_url)) {
+                projectParams.referenceAudioIdentity = (0, mediaValidation_1.parseInlineMediaDataUri)(args.reference_audio_identity_url, 'audio', {
+                    maxBytes: MAX_INPUT_MEDIA_BYTES.audio
+                }).blob;
+            }
+            if (args.audio_identity_strength !== undefined) {
+                projectParams.audioIdentityStrength = args.audio_identity_strength;
+            }
+            if (args.video_start !== undefined) {
+                projectParams.videoStart = args.video_start;
+            }
+            if (args.generate_audio !== undefined) {
+                projectParams.generateAudio = Boolean(args.generate_audio);
+            }
+            if (!isAnimateMode && !isSeedanceModel) {
+                projectParams.controlNet = {
+                    name: controlMode,
+                    strength: controlMode === 'detailer' ? 1 : 0.85
+                };
+            }
+            if (!isSeedanceModel && args.detailer_strength !== undefined) {
+                projectParams.detailerStrength = args.detailer_strength;
+            }
             if (options === null || options === void 0 ? void 0 : options.tokenType)
                 projectParams.tokenType = options.tokenType;
             if (options === null || options === void 0 ? void 0 : options.network)
@@ -286,12 +488,19 @@ class ChatToolsApi {
     }
     executeMusicGeneration(toolCall, args, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const modelId = args.model || (yield this.getDefaultModel('audio'));
+            const modelId = yield this.selectModel({
+                mediaType: 'audio',
+                requestedModel: (0, modelRouting_1.resolveHostedToolModelSelector)('sogni_generate_music', args),
+                preferredModelIds: [
+                    modelRouting_1.PREFERRED_MODEL_IDS.audio.aceStepTurbo,
+                    modelRouting_1.PREFERRED_MODEL_IDS.audio.aceStepSft
+                ]
+            });
             const projectParams = {
                 type: 'audio',
                 modelId,
                 positivePrompt: args.prompt,
-                numberOfMedia: (options === null || options === void 0 ? void 0 : options.numberOfMedia) || 1
+                numberOfMedia: getVariationCount(args, options)
             };
             if (args.duration !== undefined)
                 projectParams.duration = args.duration;
@@ -299,10 +508,24 @@ class ChatToolsApi {
                 projectParams.bpm = args.bpm;
             if (args.keyscale)
                 projectParams.keyscale = args.keyscale;
-            if (args.timesignature)
-                projectParams.timesignature = args.timesignature;
+            if (args.lyrics)
+                projectParams.lyrics = args.lyrics;
+            if (args.language)
+                projectParams.language = args.language;
             if (args.output_format)
                 projectParams.outputFormat = args.output_format;
+            const timeSignature = (0, modelRouting_1.normalizeTimeSignature)(args.timesignature);
+            if (timeSignature)
+                projectParams.timesignature = timeSignature;
+            const composerMode = (0, modelRouting_1.asBooleanValue)(args.composer_mode);
+            if (composerMode !== undefined)
+                projectParams.composerMode = composerMode;
+            const promptStrength = (0, modelRouting_1.asFiniteNumber)(args.prompt_strength);
+            if (promptStrength !== undefined)
+                projectParams.promptStrength = promptStrength;
+            const creativity = (0, modelRouting_1.asFiniteNumber)(args.creativity);
+            if (creativity !== undefined)
+                projectParams.creativity = creativity;
             if (args.seed !== undefined)
                 projectParams.seed = args.seed;
             if (options === null || options === void 0 ? void 0 : options.tokenType)