recursive-llm-ts 1.0.6 → 2.0.0

package/README.md

@@ -10,18 +10,31 @@ npm install recursive-llm-ts
 
 ### Prerequisites
 
-- **Node.js** (not Bun) - This package uses `pythonia` which requires Node.js internals
+- **Runtime:** Node.js **or** Bun (both supported with automatic detection)
 - **Python 3.9+** - Required for the underlying recursive-llm Python package
 - **pip** - Python package manager
 
 **Python dependencies are automatically installed** via the `postinstall` script when you `npm install`. The script will run `pip install -e` on the bundled Python package, installing `litellm`, `RestrictedPython`, and other dependencies.
 
+#### For Bun Users
+If using Bun, also install `bunpy`:
+```bash
+bun add bunpy
+```
+
+The package will automatically detect your runtime and use the appropriate Python bridge:
+- **Node.js** → uses `pythonia` (included)
+- **Bun** → uses `bunpy` (peer dependency)
+
 ## Usage
 
+### Automatic Runtime Detection (Recommended)
+
 ```typescript
 import { RLM } from 'recursive-llm-ts';
 
 // Initialize RLM with a model
+// Automatically detects Node.js or Bun and uses appropriate bridge
 const rlm = new RLM('gpt-4o-mini', {
   max_iterations: 15,
   api_key: process.env.OPENAI_API_KEY
@@ -37,6 +50,26 @@ console.log(result.result);
 console.log('Stats:', result.stats);
 ```
 
+### Explicit Bridge Selection
+
+If you need to explicitly specify which bridge to use:
+
+```typescript
+import { RLM } from 'recursive-llm-ts';
+
+// Force use of bunpy (for Bun)
+const rlm = new RLM('gpt-4o-mini', {
+  max_iterations: 15,
+  api_key: process.env.OPENAI_API_KEY
+}, 'bunpy');
+
+// Or force use of pythonia (for Node.js)
+const rlm2 = new RLM('gpt-4o-mini', {}, 'pythonia');
+
+// Auto-detection (default)
+const rlm3 = new RLM('gpt-4o-mini', {}, 'auto');
+```
+
 
 ## API
 

package/dist/bridge-factory.d.ts

@@ -0,0 +1,6 @@
+import { PythonBridge } from './bridge-interface';
+export type BridgeType = 'pythonia' | 'bunpy' | 'auto';
+/**
+ * Create appropriate Python bridge based on runtime or explicit choice
+ */
+export declare function createBridge(bridgeType?: BridgeType): Promise<PythonBridge>;

package/dist/bridge-factory.js

@@ -0,0 +1,91 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createBridge = createBridge;
+/**
+ * Detect the current JavaScript runtime
+ */
+function detectRuntime() {
+    // Check for Bun
+    if (typeof globalThis.Bun !== 'undefined') {
+        return 'bun';
+    }
+    // Check for Node.js
+    if (typeof process !== 'undefined' && process.versions && process.versions.node) {
+        return 'node';
+    }
+    return 'unknown';
+}
+/**
+ * Create appropriate Python bridge based on runtime or explicit choice
+ */
+function createBridge() {
+    return __awaiter(this, arguments, void 0, function* (bridgeType = 'auto') {
+        let selectedBridge;
+        if (bridgeType === 'auto') {
+            const runtime = detectRuntime();
+            if (runtime === 'bun') {
+                selectedBridge = 'bunpy';
+            }
+            else if (runtime === 'node') {
+                selectedBridge = 'pythonia';
+            }
+            else {
+                throw new Error('Unable to detect runtime. Please explicitly specify bridge type.\n' +
+                    'Supported runtimes: Node.js (pythonia) or Bun (bunpy)');
+            }
+        }
+        else {
+            selectedBridge = bridgeType;
+        }
+        if (selectedBridge === 'bunpy') {
+            const { BunpyBridge } = yield Promise.resolve().then(() => __importStar(require('./bunpy-bridge')));
+            return new BunpyBridge();
+        }
+        else {
+            const { PythoniaBridge } = yield Promise.resolve().then(() => __importStar(require('./rlm-bridge')));
+            return new PythoniaBridge();
+        }
+    });
+}

package/dist/bridge-interface.d.ts

@@ -0,0 +1,22 @@
+export interface RLMStats {
+    llm_calls: number;
+    iterations: number;
+    depth: number;
+}
+export interface RLMResult {
+    result: string;
+    stats: RLMStats;
+}
+export interface RLMConfig {
+    recursive_model?: string;
+    api_base?: string;
+    api_key?: string;
+    max_depth?: number;
+    max_iterations?: number;
+    pythonia_timeout?: number;
+    [key: string]: any;
+}
+export interface PythonBridge {
+    completion(model: string, query: string, context: string, rlmConfig: RLMConfig): Promise<RLMResult>;
+    cleanup(): Promise<void>;
+}

package/dist/bridge-interface.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/bunpy-bridge.d.ts

@@ -0,0 +1,8 @@
+import { PythonBridge, RLMConfig, RLMResult } from './bridge-interface';
+export declare class BunpyBridge implements PythonBridge {
+    private rlmModule;
+    private python;
+    private ensureRLMModule;
+    completion(model: string, query: string, context: string, rlmConfig?: RLMConfig): Promise<RLMResult>;
+    cleanup(): Promise<void>;
+}

package/dist/bunpy-bridge.js

@@ -0,0 +1,122 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BunpyBridge = void 0;
+const path = __importStar(require("path"));
+class BunpyBridge {
+    constructor() {
+        this.rlmModule = null;
+        this.python = null;
+    }
+    ensureRLMModule() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.rlmModule)
+                return;
+            // Lazy load bunpy to avoid errors in Node.js environments
+            try {
+                // Dynamic import to avoid TypeScript errors when bunpy is not installed
+                const bunpy = yield new Function('return import("bunpy")')();
+                this.python = bunpy.python;
+            }
+            catch (error) {
+                throw new Error('bunpy is not installed. Install it with: bun add bunpy\n' +
+                    'Note: bunpy only works with Bun runtime, not Node.js');
+            }
+            // Import sys module to add path
+            const sys = this.python.import('sys');
+            const pythonSrcPath = path.join(__dirname, '..', 'recursive-llm', 'src');
+            sys.path.insert(0, pythonSrcPath);
+            // Import the rlm module
+            this.rlmModule = this.python.import('rlm');
+        });
+    }
+    completion(model_1, query_1, context_1) {
+        return __awaiter(this, arguments, void 0, function* (model, query, context, rlmConfig = {}) {
+            yield this.ensureRLMModule();
+            try {
+                // Remove pythonia_timeout from config (not applicable to bunpy)
+                const { pythonia_timeout } = rlmConfig, pythonConfig = __rest(rlmConfig, ["pythonia_timeout"]);
+                // Create RLM instance with config
+                const RLMClass = this.rlmModule.RLM;
+                const rlmInstance = RLMClass(model, pythonConfig);
+                // Call completion method
+                const result = rlmInstance.completion(query, context);
+                const stats = rlmInstance.stats;
+                // Convert Python stats dict to JS object
+                const statsObj = {
+                    llm_calls: stats.llm_calls,
+                    iterations: stats.iterations,
+                    depth: stats.depth
+                };
+                return {
+                    result: String(result),
+                    stats: statsObj
+                };
+            }
+            catch (error) {
+                console.error('[RLM_DEBUG] Error details:', error);
+                throw new Error(`RLM completion failed: ${error.message || error}`);
+            }
+        });
+    }
+    cleanup() {
+        return __awaiter(this, void 0, void 0, function* () {
+            // Bunpy doesn't need explicit cleanup like pythonia
+            this.rlmModule = null;
+            this.python = null;
+        });
+    }
+}
+exports.BunpyBridge = BunpyBridge;

package/dist/index.d.ts

@@ -1,2 +1,3 @@
 export { RLM } from './rlm';
-export { RLMBridge, RLMConfig, RLMResult, RLMStats, RLMError } from './rlm-bridge';
+export { RLMConfig, RLMResult, RLMStats } from './bridge-interface';
+export { BridgeType } from './bridge-factory';

package/dist/index.js

@@ -1,7 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RLMBridge = exports.RLM = void 0;
+exports.RLM = void 0;
 var rlm_1 = require("./rlm");
 Object.defineProperty(exports, "RLM", { enumerable: true, get: function () { return rlm_1.RLM; } });
-var rlm_bridge_1 = require("./rlm-bridge");
-Object.defineProperty(exports, "RLMBridge", { enumerable: true, get: function () { return rlm_bridge_1.RLMBridge; } });

package/dist/rlm-bridge.d.ts

@@ -1,25 +1,5 @@
-export interface RLMStats {
-    llm_calls: number;
-    iterations: number;
-    depth: number;
-}
-export interface RLMResult {
-    result: string;
-    stats: RLMStats;
-}
-export interface RLMError {
-    error: string;
-}
-export interface RLMConfig {
-    recursive_model?: string;
-    api_base?: string;
-    api_key?: string;
-    max_depth?: number;
-    max_iterations?: number;
-    pythonia_timeout?: number;
-    [key: string]: any;
-}
-export declare class RLMBridge {
+import { PythonBridge, RLMConfig, RLMResult } from './bridge-interface';
+export declare class PythoniaBridge implements PythonBridge {
     private rlmModule;
     private ensureRLMModule;
     completion(model: string, query: string, context: string, rlmConfig?: RLMConfig): Promise<RLMResult>;

package/dist/rlm-bridge.js

@@ -53,10 +53,10 @@ var __rest = (this && this.__rest) || function (s, e) {
     return t;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RLMBridge = void 0;
+exports.PythoniaBridge = void 0;
 const pythonia_1 = require("pythonia");
 const path = __importStar(require("path"));
-class RLMBridge {
+class PythoniaBridge {
     constructor() {
         this.rlmModule = null;
     }
@@ -100,6 +100,7 @@ class RLMBridge {
                 };
             }
             catch (error) {
+                console.error('[RLM_DEBUG] Error details:', error);
                 throw new Error(`RLM completion failed: ${error.message || error}`);
             }
         });
@@ -113,4 +114,4 @@ class RLMBridge {
         });
     }
 }
-exports.RLMBridge = RLMBridge;
+exports.PythoniaBridge = PythoniaBridge;

package/dist/rlm.d.ts

@@ -1,9 +1,12 @@
-import { RLMConfig, RLMResult } from './rlm-bridge';
+import { RLMConfig, RLMResult } from './bridge-interface';
+import { BridgeType } from './bridge-factory';
 export declare class RLM {
     private bridge;
     private model;
     private rlmConfig;
-    constructor(model: string, rlmConfig?: RLMConfig);
+    private bridgeType;
+    constructor(model: string, rlmConfig?: RLMConfig, bridgeType?: BridgeType);
+    private ensureBridge;
     completion(query: string, context: string): Promise<RLMResult>;
     cleanup(): Promise<void>;
 }

package/dist/rlm.js

@@ -1,18 +1,44 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.RLM = void 0;
-const rlm_bridge_1 = require("./rlm-bridge");
+const bridge_factory_1 = require("./bridge-factory");
 class RLM {
-    constructor(model, rlmConfig = {}) {
-        this.bridge = new rlm_bridge_1.RLMBridge();
+    constructor(model, rlmConfig = {}, bridgeType = 'auto') {
+        this.bridge = null;
         this.model = model;
         this.rlmConfig = rlmConfig;
+        this.bridgeType = bridgeType;
+    }
+    ensureBridge() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.bridge) {
+                this.bridge = yield (0, bridge_factory_1.createBridge)(this.bridgeType);
+            }
+            return this.bridge;
+        });
     }
     completion(query, context) {
-        return this.bridge.completion(this.model, query, context, this.rlmConfig);
+        return __awaiter(this, void 0, void 0, function* () {
+            const bridge = yield this.ensureBridge();
+            return bridge.completion(this.model, query, context, this.rlmConfig);
+        });
     }
     cleanup() {
-        return this.bridge.cleanup();
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.bridge) {
+                yield this.bridge.cleanup();
+                this.bridge = null;
+            }
+        });
     }
 }
 exports.RLM = RLM;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "recursive-llm-ts",
-  "version": "1.0.6",
+  "version": "2.0.0",
   "description": "TypeScript bridge for recursive-llm: Recursive Language Models for unbounded context processing",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -41,6 +41,14 @@
   "dependencies": {
     "pythonia": "^1.2.6"
   },
+  "peerDependencies": {
+    "bunpy": "^0.1.0"
+  },
+  "peerDependenciesMeta": {
+    "bunpy": {
+      "optional": true
+    }
+  },
   "devDependencies": {
     "@types/node": "^20.11.19",
     "dotenv": "^16.4.5",