from docarray import DocumentArray, Document


record = DocumentArray([
    Document(tags={'track_id': 'a'}),
    Document(tags={'track_id': 'b'}),
])

all_tracks_a = DocumentArray(record.find({'tags__track_id': {'$eq': 'a'}}), copy=True)
all_tracks_b = DocumentArray(record.find({'tags__track_id': {'$in': ['a', 'b']}}), copy=True)

result = DocumentArray()

for doc in all_tracks_a:
    doc.tags['label'] = 'l_a'
result.extend(all_tracks_a)
print(result[0].tags['label'])  # >>> 'l_a'

for doc in all_tracks_b:
    doc.tags['label'] = 'l_b'
result.extend(all_tracks_b)
print(result[0].tags['label'])  # >>> 'l_b'