@exagent/agent 0.2.1 → 0.3.1

package/dist/cli.js

@@ -1,57 +1,379 @@
 #!/usr/bin/env node
 import {
   AgentRuntime,
+  encryptSecretPayload,
+  getDefaultSecureStorePath,
   listTemplates,
   loadConfig,
+  readConfigFile,
+  writeConfigFile,
   writeSampleConfig
-} from "./chunk-B4VHIITU.js";
+} from "./chunk-VDK4XPAC.js";
 
 // src/cli.ts
 import { Command } from "commander";
+
+// src/setup.ts
+import { chmodSync, existsSync, mkdirSync, writeFileSync } from "fs";
+import { homedir } from "os";
+import { dirname, resolve } from "path";
+import { generatePrivateKey, privateKeyToAccount } from "viem/accounts";
+import * as clack from "@clack/prompts";
+
+// src/ui.ts
+import figlet from "figlet";
+import gradient from "gradient-string";
+import boxen from "boxen";
+import pc from "picocolors";
+import { createRequire } from "module";
+var brandGradient = gradient(["#3B82F6", "#6366F1", "#7C3AED"]);
+var accentGradient = gradient(["#22D3EE", "#3B82F6"]);
+function getVersion() {
+  try {
+    const require2 = createRequire(import.meta.url);
+    const pkg = require2("../package.json");
+    return pkg.version || "0.0.0";
+  } catch {
+    return "0.0.0";
+  }
+}
+function printBanner() {
+  const art = figlet.textSync("EXAGENT", {
+    font: "Small",
+    horizontalLayout: "default"
+  });
+  console.log();
+  console.log(brandGradient(art));
+  console.log(pc.dim(`  v${getVersion()}`));
+  console.log();
+}
+function printSuccess(title, lines) {
+  const body = [
+    "",
+    pc.bold(pc.white(title)),
+    "",
+    ...lines.map((l) => `  ${l}`),
+    ""
+  ].join("\n");
+  console.log();
+  console.log(boxen(body, {
+    padding: { top: 0, bottom: 0, left: 2, right: 2 },
+    borderColor: "#3B82F6",
+    borderStyle: "round",
+    dimBorder: false
+  }));
+  console.log();
+}
+function printStep(step, total, label) {
+  console.log();
+  console.log(accentGradient(`  Step ${step} of ${total}`) + pc.dim(` \u2014 ${label}`));
+}
+function printDone(message) {
+  console.log(`  ${pc.green("\u2713")} ${message}`);
+}
+function printInfo(message) {
+  console.log(`  ${pc.dim("\u2502")} ${message}`);
+}
+function printError(message) {
+  console.log(`  ${pc.red("\u2717")} ${message}`);
+}
+
+// src/setup.ts
+function expandHomeDir(path) {
+  if (!path.startsWith("~/")) return path;
+  return resolve(homedir(), path.slice(2));
+}
+function cancelled() {
+  clack.cancel("Setup cancelled.");
+  process.exit(0);
+}
+async function consumeBootstrapPackage(config) {
+  if (!config.secrets?.bootstrapToken) {
+    return { apiToken: "" };
+  }
+  const apiHost = new URL(config.apiUrl).host;
+  printInfo(`Connecting to ${pc.cyan(apiHost)}...`);
+  const res = await fetch(`${config.apiUrl}/v1/agents/bootstrap/consume`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({
+      agentId: config.agentId,
+      token: config.secrets.bootstrapToken
+    })
+  });
+  if (!res.ok) {
+    const body = await res.text();
+    throw new Error(`Failed to consume bootstrap package: ${body}`);
+  }
+  const data = await res.json();
+  return data.payload;
+}
+async function setupWallet(config) {
+  if (config.wallet?.privateKey) {
+    const account = privateKeyToAccount(config.wallet.privateKey);
+    printDone(`Using existing wallet: ${pc.dim(account.address)}`);
+    return config.wallet.privateKey;
+  }
+  const method = await clack.select({
+    message: "How would you like to set up your wallet?",
+    options: [
+      { value: "generate", label: "Generate new wallet locally", hint: "recommended" },
+      { value: "import", label: "Import existing private key" }
+    ]
+  });
+  if (clack.isCancel(method)) cancelled();
+  if (method === "generate") {
+    const privateKey2 = generatePrivateKey();
+    const address2 = privateKeyToAccount(privateKey2).address;
+    printDone(`Wallet created: ${pc.dim(address2)}`);
+    return privateKey2;
+  }
+  const privateKey = await clack.password({
+    message: "Wallet private key (0x...):",
+    validate: (val) => {
+      if (!/^0x[a-fA-F0-9]{64}$/.test(val)) {
+        return "Invalid private key. Expected a 32-byte hex string prefixed with 0x.";
+      }
+    }
+  });
+  if (clack.isCancel(privateKey)) cancelled();
+  const address = privateKeyToAccount(privateKey).address;
+  printDone(`Wallet imported: ${pc.dim(address)}`);
+  return privateKey;
+}
+var LLM_PROVIDERS = ["openai", "anthropic", "google", "deepseek", "mistral", "groq", "together", "ollama"];
+async function setupLlm(config, bootstrapPayload) {
+  let provider = config.llm?.provider || bootstrapPayload.llm?.provider;
+  if (provider) {
+    printInfo(`Provider: ${pc.cyan(provider)} ${pc.dim("(from dashboard)")}`);
+  } else {
+    const selected = await clack.select({
+      message: "LLM provider:",
+      options: LLM_PROVIDERS.map((p) => ({ value: p, label: p }))
+    });
+    if (clack.isCancel(selected)) cancelled();
+    provider = selected;
+  }
+  let model = config.llm?.model || bootstrapPayload.llm?.model;
+  if (model) {
+    printInfo(`Model: ${pc.cyan(model)} ${pc.dim("(from dashboard)")}`);
+  } else {
+    const entered = await clack.text({
+      message: "LLM model:",
+      placeholder: "gpt-4o",
+      validate: (val) => {
+        if (!val.trim()) return "Model name is required.";
+      }
+    });
+    if (clack.isCancel(entered)) cancelled();
+    model = entered;
+  }
+  let apiKey;
+  if (bootstrapPayload.llm?.apiKey) {
+    const useBootstrap = await clack.confirm({
+      message: "LLM API key received from dashboard. Use it?",
+      initialValue: true
+    });
+    if (clack.isCancel(useBootstrap)) cancelled();
+    if (useBootstrap) {
+      apiKey = bootstrapPayload.llm.apiKey;
+    }
+  }
+  if (!apiKey) {
+    apiKey = config.llm?.apiKey;
+  }
+  if (!apiKey) {
+    const entered = await clack.password({
+      message: "LLM API key:",
+      validate: (val) => {
+        if (!val.trim()) return "API key is required.";
+      }
+    });
+    if (clack.isCancel(entered)) cancelled();
+    apiKey = entered;
+  }
+  printDone("LLM configured");
+  return { provider, model, apiKey };
+}
+async function setupEncryption() {
+  printInfo(`Secrets encrypted with ${pc.cyan("AES-256-GCM")} (${pc.cyan("scrypt")} KDF)`);
+  printInfo("The password never leaves this machine.");
+  console.log();
+  const password2 = await clack.password({
+    message: "Choose a device password (12+ characters):",
+    validate: (val) => {
+      if (val.length < 12) return "Password must be at least 12 characters.";
+    }
+  });
+  if (clack.isCancel(password2)) cancelled();
+  const confirm2 = await clack.password({
+    message: "Confirm password:",
+    validate: (val) => {
+      if (val !== password2) return "Passwords do not match.";
+    }
+  });
+  if (clack.isCancel(confirm2)) cancelled();
+  return password2;
+}
+function writeSecureStore(path, secrets, password2) {
+  const secureStorePath = expandHomeDir(path);
+  const encrypted = encryptSecretPayload(secrets, password2);
+  const dir = dirname(secureStorePath);
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true, mode: 448 });
+  }
+  writeFileSync(secureStorePath, JSON.stringify(encrypted, null, 2), { mode: 384 });
+  try {
+    chmodSync(secureStorePath, 384);
+  } catch {
+  }
+  return secureStorePath;
+}
+async function promptSecretPassword(question = "Device password:") {
+  const password2 = await clack.password({ message: question });
+  if (clack.isCancel(password2)) cancelled();
+  return password2;
+}
+async function ensureLocalSetup(configPath) {
+  const config = readConfigFile(configPath);
+  const existingSecureStorePath = config.secrets?.secureStorePath ? expandHomeDir(config.secrets.secureStorePath) : null;
+  if (existingSecureStorePath && !config.secrets?.bootstrapToken && existsSync(existingSecureStorePath) && !config.apiToken && !config.wallet?.privateKey && !config.llm.apiKey) {
+    return;
+  }
+  printBanner();
+  clack.intro(pc.bold("Agent Setup"));
+  printStep(1, 4, "Bootstrap package");
+  const bootstrapPayload = await consumeBootstrapPackage(config);
+  if (config.secrets?.bootstrapToken) {
+    printDone("Bootstrap package consumed");
+    if (bootstrapPayload.llm?.provider) {
+      printInfo(`LLM config received: ${pc.cyan(bootstrapPayload.llm.provider)}${bootstrapPayload.llm.model ? ` / ${pc.cyan(bootstrapPayload.llm.model)}` : ""}`);
+    }
+  } else {
+    printInfo("No bootstrap token \u2014 manual configuration");
+  }
+  printStep(2, 4, "Wallet setup");
+  const walletPrivateKey = await setupWallet(config);
+  printStep(3, 4, "LLM configuration");
+  const llm = await setupLlm(config, bootstrapPayload);
+  printStep(4, 4, "Device encryption");
+  const password2 = await setupEncryption();
+  const secrets = {
+    apiToken: bootstrapPayload.apiToken || config.apiToken || "",
+    walletPrivateKey,
+    llmApiKey: llm.apiKey
+  };
+  if (!secrets.apiToken) {
+    const token = await clack.password({
+      message: "Agent relay token:",
+      validate: (val) => {
+        if (!val.trim()) return "Relay token is required.";
+      }
+    });
+    if (clack.isCancel(token)) cancelled();
+    secrets.apiToken = token;
+  }
+  const nextConfig = structuredClone(config);
+  nextConfig.llm = {
+    ...nextConfig.llm,
+    provider: llm.provider,
+    model: llm.model
+  };
+  delete nextConfig.apiToken;
+  delete nextConfig.wallet;
+  delete nextConfig.llm.apiKey;
+  const secureStorePath = writeSecureStore(
+    nextConfig.secrets?.secureStorePath || getDefaultSecureStorePath(nextConfig.agentId),
+    secrets,
+    password2
+  );
+  nextConfig.secrets = { secureStorePath };
+  writeConfigFile(configPath, nextConfig);
+  printDone(`Encrypted store: ${pc.dim(secureStorePath)}`);
+  clack.outro(pc.green("Setup complete"));
+  printSuccess("Ready", [
+    `${pc.cyan("npx exagent run")}      Start trading`,
+    `${pc.cyan("npx exagent status")}   Check connection`,
+    "",
+    `${pc.dim("Dashboard:")}  ${pc.cyan("https://exagent.io")}`
+  ]);
+}
+
+// src/cli.ts
 var program = new Command();
-program.name("exagent").description("Exagent \u2014 LLM trading agent runtime").version("0.1.0");
+program.name("exagent").description("Exagent \u2014 LLM trading agent runtime").version("0.3.0");
 program.command("init").description("Create a sample agent configuration file").option("--agent-id <id>", "Agent ID (from dashboard)", "my-agent").option("--api-url <url>", "API server URL", "http://localhost:3002").option("--config <path>", "Config file path", "agent-config.json").action((opts) => {
+  printBanner();
   writeSampleConfig(opts.agentId, opts.apiUrl, opts.config);
-  console.log(`Created ${opts.config}`);
-  console.log("Edit the file with your API token, LLM key, and strategy settings.");
-  console.log("Then run: exagent run");
+  printDone(`Created ${pc.cyan(opts.config)}`);
+  console.log();
+  console.log(`  Edit only the public strategy/venue/risk settings.`);
+  console.log(`  Then run: ${pc.cyan(`exagent setup --config ${opts.config}`)}`);
+  console.log();
+});
+program.command("setup").description("Run first-time secure local setup for agent secrets").option("--config <path>", "Config file path", "agent-config.json").action(async (opts) => {
+  try {
+    await ensureLocalSetup(opts.config);
+  } catch (err) {
+    printError(err.message);
+    process.exit(1);
+  }
 });
 program.command("run").description("Start the agent").option("--config <path>", "Config file path", "agent-config.json").action(async (opts) => {
   try {
-    const config = loadConfig(opts.config);
+    await ensureLocalSetup(opts.config);
+    printBanner();
+    const config = await loadConfig(opts.config, {
+      getSecretPassword: async () => promptSecretPassword()
+    });
+    printDone(`Agent ${pc.cyan(config.agentId)} starting...`);
+    console.log();
     const runtime = new AgentRuntime(config);
     await runtime.start();
     await new Promise(() => {
     });
   } catch (err) {
-    console.error("Failed to start agent:", err.message);
+    printError(err.message);
     process.exit(1);
   }
 });
 program.command("templates").description("List available strategy templates").action(() => {
+  printBanner();
   const templates = listTemplates();
-  console.log("\nAvailable strategy templates:\n");
+  console.log(pc.bold("  Strategy Templates"));
+  console.log();
   for (const t of templates) {
-    console.log(`  ${t.id}`);
-    console.log(`    ${t.name} \u2014 ${t.description}`);
-    console.log(`    Risk: ${t.riskLevel} | Venues: ${t.venues.join(", ") || "any"}`);
+    console.log(`  ${pc.cyan(t.id)}`);
+    console.log(`  ${t.name} \u2014 ${pc.dim(t.description)}`);
+    console.log(`  ${pc.dim(`Risk: ${t.riskLevel} | Venues: ${t.venues.join(", ") || "any"}`)}`);
     console.log();
   }
 });
 program.command("status").description("Check agent status").option("--config <path>", "Config file path", "agent-config.json").action(async (opts) => {
   try {
-    const config = loadConfig(opts.config);
+    await ensureLocalSetup(opts.config);
+    printBanner();
+    const config = await loadConfig(opts.config, {
+      getSecretPassword: async () => promptSecretPassword()
+    });
     const res = await fetch(`${config.apiUrl}/v1/agents/${config.agentId}`, {
       headers: { Authorization: `Bearer ${config.apiToken}` }
     });
     if (!res.ok) {
-      console.error(`API error: ${res.status}`);
+      printError(`API error: ${res.status}`);
       process.exit(1);
     }
     const agent = await res.json();
-    console.log(JSON.stringify(agent, null, 2));
+    const name = agent.name || config.agentId;
+    const status = agent.status || "unknown";
+    const statusColor = status === "online" ? pc.green : status === "error" ? pc.red : pc.yellow;
+    printSuccess(name, [
+      `${pc.dim("Status:")}  ${statusColor(status)}`,
+      `${pc.dim("Agent:")}   ${pc.cyan(config.agentId)}`,
+      `${pc.dim("API:")}     ${pc.dim(config.apiUrl)}`
+    ]);
   } catch (err) {
-    console.error("Failed to check status:", err.message);
+    printError(err.message);
     process.exit(1);
   }
 });

package/dist/index.d.ts

@@ -19,7 +19,14 @@ interface RuntimeConfig {
     };
     strategy: {
         file?: string;
+        code?: string;
         template?: string;
+        venues?: string[];
+        prompt?: {
+            name?: string;
+            systemPrompt: string;
+            venues?: string[];
+        };
     };
     trading: {
         mode: 'live' | 'paper';
@@ -78,7 +85,10 @@ interface RuntimeConfig {
         json?: boolean;
     };
 }
-declare function loadConfig(path?: string): RuntimeConfig;
+interface LoadConfigOptions {
+    getSecretPassword?: () => Promise<string>;
+}
+declare function loadConfig(path?: string, options?: LoadConfigOptions): Promise<RuntimeConfig>;
 declare function generateSampleConfig(agentId: string, apiUrl: string): string;
 declare function writeSampleConfig(agentId: string, apiUrl: string, path?: string): void;
 
@@ -116,6 +126,18 @@ declare class AgentRuntime {
     constructor(config: RuntimeConfig);
     start(): Promise<void>;
     stop(): Promise<void>;
+    private configureLLMAdapter;
+    private teardownVenues;
+    private waitForCycleCompletion;
+    private getConfiguredSpotVenue;
+    private normalizeVenueForExecution;
+    private isVenueConfigured;
+    private canExecuteVenue;
+    private getPreferredExecutionVenues;
+    private buildRuntimeVenuesFromSelection;
+    private buildStrategyFallbackPrompt;
+    private extractStrategyConfigFromAgentConfig;
+    private applyExecutionMode;
     private initializeVenues;
     private startTrading;
     private stopTrading;
@@ -407,7 +429,13 @@ declare function createLLMAdapter(config: LLMConfig): LLMAdapter;
 
 declare function loadStrategy(config: {
     file?: string;
+    code?: string;
     template?: string;
+    prompt?: {
+        name?: string;
+        systemPrompt: string;
+        venues?: string[];
+    };
 }): Promise<StrategyFunction>;
 
 declare function getTemplate(id: string): StrategyTemplate | undefined;

package/dist/index.js

@@ -43,7 +43,7 @@ import {
   loadConfig,
   loadStrategy,
   writeSampleConfig
-} from "./chunk-B4VHIITU.js";
+} from "./chunk-VDK4XPAC.js";
 export {
   AcrossAdapter,
   AerodromeAdapter,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exagent/agent",
-  "version": "0.2.1",
+  "version": "0.3.1",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -13,24 +13,31 @@
       "types": "./dist/index.d.ts"
     }
   },
+  "scripts": {
+    "build": "tsup src/index.ts src/cli.ts --format esm --dts",
+    "dev": "tsup src/index.ts src/cli.ts --format esm --dts --watch"
+  },
   "dependencies": {
+    "@clack/prompts": "^1.1.0",
+    "@exagent/sdk": "workspace:*",
     "@polymarket/clob-client": "^4.0.0",
+    "boxen": "^8.0.1",
     "commander": "^12.0.0",
     "ethers": "^5.7.2",
+    "figlet": "^1.10.0",
+    "gradient-string": "^3.0.0",
+    "picocolors": "^1.1.1",
     "viem": "^2.21.0",
     "ws": "^8.16.0",
-    "zod": "^3.22.0",
-    "@exagent/sdk": "0.2.1"
+    "zod": "^3.22.0"
   },
   "devDependencies": {
+    "@types/figlet": "^1.7.0",
+    "@types/gradient-string": "^1.1.6",
     "@types/node": "^20.0.0",
     "@types/ws": "^8.5.0",
     "tsup": "^8.0.0",
     "tsx": "^4.0.0",
     "typescript": "^5.6.0"
-  },
-  "scripts": {
-    "build": "tsup src/index.ts src/cli.ts --format esm --dts",
-    "dev": "tsup src/index.ts src/cli.ts --format esm --dts --watch"
   }
-}
+}

package/src/cli.ts

@@ -3,13 +3,15 @@ import { Command } from 'commander';
 import { loadConfig, writeSampleConfig } from './config.js';
 import { AgentRuntime } from './runtime.js';
 import { listTemplates } from './strategy/index.js';
+import { ensureLocalSetup, promptSecretPassword } from './setup.js';
+import { printBanner, printDone, printError, printSuccess, pc } from './ui.js';
 
 const program = new Command();
 
 program
   .name('exagent')
   .description('Exagent — LLM trading agent runtime')
-  .version('0.1.0');
+  .version('0.3.0');
 
 program
   .command('init')
@@ -18,10 +20,26 @@ program
   .option('--api-url <url>', 'API server URL', 'http://localhost:3002')
   .option('--config <path>', 'Config file path', 'agent-config.json')
   .action((opts) => {
+    printBanner();
     writeSampleConfig(opts.agentId, opts.apiUrl, opts.config);
-    console.log(`Created ${opts.config}`);
-    console.log('Edit the file with your API token, LLM key, and strategy settings.');
-    console.log('Then run: exagent run');
+    printDone(`Created ${pc.cyan(opts.config)}`);
+    console.log();
+    console.log(`  Edit only the public strategy/venue/risk settings.`);
+    console.log(`  Then run: ${pc.cyan(`exagent setup --config ${opts.config}`)}`);
+    console.log();
+  });
+
+program
+  .command('setup')
+  .description('Run first-time secure local setup for agent secrets')
+  .option('--config <path>', 'Config file path', 'agent-config.json')
+  .action(async (opts) => {
+    try {
+      await ensureLocalSetup(opts.config);
+    } catch (err) {
+      printError((err as Error).message);
+      process.exit(1);
+    }
   });
 
 program
@@ -30,14 +48,20 @@ program
   .option('--config <path>', 'Config file path', 'agent-config.json')
   .action(async (opts) => {
     try {
-      const config = loadConfig(opts.config);
+      await ensureLocalSetup(opts.config);
+      printBanner();
+      const config = await loadConfig(opts.config, {
+        getSecretPassword: async () => promptSecretPassword(),
+      });
+      printDone(`Agent ${pc.cyan(config.agentId)} starting...`);
+      console.log();
       const runtime = new AgentRuntime(config);
       await runtime.start();
 
       // Keep the process running
       await new Promise(() => {});
     } catch (err) {
-      console.error('Failed to start agent:', (err as Error).message);
+      printError((err as Error).message);
       process.exit(1);
     }
   });
@@ -46,12 +70,14 @@ program
   .command('templates')
   .description('List available strategy templates')
   .action(() => {
+    printBanner();
     const templates = listTemplates();
-    console.log('\nAvailable strategy templates:\n');
+    console.log(pc.bold('  Strategy Templates'));
+    console.log();
     for (const t of templates) {
-      console.log(`  ${t.id}`);
-      console.log(`    ${t.name} — ${t.description}`);
-      console.log(`    Risk: ${t.riskLevel} | Venues: ${t.venues.join(', ') || 'any'}`);
+      console.log(`  ${pc.cyan(t.id)}`);
+      console.log(`  ${t.name} — ${pc.dim(t.description)}`);
+      console.log(`  ${pc.dim(`Risk: ${t.riskLevel} | Venues: ${t.venues.join(', ') || 'any'}`)}`);
       console.log();
     }
   });
@@ -62,20 +88,33 @@ program
   .option('--config <path>', 'Config file path', 'agent-config.json')
   .action(async (opts) => {
     try {
-      const config = loadConfig(opts.config);
+      await ensureLocalSetup(opts.config);
+      printBanner();
+      const config = await loadConfig(opts.config, {
+        getSecretPassword: async () => promptSecretPassword(),
+      });
       const res = await fetch(`${config.apiUrl}/v1/agents/${config.agentId}`, {
         headers: { Authorization: `Bearer ${config.apiToken}` },
       });
 
       if (!res.ok) {
-        console.error(`API error: ${res.status}`);
+        printError(`API error: ${res.status}`);
         process.exit(1);
       }
 
-      const agent = await res.json();
-      console.log(JSON.stringify(agent, null, 2));
+      const agent = await res.json() as Record<string, unknown>;
+      const name = (agent.name as string) || config.agentId;
+      const status = (agent.status as string) || 'unknown';
+
+      const statusColor = status === 'online' ? pc.green : status === 'error' ? pc.red : pc.yellow;
+
+      printSuccess(name, [
+        `${pc.dim('Status:')}  ${statusColor(status)}`,
+        `${pc.dim('Agent:')}   ${pc.cyan(config.agentId)}`,
+        `${pc.dim('API:')}     ${pc.dim(config.apiUrl)}`,
+      ]);
     } catch (err) {
-      console.error('Failed to check status:', (err as Error).message);
+      printError((err as Error).message);
       process.exit(1);
     }
   });