@exagent/agent 0.3.0 → 0.3.1

package/dist/cli.js

@@ -8,7 +8,7 @@ import {
   readConfigFile,
   writeConfigFile,
   writeSampleConfig
-} from "./chunk-UAP5CTHB.js";
+} from "./chunk-VDK4XPAC.js";
 
 // src/cli.ts
 import { Command } from "commander";
@@ -17,97 +17,82 @@ import { Command } from "commander";
 import { chmodSync, existsSync, mkdirSync, writeFileSync } from "fs";
 import { homedir } from "os";
 import { dirname, resolve } from "path";
-import { createInterface } from "readline/promises";
-import { stdin as input, stdout as output } from "process";
 import { generatePrivateKey, privateKeyToAccount } from "viem/accounts";
-function expandHomeDir(path) {
-  if (!path.startsWith("~/")) return path;
-  return resolve(homedir(), path.slice(2));
-}
-async function prompt(question) {
-  const rl = createInterface({ input, output });
+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 {
-    return (await rl.question(question)).trim();
-  } finally {
-    rl.close();
+    const require2 = createRequire(import.meta.url);
+    const pkg = require2("../package.json");
+    return pkg.version || "0.0.0";
+  } catch {
+    return "0.0.0";
   }
 }
-async function promptSecret(question) {
-  const rl = createInterface({ input, output, terminal: true });
-  rl.stdoutMuted = true;
-  const originalWrite = rl._writeToOutput?.bind(rl);
-  rl._writeToOutput = (text) => {
-    if (!rl.stdoutMuted) {
-      originalWrite?.(text);
-    }
-  };
-  try {
-    const answer = (await rl.question(question)).trim();
-    output.write("\n");
-    return answer;
-  } finally {
-    rl.close();
-  }
+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();
 }
-async function promptSecretPassword(question = "Device password: ") {
-  return promptSecret(question);
+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();
 }
-async function promptPasswordWithConfirmation() {
-  output.write("\n");
-  output.write("Important: this password encrypts the local wallet key, relay token, and agent LLM key for this device.\n");
-  output.write("If you lose this password and the agent wallet holds funds, those funds cannot be recovered.\n\n");
-  const ack = await prompt('Type "I UNDERSTAND" to continue: ');
-  if (ack !== "I UNDERSTAND") {
-    throw new Error("Secure setup aborted");
-  }
-  while (true) {
-    const password = await promptSecret("Create a device password (min 12 chars): ");
-    if (password.length < 12) {
-      output.write("Password must be at least 12 characters.\n");
-      continue;
-    }
-    const confirm = await promptSecret("Confirm device password: ");
-    if (password !== confirm) {
-      output.write("Passwords did not match. Try again.\n");
-      continue;
-    }
-    return password;
-  }
+function printStep(step, total, label) {
+  console.log();
+  console.log(accentGradient(`  Step ${step} of ${total}`) + pc.dim(` \u2014 ${label}`));
 }
-async function promptWalletPrivateKey() {
-  while (true) {
-    const choice = (await prompt("Wallet setup \u2014 [1] generate new wallet locally, [2] use existing private key: ")).trim();
-    if (choice === "1") {
-      const privateKey = generatePrivateKey();
-      const address = privateKeyToAccount(privateKey).address;
-      output.write(`Generated wallet address: ${address}
-`);
-      return privateKey;
-    }
-    if (choice === "2") {
-      const privateKey = await promptSecret("Wallet private key (0x...): ");
-      if (/^0x[a-fA-F0-9]{64}$/.test(privateKey)) {
-        return privateKey;
-      }
-      output.write("Invalid private key. Expected a 32-byte hex string prefixed with 0x.\n");
-      continue;
-    }
-    output.write("Enter 1 or 2.\n");
-  }
+function printDone(message) {
+  console.log(`  ${pc.green("\u2713")} ${message}`);
 }
-async function promptLlmProvider() {
-  while (true) {
-    const provider = (await prompt("Agent LLM provider (openai/anthropic/google/deepseek/mistral/groq/together/ollama): ")).toLowerCase();
-    if (["openai", "anthropic", "google", "deepseek", "mistral", "groq", "together", "ollama"].includes(provider)) {
-      return provider;
-    }
-    output.write("Unsupported provider.\n");
-  }
+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" },
@@ -118,46 +103,121 @@ async function consumeBootstrapPackage(config) {
   });
   if (!res.ok) {
     const body = await res.text();
-    throw new Error(`Failed to consume secure setup package: ${body}`);
+    throw new Error(`Failed to consume bootstrap package: ${body}`);
   }
   const data = await res.json();
   return data.payload;
 }
-async function buildLocalSecrets(config, bootstrapPayload) {
-  const nextConfig = structuredClone(config);
-  const llm = { ...nextConfig.llm || {} };
-  if (bootstrapPayload.llm?.provider && !llm.provider) {
-    llm.provider = bootstrapPayload.llm.provider;
+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;
   }
-  if (bootstrapPayload.llm?.model && !llm.model) {
-    llm.model = bootstrapPayload.llm.model;
+  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;
   }
-  if (!llm.provider) {
-    llm.provider = await promptLlmProvider();
+  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;
   }
-  if (!llm.model) {
-    llm.model = await prompt("Agent LLM model: ");
-    if (!llm.model) {
-      throw new Error("Agent LLM model is required");
+  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;
     }
   }
-  const secrets = {
-    apiToken: bootstrapPayload.apiToken || nextConfig.apiToken || await promptSecret("Agent relay token: "),
-    walletPrivateKey: bootstrapPayload.walletPrivateKey || nextConfig.wallet?.privateKey || await promptWalletPrivateKey(),
-    llmApiKey: bootstrapPayload.llm?.apiKey || nextConfig.llm.apiKey || await promptSecret("Agent LLM API key: ")
-  };
-  if (!secrets.apiToken) {
-    throw new Error("Agent relay token is required");
+  if (!apiKey) {
+    apiKey = config.llm?.apiKey;
   }
-  nextConfig.llm = llm;
-  delete nextConfig.apiToken;
-  delete nextConfig.wallet;
-  delete nextConfig.llm.apiKey;
-  return { config: nextConfig, secrets };
+  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 };
 }
-function writeSecureStore(path, secrets, password) {
+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, password);
+  const encrypted = encryptSecretPayload(secrets, password2);
   const dir = dirname(secureStorePath);
   if (!existsSync(dir)) {
     mkdirSync(dir, { recursive: true, mode: 448 });
@@ -169,74 +229,130 @@ function writeSecureStore(path, secrets, password) {
   }
   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);
-  const { config: nextConfig, secrets } = await buildLocalSecrets(config, bootstrapPayload);
-  const password = await promptPasswordWithConfirmation();
+  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,
-    password
+    password2
   );
-  nextConfig.secrets = {
-    secureStorePath
-  };
+  nextConfig.secrets = { secureStorePath };
   writeConfigFile(configPath, nextConfig);
-  output.write(`Encrypted local secret store created at ${secureStorePath}
-`);
+  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 only the public strategy/venue/risk settings in the file.");
-  console.log(`Then run: exagent setup --config ${opts.config}`);
+  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);
-    console.log("Secure local setup complete.");
   } catch (err) {
-    console.error("Failed to complete secure local setup:", err.message);
+    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 {
     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 {
     await ensureLocalSetup(opts.config);
+    printBanner();
     const config = await loadConfig(opts.config, {
       getSecretPassword: async () => promptSecretPassword()
     });
@@ -244,13 +360,20 @@ program.command("status").description("Check agent status").option("--config <pa
       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,9 @@ interface RuntimeConfig {
     };
     strategy: {
         file?: string;
+        code?: string;
         template?: string;
+        venues?: string[];
         prompt?: {
             name?: string;
             systemPrompt: string;
@@ -124,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;
@@ -415,6 +429,7 @@ declare function createLLMAdapter(config: LLMConfig): LLMAdapter;
 
 declare function loadStrategy(config: {
     file?: string;
+    code?: string;
     template?: string;
     prompt?: {
         name?: string;

package/dist/index.js

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exagent/agent",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -18,15 +18,22 @@
     "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"
   },
   "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",