agent-flow.mjs

import { applyFirstUserMessage, buildInputMessage } from './prompt-builder.mjs';
import { join } from 'node:path';
import { homedir } from 'node:os';
import { readJson } from './runtime.mjs';
import { getHomeDirectory } from './platform.mjs';

export function resolveAgentApiKey(env = process.env) {
  const apiKey = String(env.agentx_api_key || env.AGENTX_API_KEY || '').trim();
  if (apiKey) return apiKey;
  throw new Error('Set agentx_api_key or AGENTX_API_KEY in your shell environment.');
}

export async function loadPromptTemplate(promptPath, mcpPath = join(getHomeDirectory() || homedir(), '.agentx.mcp.json'), env = process.env) {
  try {
    const template = await readJson(promptPath);
    let mcpTools = null;
    try {
      const configuredTools = await readJson(mcpPath);
      mcpTools = Array.isArray(configuredTools) ? configuredTools : configuredTools?.tools || [];
    } catch (error) {
      if (error?.code !== 'ENOENT') throw error;
    }
    const merged = mcpTools === null ? template : { ...template, tools: [...(template.tools || []), ...mcpTools] };
    if (!env?.AGENTX_WORKER_ID) return merged;
    return { ...merged, tools: (merged.tools || []).filter((tool) => !['spawn_agent', 'agent_status', 'cancel_agent'].includes(tool?.name)) };
  } catch (error) {
    throw new Error(`Unable to read prompt template at ${promptPath}: ${error?.message || String(error)}`);
  }
}

function formatShellCommandOutput(output) {
  if (output && typeof output === 'object' && !Array.isArray(output)) {
    const stdout = String(output.stdout ?? '').trimEnd();
    const stderr = String(output.stderr ?? '').trimEnd();
    const parts = [];
    if (stdout) parts.push(stdout);
    if (stderr) parts.push(stdout ? `stderr:\n${stderr}` : stderr);
    return parts.join('\n\n').trimEnd();
  }
  return String(output ?? '').trimEnd();
}

export function appendCliTranscript(existingTranscript, command, outputText) {
  const entry = [`! ${command}`];
  const trimmedOutput = formatShellCommandOutput(outputText);
  if (trimmedOutput) entry.push(trimmedOutput);
  return [existingTranscript, entry.join('\n')].filter(Boolean).join('\n\n');
}

export function buildRequestMessage({ pendingCliTranscript, cwdNote, message }) {
  const contextParts = [];
  if (pendingCliTranscript) {
    contextParts.push(`Local shell commands and output since the last assistant message:\n\n${pendingCliTranscript}`);
  }
  if (cwdNote) {
    contextParts.push(cwdNote);
  }
  contextParts.push(message);
  return contextParts.join('\n\n');
}

export function buildRequestOverride(template, userMessage, agentsText, cwd, previousResponseId) {
  if (previousResponseId) {
    return {
      ...template,
      input: [buildInputMessage(userMessage)],
      store: true,
      previous_response_id: previousResponseId,
    };
  }

  return {
    ...applyFirstUserMessage(template, userMessage, agentsText, cwd),
    store: true,
  };
}