@agentvalet/register 0.1.0 → 0.1.1

package/dist/index.js

@@ -80,8 +80,20 @@ import { fetch as fetch2 } from "undici";
 async function waitForApproval(apiUrl, token, timeoutMs, onTick) {
   const deadline = Date.now() + timeoutMs;
   const sseResult = await trySSE(apiUrl, token, Math.min(3e4, timeoutMs), onTick);
-  if (sseResult) return sseResult;
-  return poll(apiUrl, token, deadline, onTick);
+  if (sseResult) return enrichWithKey(apiUrl, token, sseResult);
+  const pollResult = await poll(apiUrl, token, deadline, onTick);
+  return enrichWithKey(apiUrl, token, pollResult);
+}
+async function enrichWithKey(apiUrl, token, result) {
+  if (result.status !== "approved") return result;
+  try {
+    const res = await fetch2(`${apiUrl}/v1/register/claim-key/${token}`);
+    if (!res.ok) return result;
+    const json = await res.json();
+    return { ...result, privateKey: json.private_key ?? "" };
+  } catch {
+    return result;
+  }
 }
 async function trySSE(apiUrl, token, windowMs, onTick) {
   const controller = new AbortController();
@@ -163,7 +175,7 @@ async function saveOutput(opts) {
   const envPath = resolve(opts.envFilePath);
   await mkdir(dirname(keyPath), { recursive: true });
   await mkdir(dirname(envPath), { recursive: true });
-  await writeFile(keyPath, "", { mode: 384 });
+  await writeFile(keyPath, opts.privateKey, { mode: 384 });
   const envContent = [
     `# AgentValet \u2014 generated by agentvalet-register`,
     `AGENTVALET_AGENT_ID=${opts.agentId}`,
@@ -275,12 +287,13 @@ Approve at: https://agentvalet.ai/registrations
     }
     process.exit(2);
   }
-  const { agentId } = result;
+  const { agentId, privateKey } = result;
   let keyPath;
   let envPath;
   try {
     ({ keyPath, envPath } = await saveOutput({
       agentId,
+      privateKey: privateKey ?? "",
       apiUrl: opts.apiUrl,
       keyFilePath: opts.keyPath,
       envFilePath: opts.output
@@ -290,12 +303,17 @@ Approve at: https://agentvalet.ai/registrations
     console.error(isJson ? JSON.stringify({ error: msg }) : chalk2.red(`Failed to save output: ${msg}`));
     process.exit(3);
   }
+  const keyWritten = !!privateKey;
   if (isJson) {
-    console.log(JSON.stringify({ status: "approved", agentId, keyPath, envPath }));
+    console.log(JSON.stringify({ status: "approved", agentId, keyPath: keyWritten ? keyPath : null, envPath }));
   } else {
     console.log(chalk2.green(`
 Approved! Agent ID: ${agentId}`));
-    console.log(chalk2.gray(`Key:        ${keyPath}`));
+    if (keyWritten) {
+      console.log(chalk2.gray(`Key:        ${keyPath}`));
+    } else {
+      console.log(chalk2.yellow(`Key:        not written \u2014 copy it from the dashboard and paste into ${keyPath}`));
+    }
     console.log(chalk2.gray(`Env file:   ${envPath}`));
     console.log(`
 Source the env file before calling any tool:`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentvalet/register",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Self-register AI agents with AgentValet",
   "type": "module",
   "bin": {