modality-kit 0.9.5 → 0.10.1

package/dist/index.js

@@ -5659,6 +5659,8 @@ var logger3 = getLoggerInstance("WebSocket-Client");
 
 class WebSocketClient {
   ws = null;
+  wsStream = null;
+  streamWriter = null;
   url;
   config = {
     initialReconnectDelay: 1000,
@@ -5742,24 +5744,60 @@ class WebSocketClient {
     const url = new URL(this.url);
     return url.searchParams.get("clientId") ?? (this.connectionId ? String(this.connectionId) : "");
   }
-  send(data) {
-    if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+  async initializeWebSocketStream() {
+    if (typeof WebSocketStream === "undefined") {
+      logger3.warn("WebSocketStream not supported in this environment");
+      return false;
+    }
+    try {
+      this.wsStream = new WebSocketStream(this.url);
+      const { writable } = await this.wsStream.opened;
+      this.streamWriter = writable.getWriter();
+      logger3.info("WebSocketStream initialized successfully");
+      return true;
+    } catch (error) {
+      logger3.error("Failed to initialize WebSocketStream:", error);
+      this.wsStream = null;
+      this.streamWriter = null;
+      return false;
+    }
+  }
+  async send(data, useStream = false) {
+    const message = JSON.stringify({
+      ...data,
+      jsonrpc: "2.0",
+      timestamp: new Date().toISOString()
+    });
+    if (useStream) {
+      if (!this.streamWriter) {
+        const streamInitialized = await this.initializeWebSocketStream();
+        if (!streamInitialized) {
+          logger3.warn("WebSocketStream not available, cannot send with streaming");
+          return false;
+        }
+      }
       try {
-        const message = JSON.stringify({
-          ...data,
-          jsonrpc: "2.0",
-          timestamp: new Date().toISOString()
-        });
-        this.ws.send(message);
-        logger3.info("Message sent:", message);
+        await this.streamWriter.write(message);
+        logger3.info("Message sent via WebSocketStream:", message);
         return true;
       } catch (error) {
-        logger3.error("Error sending message:", error);
+        logger3.error("Error sending via WebSocketStream:", error);
         return false;
       }
     } else {
-      logger3.warn("WebSocket is not connected during send operation.");
-      return false;
+      if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+        try {
+          this.ws.send(message);
+          logger3.info("Message sent:", message);
+          return true;
+        } catch (error) {
+          logger3.error("Error sending message:", error);
+          return false;
+        }
+      } else {
+        logger3.warn("WebSocket is not connected during send operation.");
+        return false;
+      }
     }
   }
   isConnected() {
@@ -5848,6 +5886,41 @@ class WebSocketClient {
     }
   }
 }
+// src/lruCache.ts
+class LruCache {
+  values = new Map;
+  max;
+  constructor(max = 100) {
+    this.max = max;
+  }
+  has(key) {
+    return this.values.has(key);
+  }
+  get(key) {
+    const value = this.values.get(key);
+    if (value) {
+      this.values.delete(key);
+      this.values.set(key, value);
+    }
+    return value;
+  }
+  set(key, value) {
+    if (this.values.size >= this.max) {
+      const itemsToEvictCount = Math.max(1, Math.floor(this.max * 0.25));
+      const keys = this.values.keys();
+      let count = 0;
+      while (count < itemsToEvictCount) {
+        const next = keys.next();
+        if (next.done) {
+          break;
+        }
+        this.values.delete(next.value);
+        count++;
+      }
+    }
+    this.values.set(key, value);
+  }
+}
 export {
   withErrorHandling,
   setupAITools,
@@ -5860,6 +5933,7 @@ export {
   compressWithLanguageDetection as compressText,
   WebSocketClient,
   exports_schemas_symbol as SymbolTypes,
+  LruCache,
   JSONRPCUtils,
   JSONRPCManager,
   JSONRPCErrorCode,

package/dist/types/__tests__/lruCache.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/types/index.d.ts

@@ -15,3 +15,4 @@ export type { JSONRPCMessage, JSONRPCRequest, JSONRPCNotification, JSONRPCRespon
 export { JSONRPCManager } from "./jsonrpc-manager";
 export type { JSONRPCManagerEvents, JSONRPCManagerConfig, } from "./jsonrpc-manager";
 export { WebSocketClient } from "./websocket-client";
+export { LruCache } from "./lruCache";

package/dist/types/lruCache.d.ts

@@ -0,0 +1,8 @@
+export declare class LruCache<T> {
+    private values;
+    private max;
+    constructor(max?: number);
+    has(key: string): boolean;
+    get(key: string): T | undefined;
+    set(key: string, value: T): void;
+}

package/dist/types/schemas/schemas_symbol.d.ts

@@ -60,7 +60,6 @@ interface SymbolBase {
     id?: string;
     uri?: string;
     detail?: string;
-    content?: string;
     containerName?: string;
 }
 export interface Symbol extends SymbolBase {

package/dist/types/websocket-client.d.ts

@@ -1,4 +1,17 @@
 import type { JSONRPCValidationResult } from "./schemas/jsonrpc";
+declare global {
+    interface WebSocketStream {
+        readonly opened: Promise<{
+            readable: ReadableStream;
+            writable: WritableStream;
+        }>;
+        readonly closed: Promise<void>;
+        close(code?: number, reason?: string): void;
+    }
+    const WebSocketStream: {
+        new (url: string): WebSocketStream;
+    };
+}
 interface WebSocketConfig {
     maxReconnectAttempts: number;
     initialReconnectDelay: number;
@@ -18,6 +31,8 @@ interface WebSocketInfo {
 }
 export declare class WebSocketClient {
     private ws;
+    private wsStream;
+    private streamWriter;
     private url;
     private config;
     private connectionId;
@@ -35,7 +50,8 @@ export declare class WebSocketClient {
     private onOpen;
     private onClose;
     private getClientId;
-    send(data: any): boolean;
+    private initializeWebSocketStream;
+    send(data: any, useStream?: boolean): Promise<boolean>;
     isConnected(): boolean;
     getInfo(): WebSocketInfo;
     /**

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.9.5",
+  "version": "0.10.1",
   "name": "modality-kit",
   "repository": {
     "type": "git",