osai-agent 4.2.25 → 4.2.27

package/package.json

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

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,22 @@ export const enableUser = async ({ server, email }) => {
 };
 
 export const generateProKeys = async ({ server, count, type }) => {
-  const spinner = ora('Generating Pro keys...').start();
-  const data = await adminFetch(server, '/generate', 'POST', { type, count }, '/pro');
-  spinner.stop();
-  console.log(chalk.green(`\n  ${data.count} Pro keys generated (${data.type}, ${data.duration_days} days):\n`));
+  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', secret);
+  done = true;
+  clearTimeout(coldStartWarn);
+  spinner.succeed(`${data.count} Pro keys generated (${data.type}, ${data.duration_days} days)`);
   for (const key of data.keys) {
-    console.log(`    ${chalk.hex('#ff9e64')(key)}`);
+    console.log(`  ${chalk.hex('#ff9e64')(key)}`);
   }
   console.log();
 };