pairai 0.2.1 → 0.2.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pairai",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "pairai CLI — connect AI agents to collaborate via the pairai hub",
   "type": "module",
   "license": "MIT",

package/pairai.ts

@@ -3,7 +3,7 @@
  * pairai CLI — connect AI agents via the pairai hub
  *
  * Commands:
- *   npx pairai setup "Agent Name" [--hub URL] [--provider claude|gemini] [--global]
+ *   npx pairai setup "Agent Name" [--hub URL] [--provider claude|gemini] [--global] [--force]
  *   npx pairai serve [--provider claude|gemini]
  *   npx pairai upgrade     — update to latest version (preserves keys and config)
  *   npx pairai version     — show current version
@@ -93,16 +93,41 @@ if (command === "setup") {
   const hubIdx = rest.indexOf("--hub");
   const hubUrl = hubIdx !== -1 ? rest.splice(hubIdx, 2)[1] : "https://pairai.pro";
   const providerIdx = rest.indexOf("--provider");
-  const provider = providerIdx !== -1 ? rest.splice(providerIdx, 2)[1] as "claude" | "gemini" : detectProvider();
+  const providerArg = providerIdx !== -1 ? rest.splice(providerIdx, 2)[1] : undefined;
+  if (providerArg && providerArg !== "claude" && providerArg !== "gemini") {
+    console.error(`  Unknown provider "${providerArg}". Must be "claude" or "gemini".`);
+    process.exit(1);
+  }
+  const provider = providerArg ?? detectProvider();
   const globalIdx = rest.indexOf("--global");
   const useGlobal = globalIdx !== -1 ? (rest.splice(globalIdx, 1), true) : false;
   const agentName = rest.find((a) => !a.startsWith("--"));
 
+  const forceIdx = rest.indexOf("--force");
+  const useForce = forceIdx !== -1 ? (rest.splice(forceIdx, 1), true) : false;
   if (!agentName) {
-    console.error('Usage: npx pairai setup "Agent Name" [--hub URL] [--provider claude|gemini] [--global]');
+    console.error('Usage: npx pairai setup "Agent Name" [--hub URL] [--provider claude|gemini] [--global] [--force]');
     process.exit(1);
   }
 
+  // Check for existing config to avoid accidental overwrites
+  const existingConfigPath = provider === "gemini"
+    ? join(useGlobal ? join(homedir(), ".gemini") : join(process.cwd(), ".gemini"), "settings.json")
+    : join(process.cwd(), ".mcp.json");
+  const mcpKey = provider === "gemini" ? "pairai" : "pairai-channel";
+  if (!useForce && existsSync(existingConfigPath)) {
+    try {
+      const existing = JSON.parse(readFileSync(existingConfigPath, "utf-8"));
+      const servers = existing.mcpServers ?? {};
+      if (servers[mcpKey]) {
+        console.error(`\n  pairai is already configured in ${existingConfigPath}`);
+        console.error(`  Running setup again would overwrite the existing API key and config.`);
+        console.error(`\n  To force a fresh setup, run: npx pairai setup "${agentName}" --force\n`);
+        process.exit(1);
+      }
+    } catch {}
+  }
+
   console.log(`\n  Registering "${agentName}" on ${hubUrl}...\n`);
 
   console.log("  Generating RSA-4096 keypair...");
@@ -133,18 +158,22 @@ if (command === "setup") {
   const keyPath = join(keyDir, `${id}.pem`);
   writeFileSync(keyPath, privateKey, { mode: 0o600 });
   console.log(`  Private key: ${keyPath}`);
+  const lines = [
+    "BACK UP YOUR PRIVATE KEY",
+    "",
+    keyPath,
+    "",
+    "This key is stored only on your machine.",
+    "The hub never sees it. If lost, you must re-register",
+    "and re-pair — all encrypted history becomes unreadable.",
+    "",
+    "Copy it to a password manager or secure backup now.",
+  ];
+  const w = Math.max(...lines.map((l) => l.length)) + 2;
   console.log();
-  console.log(`  ┌──────────────────────────────────────────────────────────┐`);
-  console.log(`  │  BACK UP YOUR PRIVATE KEY                               │`);
-  console.log(`  │                                                          │`);
-  console.log(`  │  ${keyPath.padEnd(56)}│`);
-  console.log(`  │                                                          │`);
-  console.log(`  │  This key is stored only on your machine.                │`);
-  console.log(`  │  The hub never sees it. If lost, you must re-register    │`);
-  console.log(`  │  and re-pair — all encrypted history becomes unreadable. │`);
-  console.log(`  │                                                          │`);
-  console.log(`  │  Copy it to a password manager or secure backup now.     │`);
-  console.log(`  └──────────────────────────────────────────────────────────┘`);
+  console.log(`  ┌${"─".repeat(w + 2)}┐`);
+  for (const l of lines) console.log(`  │  ${l.padEnd(w)}│`);
+  console.log(`  └${"─".repeat(w + 2)}┘`);
   console.log();
 
   if (provider === "gemini") {
@@ -213,7 +242,7 @@ if (command === "setup") {
 if (command !== "serve") {
   console.error(`pairai v${VERSION}\n`);
   console.error("Usage:");
-  console.error('  npx pairai setup "Agent Name" [--hub URL] [--provider claude|gemini] [--global]');
+  console.error('  npx pairai setup "Agent Name" [--hub URL] [--provider claude|gemini] [--global] [--force]');
   console.error("  npx pairai serve [--provider claude|gemini]");
   console.error("  npx pairai upgrade        — update to latest version");
   console.error("  npx pairai version        — show current version");
@@ -226,7 +255,12 @@ const { ListToolsRequestSchema, CallToolRequestSchema } = await import("@modelco
 
 const serveArgs = args.slice(1);
 const serveProviderIdx = serveArgs.indexOf("--provider");
-const serveProvider = serveProviderIdx !== -1 ? serveArgs[serveProviderIdx + 1] : "claude";
+const serveProviderArg = serveProviderIdx !== -1 ? serveArgs[serveProviderIdx + 1] : undefined;
+if (serveProviderArg && serveProviderArg !== "claude" && serveProviderArg !== "gemini") {
+  console.error(`  Unknown provider "${serveProviderArg}". Must be "claude" or "gemini".`);
+  process.exit(1);
+}
+const serveProvider = serveProviderArg ?? "claude";
 
 const HUB_URL = process.env.PAIRAI_HUB_URL ?? process.env.PAIRAI_URL ?? "https://pairai.pro";
 const API_KEY = process.env.PAIRAI_AGENT_CRED ?? process.env.PAIRAI_API_KEY;
@@ -402,6 +436,11 @@ mcp.setRequestHandler(ListToolsRequestSchema, async () => ({
         required: ["task_id", "status"],
       },
     },
+    {
+      name: "pairai_get_profile",
+      description: "Get your own agent profile — name, ID, description, capabilities.",
+      inputSchema: { type: "object" as const, properties: {} },
+    },
     {
       name: "pairai_list_connections",
       description: "List agents you are connected with.",
@@ -436,6 +475,66 @@ mcp.setRequestHandler(ListToolsRequestSchema, async () => ({
         required: ["code"],
       },
     },
+    {
+      name: "pairai_update_profile",
+      description: "Update your agent's profile — name, description, capabilities, and metadata. Returns the updated profile.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          name: { type: "string", description: "Display name (1-64 chars)" },
+          description: { type: "string", description: "What this agent does (max 500 chars)" },
+          capabilities: { type: "array", items: { type: "string" }, description: "List of capabilities, e.g. ['scheduling', 'code-review']" },
+          metadata: { type: "object", description: "Arbitrary JSON metadata (max 4KB)" },
+        },
+      },
+    },
+    {
+      name: "pairai_set_alias",
+      description: "Set a local alias for a connected agent. Only you see this alias. Set to null to clear.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          connection_id: { type: "string", description: "Connection ID" },
+          alias: { type: ["string", "null"], description: "Local alias, or null to clear" },
+        },
+        required: ["connection_id"],
+      },
+    },
+    {
+      name: "pairai_list_tasks",
+      description: "List all tasks you are involved in (as initiator or target).",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          status: { type: "string", enum: ["submitted", "working", "input-required", "completed", "failed", "cancelled"], description: "Filter by status" },
+        },
+      },
+    },
+    {
+      name: "pairai_get_task",
+      description: "Get full details of a task including all messages.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          task_id: { type: "string", description: "Task ID" },
+        },
+        required: ["task_id"],
+      },
+    },
+    {
+      name: "pairai_upload_file",
+      description: "Upload a file to a task. Provide base64-encoded content.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          task_id: { type: "string", description: "Task ID" },
+          filename: { type: "string", description: "Original filename, e.g. photo.png" },
+          mime_type: { type: "string", description: "MIME type, e.g. image/png" },
+          base64_content: { type: "string", description: "Base64-encoded file content" },
+        },
+        required: ["task_id", "filename", "mime_type", "base64_content"],
+      },
+    },
     {
       name: "pairai_create_encrypted_task",
       description: "Create an encrypted task. Title and description are encrypted — the hub cannot read them.",
@@ -496,6 +595,11 @@ mcp.setRequestHandler(CallToolRequestSchema, async (req) => {
     return { content: [{ type: "text" as const, text: `Status → ${args.status}` }] };
   }
 
+  if (name === "pairai_get_profile") {
+    const data = await hubGet("/agents/me");
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
   if (name === "pairai_list_connections") {
     const data = await hubGet("/connections");
     return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
@@ -550,6 +654,43 @@ mcp.setRequestHandler(CallToolRequestSchema, async (req) => {
     return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
   }
 
+  if (name === "pairai_update_profile") {
+    const body: Record<string, unknown> = {};
+    if (args.name !== undefined) body.name = args.name;
+    if (args.description !== undefined) body.description = args.description;
+    if (args.capabilities !== undefined) body.capabilities = args.capabilities;
+    if (args.metadata !== undefined) body.metadata = args.metadata;
+    const data = await hubPatch("/agents/me", body);
+    return { content: [{ type: "text" as const, text: "Profile updated.\n" + JSON.stringify(data, null, 2) }] };
+  }
+
+  if (name === "pairai_set_alias") {
+    const { connection_id, alias } = args as { connection_id: string; alias?: string | null };
+    const data = await hubPatch(`/connections/${connection_id}`, { alias: alias ?? null });
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
+  if (name === "pairai_list_tasks") {
+    const data = (await hubGet("/tasks")) as Array<{ status: string }>;
+    const filtered = args.status ? data.filter((t) => t.status === args.status) : data;
+    return { content: [{ type: "text" as const, text: JSON.stringify(filtered, null, 2) }] };
+  }
+
+  if (name === "pairai_get_task") {
+    const data = await hubGet(`/tasks/${args.task_id}`);
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
+  if (name === "pairai_upload_file") {
+    const { task_id, filename, mime_type, base64_content } = args as {
+      task_id: string; filename: string; mime_type: string; base64_content: string;
+    };
+    const data = await hubPost(`/tasks/${task_id}/files/json`, {
+      filename, mimeType: mime_type, base64Content: base64_content,
+    });
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
   if (name === "pairai_create_encrypted_task") {
     if (!PRIVATE_KEY)
       return { content: [{ type: "text" as const, text: "No private key configured. Re-run setup." }] };
@@ -701,6 +842,7 @@ async function poll() {
         }>;
       };
 
+      if (!full?.messages) continue;
       for (const msg of full.messages.slice(-unread.count)) {
         const key = `msg:${msg.id}`;
         if (seenMessages.has(key)) continue;