@agntk/client 0.1.0

package/dist/websocket-client.js

@@ -0,0 +1,140 @@
+// WebSocket client for streaming
+import { createLogger } from '@agntk/logger';
+import { WebSocketError } from './errors.js';
+const log = createLogger('@agntk/client:ws');
+export class AgentWebSocketClient {
+    ws = null;
+    config;
+    state = 'disconnected';
+    reconnectAttempts = 0;
+    reconnectTimer = null;
+    onStateChange;
+    constructor(config) {
+        this.config = {
+            reconnect: true,
+            reconnectInterval: 1000,
+            maxReconnectAttempts: 5,
+            ...config
+        };
+        log.debug('Created WebSocket client', { url: config.url });
+    }
+    getState() {
+        return this.state;
+    }
+    onConnectionStateChange(handler) {
+        this.onStateChange = handler;
+    }
+    setState(newState) {
+        if (this.state !== newState) {
+            this.state = newState;
+            this.onStateChange?.(newState);
+            log.debug('Connection state changed', { state: newState });
+        }
+    }
+    connect() {
+        if (this.state === 'connected')
+            return Promise.resolve();
+        this.setState('connecting');
+        log.debug('Connecting');
+        return new Promise((resolve, reject) => {
+            try {
+                this.ws = new WebSocket(this.config.url);
+                this.ws.onopen = () => {
+                    log.info('Connected');
+                    this.setState('connected');
+                    this.reconnectAttempts = 0;
+                    resolve();
+                };
+                this.ws.onclose = () => {
+                    log.info('Connection closed');
+                    if (this.state !== 'disconnected') {
+                        this.handleReconnect();
+                    }
+                };
+                this.ws.onerror = (error) => {
+                    log.error('Connection failed', { error });
+                    if (this.state === 'connecting') {
+                        this.setState('disconnected');
+                        reject(new WebSocketError('Connection failed'));
+                    }
+                };
+            }
+            catch (error) {
+                this.setState('disconnected');
+                reject(new WebSocketError('Failed to create WebSocket'));
+            }
+        });
+    }
+    handleReconnect() {
+        if (!this.config.reconnect || this.state === 'disconnected')
+            return;
+        if (this.reconnectAttempts >= (this.config.maxReconnectAttempts || 5)) {
+            log.error('Max reconnect attempts reached');
+            this.setState('disconnected');
+            return;
+        }
+        this.setState('reconnecting');
+        this.reconnectAttempts++;
+        const delay = Math.min((this.config.reconnectInterval || 1000) * Math.pow(2, this.reconnectAttempts - 1), 30000);
+        log.info('Scheduling reconnect', { attempt: this.reconnectAttempts, delay });
+        if (this.reconnectTimer)
+            clearTimeout(this.reconnectTimer);
+        this.reconnectTimer = setTimeout(() => {
+            this.connect().catch(() => {
+                // Error handling is done in connect()
+            });
+        }, delay);
+    }
+    send(message, callbacks) {
+        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+            throw new WebSocketError('WebSocket not connected');
+        }
+        log.debug('Sending message');
+        this.ws.send(JSON.stringify(message));
+        this.ws.onmessage = (event) => {
+            try {
+                const data = JSON.parse(event.data);
+                this.dispatch(data, callbacks);
+            }
+            catch (_err) {
+                callbacks.onError?.('Failed to parse message');
+            }
+        };
+    }
+    dispatch(event, callbacks) {
+        switch (event.type) {
+            case 'text-delta':
+                callbacks.onTextDelta?.(event.textDelta);
+                break;
+            case 'tool-call':
+                callbacks.onToolCall?.(event.toolCallId, event.toolName, event.args);
+                break;
+            case 'tool-result':
+                callbacks.onToolResult?.(event.toolCallId, event.toolName, event.result);
+                break;
+            case 'step-start':
+                callbacks.onStepStart?.(event.stepIndex);
+                break;
+            case 'step-finish':
+                callbacks.onStepFinish?.(event.stepIndex, event.finishReason);
+                break;
+            case 'finish':
+                callbacks.onComplete?.({ text: event.text, usage: event.usage });
+                break;
+            case 'error':
+                callbacks.onError?.(event.error);
+                break;
+        }
+    }
+    disconnect() {
+        log.debug('Disconnecting');
+        this.setState('disconnected');
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+        this.ws?.close();
+        this.ws = null;
+    }
+}
+//# sourceMappingURL=websocket-client.js.map

package/dist/websocket-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-client.js","sourceRoot":"","sources":["../src/websocket-client.ts"],"names":[],"mappings":"AAAA,iCAAiC;AAEjC,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAG1C,MAAM,GAAG,GAAG,YAAY,CAAC,kBAAkB,CAAC,CAAC;AAqB7C,MAAM,OAAO,oBAAoB;IACvB,EAAE,GAAqB,IAAI,CAAC;IAC5B,MAAM,CAAwB;IAC9B,KAAK,GAAoB,cAAc,CAAC;IACxC,iBAAiB,GAAG,CAAC,CAAC;IACtB,cAAc,GAA0B,IAAI,CAAC;IAC7C,aAAa,CAAoC;IAEzD,YAAY,MAA6B;QACvC,IAAI,CAAC,MAAM,GAAG;YACZ,SAAS,EAAE,IAAI;YACf,iBAAiB,EAAE,IAAI;YACvB,oBAAoB,EAAE,CAAC;YACvB,GAAG,MAAM;SACV,CAAC;QACF,GAAG,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,uBAAuB,CAAC,OAAyC;QAC/D,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;IAC/B,CAAC;IAEO,QAAQ,CAAC,QAAyB;QACxC,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;YACtB,IAAI,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,CAAC;YAC/B,GAAG,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,KAAK,KAAK,WAAW;YAAE,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAEzD,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC5B,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAExB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAEzC,IAAI,CAAC,EAAE,CAAC,MAAM,GAAG,GAAG,EAAE;oBACpB,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBACtB,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;oBAC3B,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;oBAC3B,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC;gBAEF,IAAI,CAAC,EAAE,CAAC,OAAO,GAAG,GAAG,EAAE;oBACrB,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;oBAC9B,IAAI,IAAI,CAAC,KAAK,KAAK,cAAc,EAAE,CAAC;wBAClC,IAAI,CAAC,eAAe,EAAE,CAAC;oBACzB,CAAC;gBACH,CAAC,CAAC;gBAEF,IAAI,CAAC,EAAE,CAAC,OAAO,GAAG,CAAC,KAAK,EAAE,EAAE;oBAC1B,GAAG,CAAC,KAAK,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;oBAC1C,IAAI,IAAI,CAAC,KAAK,KAAK,YAAY,EAAE,CAAC;wBAChC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;wBAC9B,MAAM,CAAC,IAAI,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;gBAC9B,MAAM,CAAC,IAAI,cAAc,CAAC,4BAA4B,CAAC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,KAAK,cAAc;YAAE,OAAO;QAEpE,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,CAAC,CAAC,EAAE,CAAC;YACtE,GAAG,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAC5C,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YAC9B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QAC9B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CACpB,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC,EACjF,KAAK,CACN,CAAC;QAEF,GAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;QAE7E,IAAI,IAAI,CAAC,cAAc;YAAE,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAE3D,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;gBACxB,sCAAsC;YACxC,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAED,IAAI,CAAC,OAAoB,EAAE,SAA6B;QACtD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACtD,MAAM,IAAI,cAAc,CAAC,yBAAyB,CAAC,CAAC;QACtD,CAAC;QACD,GAAG,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAC7B,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QAEtC,IAAI,CAAC,EAAE,CAAC,SAAS,GAAG,CAAC,KAAK,EAAE,EAAE;YAC5B,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAgB,CAAC;gBACnD,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACjC,CAAC;YAAC,OAAO,IAAa,EAAE,CAAC;gBACvB,SAAS,CAAC,OAAO,EAAE,CAAC,yBAAyB,CAAC,CAAC;YACjD,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IAEO,QAAQ,CAAC,KAAkB,EAAE,SAA6B;QAChE,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,YAAY;gBACf,SAAS,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBACzC,MAAM;YAER,KAAK,WAAW;gBACd,SAAS,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBACrE,MAAM;YAER,KAAK,aAAa;gBAChB,SAAS,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBACzE,MAAM;YAER,KAAK,YAAY;gBACf,SAAS,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBACzC,MAAM;YAER,KAAK,aAAa;gBAChB,SAAS,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC9D,MAAM;YAER,KAAK,QAAQ;gBACX,SAAS,CAAC,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;gBACjE,MAAM;YAER,KAAK,OAAO;gBACV,SAAS,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACjC,MAAM;QACV,CAAC;IACH,CAAC;IAED,UAAU;QACR,GAAG,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;QAC3B,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QAC9B,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;IACjB,CAAC;CACF"}

package/package.json

@@ -0,0 +1,58 @@
+{
+  "name": "@agntk/client",
+  "version": "0.1.0",
+  "type": "module",
+  "description": "HTTP/WebSocket client for @agntk/server",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc && node ../../scripts/fix-esm-imports.mjs ./dist",
+    "typecheck": "tsc --noEmit",
+    "dev": "tsc --watch",
+    "test": "vitest",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@agntk/logger": "workspace:*",
+    "ai": "6.0.77"
+  },
+  "devDependencies": {
+    "@agntk/config-typescript": "workspace:*",
+    "typescript": "^5.7.3",
+    "vitest": "^3.2.3",
+    "@types/node": "^20.17.19"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/agntk/agntk.git",
+    "directory": "packages/sdk-client"
+  },
+  "homepage": "https://github.com/agntk/agntk#readme",
+  "bugs": {
+    "url": "https://github.com/agntk/agntk/issues"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  },
+  "keywords": [
+    "ai",
+    "agent",
+    "client",
+    "http",
+    "websocket",
+    "sse"
+  ],
+  "license": "MIT"
+}