Function convertAgentTraceToUiChunk

backend/src/agents/agents.controller.ts:134–254

(event: AgentTracePartEntry)

const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));

function convertAgentTraceToUiChunk(event: AgentTracePartEntry): UIMessageChunk | null {
  const payload = (event.part ?? {}) as Record<string, unknown>;
  const type = typeof payload.type === 'string' ? payload.type : undefined;
  if (!type) {
    return null;
  }

  const baseMessageId =
    typeof payload.messageId === 'string' && payload.messageId.length > 0
      ? payload.messageId
      : event.agentRunId;

  if (type === 'message-start') {
    return {
      type: 'start',
      messageId: baseMessageId,
      messageMetadata: {
        workflowRunId: event.workflowRunId,
        nodeRef: event.nodeRef,
        role: typeof payload.role === 'string' ? payload.role : 'assistant',
        sequence: event.sequence,
      },
    };
  }

  if (type === 'text-start') {
    return {
      type: 'text-start',
      id: ensureString(payload.id) ?? baseMessageId,
    };
  }

  if (type === 'data-text-start') {
    return {
      type: 'text-start',
      id: ensureString(payload.id) ?? baseMessageId,
    };
  }

  if (type === 'text-end') {
    return {
      type: 'text-end',
      id: ensureString(payload.id) ?? baseMessageId,
    };
  }

  if (type === 'data-text-end') {
    return {
      type: 'text-end',
      id: ensureString(payload.id) ?? baseMessageId,
    };
  }

  if (type === 'text-delta') {
    return {
      type: 'text-delta',
      id: ensureString(payload.id) ?? baseMessageId,
      delta: typeof payload.textDelta === 'string' ? payload.textDelta : '',