Method attachWorkflowRunToToolCall

src/node/services/streamManager.ts:693–725

(event: WorkflowRunAttachedEvent)

  }

  async attachWorkflowRunToToolCall(event: WorkflowRunAttachedEvent): Promise<boolean> {
    const workspaceId = event.workspaceId as WorkspaceId;
    const streamInfo = this.workspaceStreams.get(workspaceId);
    if (
      streamInfo == null ||
      (event.messageId != null && streamInfo.messageId !== event.messageId)
    ) {
      return false;
    }

    const attachment = this.getWorkflowRunAttachment(event);
    const partIndex = streamInfo.parts.findIndex(
      (part) => part.type === "dynamic-tool" && part.toolCallId === event.toolCallId
    );
    if (partIndex === -1) {
      (streamInfo.pendingWorkflowRunAttachments ??= new Map()).set(event.toolCallId, attachment);
      return true;
    }

    const part = streamInfo.parts[partIndex];
    if (part.type !== "dynamic-tool") {
      return false;
    }

    streamInfo.pendingWorkflowRunAttachments?.delete(event.toolCallId);
    streamInfo.parts[partIndex] = {
      ...part,
      workflowRun: attachment,
    };

    await this.flushPartialWrite(workspaceId, streamInfo);
    return true;
  }

  /**
   * Write the current partial message to disk (throttled by mtime)