Function mergeProcessRecordsWithToolCalls

api/controller/message.go:209–251

(processRecordMap map[int64][]*model.MessageProcessStep, toolCallMap map[int64][]*model.MessageToolCall)

}

func mergeProcessRecordsWithToolCalls(processRecordMap map[int64][]*model.MessageProcessStep, toolCallMap map[int64][]*model.MessageToolCall) map[int64][]*model.MessageProcessStep {
	merged := make(map[int64][]*model.MessageProcessStep, len(processRecordMap)+len(toolCallMap))
	for messageID, records := range processRecordMap {
		if len(records) == 0 {
			continue
		}
		merged[messageID] = append(merged[messageID], records...)
	}

	for messageID, toolCalls := range toolCallMap {
		for _, toolCall := range toolCalls {
			record := convertToolCallToProcessRecord(toolCall)
			if record == nil {
				continue
			}
			merged[messageID] = append(merged[messageID], record)
		}
	}

	for messageID := range merged {
		records := merged[messageID]
		sort.SliceStable(records, func(i, j int) bool {
			left := records[i]
			right := records[j]
			if left == nil {
				return false
			}
			if right == nil {
				return true
			}
			if left.StepTimestamp != right.StepTimestamp {
				return left.StepTimestamp < right.StepTimestamp
			}
			if left.CreatedTime != right.CreatedTime {
				return left.CreatedTime < right.CreatedTime
			}
			return left.ID < right.ID
		})
		merged[messageID] = records
	}

	return merged
}

func parseConversationIDParam(rawID string) (int64, error) {
	return hashids.TryParseID(rawID)