agent-swarm-kit 1.0.20 → 1.0.22

package/build/index.cjs

@@ -294,13 +294,13 @@ var CC_SWARM_DEFAULT_AGENT = function (_a, _b, defaultAgent_1) { return __awaite
     });
 }); };
 var CC_KEEP_MESSAGES = 5;
-var CC_ANSWER_TIMEOUT_SECONDS = 120000;
+var CC_ANSWER_TIMEOUT = 120000;
 var CC_GET_AGENT_HISTORY = function () { return new functoolsKit.PubsubArrayAdapter(); };
 var GLOBAL_CONFIG = {
     CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
     CC_KEEP_MESSAGES: CC_KEEP_MESSAGES,
-    CC_ANSWER_TIMEOUT_SECONDS: CC_ANSWER_TIMEOUT_SECONDS,
+    CC_ANSWER_TIMEOUT: CC_ANSWER_TIMEOUT,
     CC_GET_AGENT_HISTORY: CC_GET_AGENT_HISTORY,
     CC_SWARM_AGENT_CHANGED: CC_SWARM_AGENT_CHANGED,
     CC_SWARM_DEFAULT_AGENT: CC_SWARM_DEFAULT_AGENT,
@@ -1249,7 +1249,7 @@ var ClientSwarm = /** @class */ (function () {
                         START_TIME = Date.now();
                         _c.label = 1;
                     case 1:
-                        if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT_SECONDS) {
+                        if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT) {
                             throw new Error("agent-swarm ClientSwarm waitForOutput timeout reached for ".concat(this.params.swarmName));
                         }
                         return [4 /*yield*/, Promise.race(__spreadArray(__spreadArray([], __read(Object.entries(this.params.agentMap).map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
@@ -4189,6 +4189,60 @@ var getLastSystemMessage = function (clientId) { return __awaiter(void 0, void 0
     });
 }); };
 
+var DEFAULT_TIMEOUT = 15 * 60;
+/**
+ * Creates an auto-dispose mechanism for a client in a swarm.
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {SwarmName} swarmName - The name of the swarm.
+ * @param {Partial<IMakeDisposeParams>} [params={}] - Optional parameters for auto-dispose.
+ * @returns {Object} An object with tick and stop methods to control the auto-dispose.
+ */
+var makeAutoDispose = function (clientId, swarmName, _a) {
+    var _b = _a === void 0 ? {} : _a, _c = _b.timeoutSeconds, timeoutSeconds = _c === void 0 ? DEFAULT_TIMEOUT : _c;
+    var stateEmitter = new functoolsKit.Subject();
+    var unSource = functoolsKit.Source.join([
+        stateEmitter.toObserver(),
+        functoolsKit.Source.fromInterval(1000),
+    ])
+        .reduce(function (acm, _a) {
+        var _b = __read(_a, 1), isOk = _b[0];
+        if (isOk) {
+            return acm + 1;
+        }
+        return 0;
+    }, 0)
+        .filter(function (ticker) { return ticker >= timeoutSeconds; })
+        .once(function () { return __awaiter(void 0, void 0, void 0, function () {
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    unSource();
+                    if (!swarm.sessionValidationService.hasSession(clientId)) return [3 /*break*/, 2];
+                    return [4 /*yield*/, disposeConnection(clientId, swarmName)];
+                case 1:
+                    _a.sent();
+                    _a.label = 2;
+                case 2: return [2 /*return*/];
+            }
+        });
+    }); });
+    return {
+        /**
+         * Signals that the client is active, resetting the auto-dispose timer.
+         */
+        tick: function () {
+            stateEmitter.next(true);
+        },
+        /**
+         * Stops the auto-dispose mechanism.
+         */
+        stop: function () {
+            unSource();
+        },
+    };
+};
+
 exports.ContextService = ContextService;
 exports.addAgent = addAgent;
 exports.addCompletion = addCompletion;
@@ -4210,6 +4264,7 @@ exports.getLastSystemMessage = getLastSystemMessage;
 exports.getLastUserMessage = getLastUserMessage;
 exports.getRawHistory = getRawHistory;
 exports.getUserHistory = getUserHistory;
+exports.makeAutoDispose = makeAutoDispose;
 exports.makeConnection = makeConnection;
 exports.session = session;
 exports.setConfig = setConfig;

package/build/index.mjs

@@ -1,5 +1,5 @@
 import { scoped } from 'di-scoped';
-import { ToolRegistry, sleep, PubsubArrayAdapter, Subject, queued, not, trycatch, memoize, schedule, ttl, singleshot } from 'functools-kit';
+import { ToolRegistry, sleep, PubsubArrayAdapter, Subject, queued, not, trycatch, memoize, schedule, ttl, singleshot, Source } from 'functools-kit';
 import { createActivator } from 'di-kit';
 import { omit } from 'lodash-es';
 import xml2js from 'xml2js';
@@ -292,13 +292,13 @@ var CC_SWARM_DEFAULT_AGENT = function (_a, _b, defaultAgent_1) { return __awaite
     });
 }); };
 var CC_KEEP_MESSAGES = 5;
-var CC_ANSWER_TIMEOUT_SECONDS = 120000;
+var CC_ANSWER_TIMEOUT = 120000;
 var CC_GET_AGENT_HISTORY = function () { return new PubsubArrayAdapter(); };
 var GLOBAL_CONFIG = {
     CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
     CC_KEEP_MESSAGES: CC_KEEP_MESSAGES,
-    CC_ANSWER_TIMEOUT_SECONDS: CC_ANSWER_TIMEOUT_SECONDS,
+    CC_ANSWER_TIMEOUT: CC_ANSWER_TIMEOUT,
     CC_GET_AGENT_HISTORY: CC_GET_AGENT_HISTORY,
     CC_SWARM_AGENT_CHANGED: CC_SWARM_AGENT_CHANGED,
     CC_SWARM_DEFAULT_AGENT: CC_SWARM_DEFAULT_AGENT,
@@ -1247,7 +1247,7 @@ var ClientSwarm = /** @class */ (function () {
                         START_TIME = Date.now();
                         _c.label = 1;
                     case 1:
-                        if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT_SECONDS) {
+                        if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT) {
                             throw new Error("agent-swarm ClientSwarm waitForOutput timeout reached for ".concat(this.params.swarmName));
                         }
                         return [4 /*yield*/, Promise.race(__spreadArray(__spreadArray([], __read(Object.entries(this.params.agentMap).map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
@@ -4187,4 +4187,58 @@ var getLastSystemMessage = function (clientId) { return __awaiter(void 0, void 0
     });
 }); };
 
-export { ContextService, addAgent, addCompletion, addSwarm, addTool, changeAgent, commitSystemMessage, commitToolOutput, commitUserMessage, complete, disposeConnection, emit, execute, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getUserHistory, makeConnection, session, setConfig, swarm };
+var DEFAULT_TIMEOUT = 15 * 60;
+/**
+ * Creates an auto-dispose mechanism for a client in a swarm.
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {SwarmName} swarmName - The name of the swarm.
+ * @param {Partial<IMakeDisposeParams>} [params={}] - Optional parameters for auto-dispose.
+ * @returns {Object} An object with tick and stop methods to control the auto-dispose.
+ */
+var makeAutoDispose = function (clientId, swarmName, _a) {
+    var _b = _a === void 0 ? {} : _a, _c = _b.timeoutSeconds, timeoutSeconds = _c === void 0 ? DEFAULT_TIMEOUT : _c;
+    var stateEmitter = new Subject();
+    var unSource = Source.join([
+        stateEmitter.toObserver(),
+        Source.fromInterval(1000),
+    ])
+        .reduce(function (acm, _a) {
+        var _b = __read(_a, 1), isOk = _b[0];
+        if (isOk) {
+            return acm + 1;
+        }
+        return 0;
+    }, 0)
+        .filter(function (ticker) { return ticker >= timeoutSeconds; })
+        .once(function () { return __awaiter(void 0, void 0, void 0, function () {
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    unSource();
+                    if (!swarm.sessionValidationService.hasSession(clientId)) return [3 /*break*/, 2];
+                    return [4 /*yield*/, disposeConnection(clientId, swarmName)];
+                case 1:
+                    _a.sent();
+                    _a.label = 2;
+                case 2: return [2 /*return*/];
+            }
+        });
+    }); });
+    return {
+        /**
+         * Signals that the client is active, resetting the auto-dispose timer.
+         */
+        tick: function () {
+            stateEmitter.next(true);
+        },
+        /**
+         * Stops the auto-dispose mechanism.
+         */
+        stop: function () {
+            unSource();
+        },
+    };
+};
+
+export { ContextService, addAgent, addCompletion, addSwarm, addTool, changeAgent, commitSystemMessage, commitToolOutput, commitUserMessage, complete, disposeConnection, emit, execute, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getUserHistory, makeAutoDispose, makeConnection, session, setConfig, swarm };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-swarm-kit",
-  "version": "1.0.20",
+  "version": "1.0.22",
   "description": "A TypeScript library for building orchestrated framework-agnostic multi-agent AI systems",
   "author": {
     "name": "Petr Tripolsky",

package/types.d.ts

@@ -1766,15 +1766,43 @@ declare const getLastAssistantMessage: (clientId: string) => Promise<string>;
  */
 declare const getLastSystemMessage: (clientId: string) => Promise<string>;
 
+/**
+ * Interface for the parameters of the makeAutoDispose function.
+ */
+interface IMakeDisposeParams {
+    /**
+     * Timeout in seconds before auto-dispose is triggered.
+     */
+    timeoutSeconds: number;
+}
+/**
+ * Creates an auto-dispose mechanism for a client in a swarm.
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {SwarmName} swarmName - The name of the swarm.
+ * @param {Partial<IMakeDisposeParams>} [params={}] - Optional parameters for auto-dispose.
+ * @returns {Object} An object with tick and stop methods to control the auto-dispose.
+ */
+declare const makeAutoDispose: (clientId: string, swarmName: SwarmName, { timeoutSeconds }?: Partial<IMakeDisposeParams>) => {
+    /**
+     * Signals that the client is active, resetting the auto-dispose timer.
+     */
+    tick(): void;
+    /**
+     * Stops the auto-dispose mechanism.
+     */
+    stop(): void;
+};
+
 declare const GLOBAL_CONFIG: {
     CC_TOOL_CALL_EXCEPTION_PROMPT: string;
     CC_EMPTY_OUTPUT_PLACEHOLDERS: string[];
     CC_KEEP_MESSAGES: number;
-    CC_ANSWER_TIMEOUT_SECONDS: number;
+    CC_ANSWER_TIMEOUT: number;
     CC_GET_AGENT_HISTORY: (clientId: string, agentName: AgentName) => IPubsubArray<IModelMessage>;
     CC_SWARM_AGENT_CHANGED: (clientId: string, agentName: AgentName, swarmName: SwarmName) => Promise<void>;
     CC_SWARM_DEFAULT_AGENT: (clientId: string, swarmName: SwarmName, defaultAgent: AgentName) => Promise<AgentName>;
 };
 declare const setConfig: (config: typeof GLOBAL_CONFIG) => void;
 
-export { ContextService, type IAgentSchema, type IAgentTool, type ICompletionSchema, type IIncomingMessage, type IMakeConnectionConfig, type IModelMessage, type IOutgoingMessage, type ISessionConfig, type ISwarmSchema, type ITool, type IToolCall, type ReceiveMessageFn, type SendMessageFn$1 as SendMessageFn, addAgent, addCompletion, addSwarm, addTool, changeAgent, commitSystemMessage, commitToolOutput, commitUserMessage, complete, disposeConnection, emit, execute, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getUserHistory, makeConnection, session, setConfig, swarm };
+export { ContextService, type IAgentSchema, type IAgentTool, type ICompletionSchema, type IIncomingMessage, type IMakeConnectionConfig, type IMakeDisposeParams, type IModelMessage, type IOutgoingMessage, type ISessionConfig, type ISwarmSchema, type ITool, type IToolCall, type ReceiveMessageFn, type SendMessageFn$1 as SendMessageFn, addAgent, addCompletion, addSwarm, addTool, changeAgent, commitSystemMessage, commitToolOutput, commitUserMessage, complete, disposeConnection, emit, execute, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getUserHistory, makeAutoDispose, makeConnection, session, setConfig, swarm };