@awarevue/agent-sdk 2.0.78 → 2.0.80

package/dist/agent-protocol.d.ts

@@ -13,6 +13,11 @@ export type Inbound<D extends Direction> = D extends 'server' ? FromAgent : From
 export type Outbound<D extends Direction> = D extends 'server' ? FromServer : FromAgent;
 export type InMsg<D extends Direction> = Message<Inbound<D>>;
 export type OutMsg<D extends Direction> = Message<Outbound<D>>;
+export interface GetReplyOptions {
+    /** Timeout for awaiting replies to sent messages (default 10s) */
+    timeoutMs?: number;
+    onProgress?: (progress: Message<PayloadByKind['progress']>) => void;
+}
 export interface AgentProtocolOptions {
     /** Timeout for awaiting replies to sent messages (default 10s) */
     replyTimeout?: number;
@@ -27,6 +32,6 @@ export declare class AgentProtocol<D extends Direction> {
     constructor(transport: DuplexTransport<Message<Inbound<D>>, Message<Outbound<D>>>, options: AgentProtocolOptions);
     getReply$: <K extends RequestKind>(payload: Extract<Outbound<D>, {
         kind: K;
-    }>, timeoutMs?: number) => Observable<Message<PayloadByKind[ResponseKind<K>]>>;
+    }>, { timeoutMs, onProgress }?: GetReplyOptions) => Observable<Message<PayloadByKind[ResponseKind<K>]>>;
     send: (payload: Outbound<D>) => void;
 }

package/dist/agent-protocol.js

@@ -15,7 +15,7 @@ class AgentProtocol {
             version: constants_1.AGENT_PROTOCOL_VERSION,
             on: Date.now(),
         });
-        this.getReply$ = (payload, timeoutMs) => {
+        this.getReply$ = (payload, { timeoutMs, onProgress } = {}) => {
             const responseKind = api_types_1.ReplyKindByRequestKind[payload.kind];
             const reply$ = (id) => this.transport.messages$.pipe(
             // pass through only messages with matching requestId (it could be response, error or progress update)
@@ -29,6 +29,12 @@ class AgentProtocol {
             }), 
             // enforce timeout (progress updates will reset the timer, but if no messages arrive for `timeoutMs` duration, the request is considered failed)
             (0, rxjs_1.timeout)(timeoutMs || this.options.replyTimeout || 10000), 
+            // if an onProgress callback was provided, tap into the stream to route progress updates to the callback
+            (0, rxjs_1.tap)((message) => {
+                if (message.kind === 'progress' && onProgress) {
+                    onProgress(message);
+                }
+            }), 
             // pass through only the final response message (filter out progress updates)
             (0, rxjs_1.filter)((message) => message.kind === responseKind), 
             // we only expect one response message, so complete the stream after the response arrives

package/dist/agent-server.d.ts

@@ -47,7 +47,8 @@ export declare class AgentServer<TPeer extends PeerId = string> {
     private onPeerLeave;
     private processMessage;
     startAgent({ agentId, ...rest }: StartAgentArgs): void;
-    stopAgent(agentId: string, provider: string): void;
+    stopAgent(agentId: string, provider: string): Promise<void>;
+    restartAgent({ agentId, ...rest }: StartAgentArgs): Promise<void>;
     finalize(): void;
     getAgentSender(agentId: string): AgentProtocol<'server'> | null;
     private getPeerSender;

package/dist/agent-server.js

@@ -119,9 +119,19 @@ class AgentServer {
         const protocol = this.getAgentSender(agentId);
         protocol === null || protocol === void 0 ? void 0 : protocol.send({ kind: 'start', ...rest });
     }
-    stopAgent(agentId, provider) {
+    async stopAgent(agentId, provider) {
         const protocol = this.getAgentSender(agentId);
-        protocol === null || protocol === void 0 ? void 0 : protocol.send({ kind: 'stop', provider });
+        if (!protocol) {
+            this.notifications$.next(`Cannot stop agent ${agentId} - no connection found`);
+            return;
+        }
+        await (0, rxjs_1.lastValueFrom)(protocol.getReply$({ kind: 'stop', provider })).catch((err) => {
+            this.notifications$.next(`Failed to stop agent ${agentId}: ${err.message}`);
+        });
+    }
+    async restartAgent({ agentId, ...rest }) {
+        await this.stopAgent(agentId, rest.provider);
+        this.startAgent({ agentId, ...rest });
     }
     finalize() {
         var _a;

package/dist/package.json

@@ -4,7 +4,7 @@
     "type": "git",
     "url": "git+https://github.com/Linc-Security-Systems/aware-essentials.git"
   },
-  "version": "2.0.78",
+  "version": "2.0.80",
   "description": "SDK for building Agent implementations that speak the agent protocol.",
   "author": "Yaser Awajan",
   "license": "MIT",
@@ -31,12 +31,12 @@
     "lint:fix": "yarn lint --fix"
   },
   "peerDependencies": {
-    "@awarevue/api-types": "2.0.78",
+    "@awarevue/api-types": "2.0.80",
     "rxjs": "^7.8.2",
     "ws": "^8"
   },
   "devDependencies": {
-    "@awarevue/api-types": "2.0.78",
+    "@awarevue/api-types": "2.0.80",
     "@types/node": "^20.12.7",
     "@types/ws": "^8.18.1",
     "@typescript-eslint/eslint-plugin": "^8.31.1",

package/package.json

@@ -4,7 +4,7 @@
     "type": "git",
     "url": "git+https://github.com/Linc-Security-Systems/aware-essentials.git"
   },
-  "version": "2.0.78",
+  "version": "2.0.80",
   "description": "SDK for building Agent implementations that speak the agent protocol.",
   "author": "Yaser Awajan",
   "license": "MIT",
@@ -31,12 +31,12 @@
     "lint:fix": "yarn lint --fix"
   },
   "peerDependencies": {
-    "@awarevue/api-types": "2.0.78",
+    "@awarevue/api-types": "2.0.80",
     "rxjs": "^7.8.2",
     "ws": "^8"
   },
   "devDependencies": {
-    "@awarevue/api-types": "2.0.78",
+    "@awarevue/api-types": "2.0.80",
     "@types/node": "^20.12.7",
     "@types/ws": "^8.18.1",
     "@typescript-eslint/eslint-plugin": "^8.31.1",