lQjuDddlmZmZmZmZddlmZmZmZGddZ y))DictListSetcast) LogRecord OperationVectorc eZdZUdZeeefed<eeed<eeed<eeed<e ed<e ed<dd Z de fd Z de efd Z de efd Zde ede efdZdedefdZdedefdZede fdZddededd fdZy )Batchz7Used to model the set of changes as an atomic operation_ids_to_records _deleted_ids _written_ids_upsert_add_ids add_count update_countreturnNci|_t|_t|_t|_d|_d|_y)Nr)r setr rrrrselfs g/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/segment/impl/vector/batch.py__init__zBatch.__init__s7!EE"ucXt|jt|jzS)z'Get the number of changes in this batch)lenrr rs r__len__z Batch.__len__s#4$$%D,=,=(>>>rc,t|jS)z0Get the list of deleted embeddings in this batch)listr rs rget_deleted_idszBatch.get_deleted_idsD%%&&rc,t|jS)z0Get the list of written embeddings in this batch)rrrs rget_written_idszBatch.get_written_idsr ridscn|Dcgc]%}tt|j|dd'c}Scc}w)z.Get the list of vectors to write in this batchrecord embedding)rr r )rr#ids rget_written_vectorszBatch.get_written_vectors#s?UX NPD--b1(;KH I   s*2r'c |j|S)zGet the record for a given ID)r rr's r get_recordzBatch.get_record)s##B''rc||jvS)zCheck if a given ID is deleted)r r*s r is_deletedzBatch.is_deleted-sT&&&&rc,t|jS)N)rr rs r delete_countzBatch.delete_count1s4$$%%rr%exists_alreadyc|dd}|ddtjk(r||jvrE|jj||j|ddtj k(r|xj dzc_n|j|ddtjk(r1|xjdzc_|jj|n|j|ddtjk(r||jvr1|xj dzc_|jj|nZ|xjdzc_|jj|n)||jvr|jj|||jvr|j|=yy||j|<|jj|||jvr|jj||ddtjk(rI|s1|xj dz c_|jj|y|xjdz c_y|ddtj k(r|xj dz c_y|ddtjk(r|xjdz c_yy)akApply an embedding record to this batch. Records passed to this method are assumed to be validated for correctness. For example, a delete or update presumes the ID exists in the index. An add presumes the ID does not exist in the index. The exists_already flag should be set to True if the ID does exist in the index, and False otherwise. r%r' operationN) rDELETErremover ADDrUPDATErr addUPSERTr)rr%r0r's rapplyz Batch.apply5sp H d # ( K (I,<,< <T&&&!!((,''+H5kBimmSNNa'N((,X6{CyGWGWW%%*%%%))"-((,X6{CyGWGWWT111!+,,33B7))Q.)))--b14,,,!!%%b)T)))((,*(.D  $    ! !" %T&&&!!((,h , 0@0@@%NNa'N((,,R0%%*%!+.)--?!#!+.)2B2BB!!Q&!Cr)rN)F)__name__ __module__ __qualname____doc__rstrr__annotations__rintrrrrr"r r(r+boolr-propertyr/r:rrr r sA#y.))c(c(XN??'c''c' tCy T&\ (S(Y('S'T'&c&&5'I5't5'5'rr N) typingrrrrchromadb.typesrrr r rDrrrGs((77e'e'r