parseMultiAgentResponse.ts

/**
 * Parses multi-agent LLM responses where multiple agents reply in a single stream.
 * Format: **AgentDisplayName:** message text
 *
 * Returns segments for each detected agent (excluding the originating agent).
 * System messages like **Party Mode encerrado.** (no colon) are ignored.
 */

export interface ParsedAgentSegment {
  agentId: string;
  displayName: string;
  message: string;
}

/**
 * Matches **DisplayName:** at start of line or after newline.
 * Requires colon after the bold name (excludes **Party Mode encerrado.**).
 */
const AGENT_PATTERN = /(?:\n|^)\*\*([^*]+)\*\*:\s*/g;

export function parseMultiAgentResponse(
  fullText: string,
  displayNameToId: Map<string, string>,
  originatingAgentId: string
): ParsedAgentSegment[] {
  const segments: ParsedAgentSegment[] = [];
  const knownIds = new Set(displayNameToId.values());

  let lastIndex = 0;
  let lastDisplayName: string | null = null;
  let lastAgentId: string | null = null;

  let match: RegExpExecArray | null;
  AGENT_PATTERN.lastIndex = 0;

  while ((match = AGENT_PATTERN.exec(fullText)) !== null) {
    const displayName = match[1].trim();
    const agentId = displayNameToId.get(displayName.toLowerCase());

    if (lastDisplayName !== null && lastAgentId !== null && lastAgentId !== originatingAgentId && knownIds.has(lastAgentId)) {
      const message = fullText.slice(lastIndex, match.index).trim();
      if (message.length > 0) {
        segments.push({ agentId: lastAgentId, displayName: lastDisplayName, message });
      }
    }

    lastIndex = match.index + match[0].length;
    lastDisplayName = displayName;
    lastAgentId = agentId ?? null;
  }

  if (lastDisplayName !== null && lastAgentId !== null && lastAgentId !== originatingAgentId && knownIds.has(lastAgentId)) {
    const message = fullText.slice(lastIndex).trim();
    if (message.length > 0) {
      segments.push({ agentId: lastAgentId, displayName: lastDisplayName, message });
    }
  }

  return segments;
}