osai-agent 4.2.26 → 4.2.28

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "osai-agent",
-  "version": "4.2.26",
+  "version": "4.2.28",
   "type": "module",
   "description": "OS AI Agent - YOUR AI AGENT",
   "main": "src/index.js",

package/src/agent/loop/websocket.js

@@ -39,7 +39,7 @@ export default {
                  if (msg.type === 'REGISTERED') { clearTimeout(connectTimeout); this.ws = ws; resolve(msg); }
                  else if (msg.type === 'COMMANDS') { this.pendingCommands = msg.commands; this.currentTaskId = msg.taskId; }
                  else if (msg.type === 'CANCEL_ACK') { logger.info('Cancellation acknowledged by server'); }
-                 else if (msg.type === 'ERROR') { logger.warn('WebSocket error message', { message: msg.message }); }
+                  else if (msg.type === 'ERROR') { clearTimeout(connectTimeout); ws.close(); reject(new Error(msg.message)); }
                  else if (msg.type === 'CANCEL_SUBAGENT') {
                    if (this._subagentActive && this._subagentLoop) {
                      this._subagentLoop.cancel();

package/src/agent/react-loop.js

@@ -274,6 +274,10 @@ export class AgentLoop {
       await this._connectWebSocket();
     } catch (err) {
       this.onConnectionChange(false);
+      if (err.message.toLowerCase().includes('disabled')) {
+        console.log(chalk.red(`\n  ${err.message}\n`));
+        process.exit(1);
+      }
       logger.debug(`WebSocket connection failed: ${err.message}`);
       logger.warn('Server unavailable, running in offline mode');
       this.ws = null;

package/src/commands/admin.js

@@ -12,12 +12,12 @@ const promptSecret = () => new Promise((resolve) => {
   });
 });
 
-const adminFetch = async (serverArg, path, method = 'GET', body = null, pathPrefix = '/auth') => {
+const adminFetch = async (serverArg, path, method = 'GET', body = null, pathPrefix = '/auth', secret = null) => {
   const config = new Conf({ projectName: 'osai-agent' });
   const serverUrl = serverArg || config.get('server') || DEFAULT_SERVER_URL;
   const httpUrl = toHttpUrl(serverUrl);
 
-  const secret = await promptSecret();
+  if (!secret) secret = await promptSecret();
   if (!secret) {
     console.log(chalk.red('Admin secret is required.'));
     process.exit(1);
@@ -82,12 +82,17 @@ export const enableUser = async ({ server, email }) => {
 };
 
 export const generateProKeys = async ({ server, count, type }) => {
+  const secret = await promptSecret();
+  if (!secret) {
+    console.log(chalk.red('Admin secret is required.'));
+    process.exit(1);
+  }
   const spinner = ora({ text: 'Generating Pro keys...', spinner: 'bouncingBar' }).start();
   let done = false;
   const coldStartWarn = setTimeout(() => {
     if (!done) spinner.text = 'Server is waking up (HF Spaces cold start), please wait...';
   }, 15000);
-  const data = await adminFetch(server, '/generate', 'POST', { type, count }, '/pro');
+  const data = await adminFetch(server, '/generate', 'POST', { type, count }, '/pro', secret);
   done = true;
   clearTimeout(coldStartWarn);
   spinner.succeed(`${data.count} Pro keys generated (${data.type}, ${data.duration_days} days)`);