@felipenmoura/laya-node 1.0.2 → 1.0.3

package/README.md

@@ -1,6 +1,6 @@
 # Laya Node Wrapper
 
-The Node.js wrapper for running the [Laya Text Classification Model](https://huggingface.co/convaiinnovations/laya).
+The Node.js wrapper for running the [Laya Text Classification Model](https://huggingface.co/convaiinnovations/laya), the multilingual, non-autoregressive System 1 decision model for AI (for artificial inteligence).
 
 Laya is a non-autoregressive decision model designed for text classification, email triage, and moderation. This package exposes Laya's API through an Express HTTP server, allowing you to easily integrate it into your Node.js applications while matching the exact Python API structure natively.
 

package/index.js

@@ -81,16 +81,27 @@ const uvicornExec = process.platform === 'win32'
   : path.join(venvPath, 'bin', 'uvicorn');
 
 // Check if venv exists, otherwise fallback to system python
-const useVenv = fs.existsSync(uvicornExec);
-const command = useVenv ? uvicornExec : 'uvicorn';
+const useVenv = fs.existsSync(pythonExec);
+const command = useVenv ? pythonExec : 'python3';
+const args = ['-m', 'uvicorn', 'server:app', '--port', PYTHON_PORT.toString()];
 
-console.log(`Starting python server via ${command}...`);
+console.log(`Starting python server via ${command} ${args.join(' ')}...`);
 
-const pythonProcess = spawn(command, ['server:app', '--port', PYTHON_PORT.toString()], {
+const pythonProcess = spawn(command, args, {
   cwd: __dirname,
   stdio: ['ignore', 'pipe', 'pipe']
 });
 
+pythonProcess.on('error', (err) => {
+  if (err.code === 'ENOENT') {
+    console.error(`\n\x1b[31m✖ Error: Could not find Python executable or uvicorn.\x1b[0m`);
+    console.error(`\x1b[33mPlease ensure you have run the setup or installed dependencies in the virtual environment.\x1b[0m\n`);
+  } else {
+    console.error(`\n\x1b[31m✖ Failed to start Python server:\x1b[0m ${err.message}\n`);
+  }
+  process.exit(1);
+});
+
 let warmedUp = false;
 async function warmupModel() {
   if (warmedUp) return;
@@ -175,7 +186,7 @@ app.use('/', createProxyMiddleware({
   }
 }));
 
-app.listen(PORT, () => {
+const server = app.listen(PORT, () => {
   console.log(`Laya Node wrapper is running on http://localhost:${PORT}`);
   if (SECRET_API_KEY) {
     console.log(`API Key protection is ENABLED. Please send "Authorization: Bearer <your-key>"`);
@@ -203,6 +214,16 @@ app.listen(PORT, () => {
   console.log(`\x1b[1mExample Request:\x1b[0m\n${curlExample}`);
 });
 
+server.on('error', (err) => {
+  if (err.code === 'EADDRINUSE') {
+    console.error(`\n\x1b[31m✖ Error: Port ${PORT} is already in use.\x1b[0m`);
+    console.error(`\x1b[33mPlease ensure no other process is using this port, or specify a different PORT environment variable.\x1b[0m\n`);
+    process.exit(1);
+  } else {
+    console.error(`\n\x1b[31m✖ Server error:\x1b[0m`, err.message, '\n');
+  }
+});
+
 // Clean up child process on exit
 process.on('SIGINT', () => {
   pythonProcess.kill('SIGINT');

package/install.js

@@ -117,11 +117,11 @@ async function runSetup() {
       }
       
       spinner.text = 'Installing dependencies (laya, fastapi, uvicorn)...';
-      const pipCommand = process.platform === 'win32' 
-        ? path.join(venvPath, 'Scripts', 'pip')
-        : path.join(venvPath, 'bin', 'pip');
+      const pythonCommand = process.platform === 'win32' 
+        ? path.join(venvPath, 'Scripts', 'python')
+        : path.join(venvPath, 'bin', 'python');
         
-      await execAsync(`${pipCommand} install fastapi uvicorn laya`, { cwd: __dirname });
+      await execAsync(`"${pythonCommand}" -m pip install fastapi uvicorn laya`, { cwd: __dirname });
       spinner.succeed('Setup complete!');
     } catch (error) {
       spinner.fail('Setup failed during python environment creation or pip install.');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@felipenmoura/laya-node",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Node wrapper for laya text classification",
   "license": "MIT",
   "author": "Felipe N. Moura",

package/server.py

@@ -24,9 +24,23 @@ def predict(req: PredictRequest):
         kwargs["model"] = req.model
         
     max_len = req.max_len if req.max_len is not None else int(os.getenv("MAX_LEN", 8192))
-    kwargs["max_len"] = max_len
-        
-    result = router.predict(req.state, req.questions, **kwargs)
+    
+    # Manually route to support max_len since Router.predict doesn't accept it
+    decision = router.route(req.state, req.questions, **kwargs)
+    agent = router.load(decision["model"])
+    
+    original_max_len = agent.cfg.get("max_len")
+    agent.cfg["max_len"] = max_len
+    
+    try:
+        result = agent.system_one(req.state, req.questions)
+        result["routing"] = dict(decision)
+    finally:
+        if original_max_len is not None:
+            agent.cfg["max_len"] = original_max_len
+        elif "max_len" in agent.cfg:
+            del agent.cfg["max_len"]
+            
     return result
 
 @app.get("/health")