@exagent/agent 0.2.0 → 0.3.0

package/src/setup.ts

@@ -0,0 +1,233 @@
+import { chmodSync, existsSync, mkdirSync, writeFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { dirname, resolve } from 'node:path';
+import { createInterface } from 'node:readline/promises';
+import { stdin as input, stdout as output } from 'node:process';
+import { generatePrivateKey, privateKeyToAccount } from 'viem/accounts';
+import {
+  encryptSecretPayload,
+  getDefaultSecureStorePath,
+  readConfigFile,
+  type LocalSecretPayload,
+  type RuntimeConfigFile,
+  writeConfigFile,
+} from './config.js';
+
+interface BootstrapPayload {
+  apiToken: string;
+  walletPrivateKey?: string;
+  llm?: {
+    provider?: string;
+    model?: string;
+    apiKey?: string;
+  };
+}
+
+function expandHomeDir(path: string): string {
+  if (!path.startsWith('~/')) return path;
+  return resolve(homedir(), path.slice(2));
+}
+
+async function prompt(question: string): Promise<string> {
+  const rl = createInterface({ input, output });
+  try {
+    return (await rl.question(question)).trim();
+  } finally {
+    rl.close();
+  }
+}
+
+async function promptSecret(question: string): Promise<string> {
+  const rl = createInterface({ input, output, terminal: true }) as ReturnType<typeof createInterface> & {
+    stdoutMuted?: boolean;
+    _writeToOutput?: (text: string) => void;
+  };
+  rl.stdoutMuted = true;
+  const originalWrite = rl._writeToOutput?.bind(rl);
+  rl._writeToOutput = (text: string) => {
+    if (!rl.stdoutMuted) {
+      originalWrite?.(text);
+    }
+  };
+
+  try {
+    const answer = (await rl.question(question)).trim();
+    output.write('\n');
+    return answer;
+  } finally {
+    rl.close();
+  }
+}
+
+export async function promptSecretPassword(question: string = 'Device password: '): Promise<string> {
+  return promptSecret(question);
+}
+
+async function promptPasswordWithConfirmation(): Promise<string> {
+  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;
+  }
+}
+
+async function promptWalletPrivateKey(): Promise<string> {
+  while (true) {
+    const choice = (await prompt('Wallet setup — [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}\n`);
+      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');
+  }
+}
+
+async function promptLlmProvider(): Promise<string> {
+  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');
+  }
+}
+
+async function consumeBootstrapPackage(config: RuntimeConfigFile): Promise<BootstrapPayload> {
+  if (!config.secrets?.bootstrapToken) {
+    return { apiToken: '' };
+  }
+
+  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 secure setup package: ${body}`);
+  }
+
+  const data = await res.json() as { payload: BootstrapPayload };
+  return data.payload;
+}
+
+async function buildLocalSecrets(config: RuntimeConfigFile, bootstrapPayload: BootstrapPayload): Promise<{ config: RuntimeConfigFile; secrets: LocalSecretPayload }> {
+  const nextConfig = structuredClone(config);
+  const llm = { ...(nextConfig.llm || {}) };
+
+  if (bootstrapPayload.llm?.provider && !llm.provider) {
+    llm.provider = bootstrapPayload.llm.provider as RuntimeConfigFile['llm']['provider'];
+  }
+  if (bootstrapPayload.llm?.model && !llm.model) {
+    llm.model = bootstrapPayload.llm.model;
+  }
+
+  if (!llm.provider) {
+    llm.provider = await promptLlmProvider() as RuntimeConfigFile['llm']['provider'];
+  }
+  if (!llm.model) {
+    llm.model = await prompt('Agent LLM model: ');
+    if (!llm.model) {
+      throw new Error('Agent LLM model is required');
+    }
+  }
+
+  const secrets: LocalSecretPayload = {
+    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');
+  }
+
+  nextConfig.llm = llm;
+  delete nextConfig.apiToken;
+  delete nextConfig.wallet;
+  delete nextConfig.llm.apiKey;
+
+  return { config: nextConfig, secrets };
+}
+
+function writeSecureStore(path: string, secrets: LocalSecretPayload, password: string): string {
+  const secureStorePath = expandHomeDir(path);
+  const encrypted = encryptSecretPayload(secrets, password);
+  const dir = dirname(secureStorePath);
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true, mode: 0o700 });
+  }
+  writeFileSync(secureStorePath, JSON.stringify(encrypted, null, 2), { mode: 0o600 });
+  try {
+    chmodSync(secureStorePath, 0o600);
+  } catch {
+    // Best effort only — some platforms ignore chmod semantics.
+  }
+  return secureStorePath;
+}
+
+export async function ensureLocalSetup(configPath: string): Promise<void> {
+  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;
+  }
+
+  const bootstrapPayload = await consumeBootstrapPackage(config);
+  const { config: nextConfig, secrets } = await buildLocalSecrets(config, bootstrapPayload);
+  const password = await promptPasswordWithConfirmation();
+  const secureStorePath = writeSecureStore(
+    nextConfig.secrets?.secureStorePath || getDefaultSecureStorePath(nextConfig.agentId),
+    secrets,
+    password,
+  );
+
+  nextConfig.secrets = {
+    secureStorePath,
+  };
+
+  writeConfigFile(configPath, nextConfig);
+  output.write(`Encrypted local secret store created at ${secureStorePath}\n`);
+}

package/src/strategy/loader.ts

@@ -1,16 +1,44 @@
 import { existsSync } from 'node:fs';
 import { resolve } from 'node:path';
+import { z } from 'zod';
 import type { StrategyFunction, StrategyContext, TradeSignal } from '@exagent/sdk';
 import { getTemplate } from './templates.js';
 
+const promptSignalSchema = z.object({
+  symbol: z.string().min(1),
+  side: z.enum(['buy', 'sell', 'long', 'short']),
+  confidence: z.number().min(0).max(1).optional(),
+  reasoning: z.string().optional(),
+  venue: z.string().optional(),
+  chain: z.string().optional(),
+  size: z.number().positive().optional(),
+  price: z.number().positive().optional(),
+  fee: z.number().min(0).optional(),
+  venueFillId: z.string().optional(),
+  venueTimestamp: z.string().optional(),
+  leverage: z.number().positive().optional(),
+  orderType: z.string().optional(),
+});
+
+const promptSignalArraySchema = z.array(promptSignalSchema);
+
 export async function loadStrategy(config: {
   file?: string;
   template?: string;
+  prompt?: {
+    name?: string;
+    systemPrompt: string;
+    venues?: string[];
+  };
 }): Promise<StrategyFunction> {
   if (config.file) {
     return loadFromFile(config.file);
   }
 
+  if (config.prompt) {
+    return loadFromPrompt(config.prompt);
+  }
+
   if (config.template) {
     const template = getTemplate(config.template);
     if (!template) {
@@ -51,6 +79,76 @@ async function loadFromCode(code: string): Promise<StrategyFunction> {
   return fn;
 }
 
+function loadFromPrompt(config: {
+  name?: string;
+  systemPrompt: string;
+  venues?: string[];
+}): StrategyFunction {
+  return async (context: StrategyContext): Promise<TradeSignal[]> => {
+    const prices = context.market.getPrices();
+    const positions = context.position.openPositions.map((position) => ({
+      token: position.token,
+      quantity: position.quantity,
+      costBasisPerUnit: position.costBasisPerUnit,
+      venue: position.venue,
+      chain: position.chain,
+    }));
+
+    const response = await context.llm.chat([
+      { role: 'system', content: config.systemPrompt },
+      {
+        role: 'user',
+        content: [
+          `Strategy: ${config.name || 'Prompt Strategy'}`,
+          `Allowed venues: ${(config.venues || []).join(', ') || 'any'}`,
+          `Current prices: ${JSON.stringify(prices)}`,
+          `Open positions: ${JSON.stringify(positions)}`,
+          `Risk config: ${JSON.stringify(context.config)}`,
+          'Return ONLY a JSON array of trade signals.',
+        ].join('\n'),
+      },
+    ]);
+
+    const match = response.content.match(/\[[\s\S]*\]/);
+    if (!match) {
+      context.log('Prompt strategy returned a non-JSON response; no signals emitted.');
+      return [];
+    }
+
+    try {
+      const parsed = promptSignalArraySchema.parse(JSON.parse(match[0]));
+      const signals: TradeSignal[] = [];
+      for (const signal of parsed) {
+        const price = signal.price ?? prices[signal.symbol.toUpperCase()];
+        if (!price || price <= 0) {
+          context.log(`Prompt strategy skipped ${signal.symbol}: no usable price in response or market cache.`);
+          continue;
+        }
+
+        signals.push({
+          symbol: signal.symbol,
+          side: signal.side,
+          confidence: signal.confidence ?? 0.5,
+          reasoning: signal.reasoning,
+          venue: signal.venue || config.venues?.[0] || 'manual',
+          chain: signal.chain,
+          size: signal.size ?? 1,
+          price,
+          fee: signal.fee ?? 0,
+          venueFillId: signal.venueFillId ?? '',
+          venueTimestamp: signal.venueTimestamp ?? new Date().toISOString(),
+          leverage: signal.leverage,
+          orderType: signal.orderType,
+        });
+      }
+      return signals;
+    } catch (err) {
+      context.log(`Prompt strategy parse failed: ${(err as Error).message}`);
+      return [];
+    }
+  };
+}
+
 const holdStrategy: StrategyFunction = async (_context: StrategyContext): Promise<TradeSignal[]> => {
   return []; // No trades — hold position
 };