uiPddlmZmZmZmZddlmZmZmZm Z GddZ dS))DictListSetcast) LogRecord OperationSeqIdVectorcLeZdZUdZeeefed<eeed<eeed<eeed<e ed<e ed<e ed<dd Z d e fd Z d e efd Zd e efdZde ed e efdZded efdZded efdZed e fdZddeded d fdZd S)Batchz7Used to model the set of changes as an atomic operation_ids_to_records _deleted_ids _written_ids_upsert_add_ids add_count update_count max_seq_idreturnNci|_t|_t|_t|_d|_d|_d|_dS)Nr)r setrrrrrrselfs U/var/www/icac/venv/lib/python3.11/site-packages/chromadb/segment/impl/vector/batch.py__init__zBatch.__init__sH!EEEE"uucTt|jt|jzS)z'Get the number of changes in this batch)lenrrrs r__len__z Batch.__len__s#4$%%D,=(>(>>>rc*t|jS)z0Get the list of deleted embeddings in this batch)listrrs rget_deleted_idszBatch.get_deleted_idsD%&&&rc*t|jS)z0Get the list of written embeddings in this batch)r rrs rget_written_idszBatch.get_written_ids"r"ridsc fd|DS)z.Get the list of vectors to write in this batchchg|].}ttj|dd/S)operation_record embedding)rr r ).0idrs r z-Batch.get_written_vectors..(sF    -b12DEkR S S   r)rr%s` rget_written_vectorszBatch.get_written_vectors&s.        rr+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)rr0s r is_deletedzBatch.is_deleted1sT&&&rc*t|jS)N)rrrs r delete_countzBatch.delete_count5s4$%%%rFrecordexists_alreadycb|dd}|ddtjkri||jvr*|j||j|ddtjkr|xjdzc_n|j|ddtjkr+|xjdzc_|j |n|j|ddtj kr^||j vr+|xjdzc_|j |nN|xjdzc_|j |n#||j vr|j |||jvr|j|=n||j|<|j |||j vr|j ||ddtj kr>|s+|xjdz c_|j |nj|xjdz c_nY|ddtjkr|xjdz c_n,|ddtjkr|xjdz c_t|j|d|_dS)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+ operation log_offsetN)rDELETErremover ADDrUPDATErraddUPSERTrmaxr)rr6r7r+s rapplyz Batch.apply9s & ' - $ %k 2i6F F FT&&&!((,,,(,-?@M }%%NNa'NNN(,-?@M '((%%*%%%))"----(,-?@M '((T111!+,33B7777))Q.)))--b1114,,,!%%b)))T)))(,(.D  $   ! !" % % %T&&&!((,,,()+6):JJJ%+NNa'NN(,,R0000%%*%%%*+K8IMII!#*+K8IrOs((((((((((((>>>>>>>>>>>>pEpEpEpEpEpEpEpEpEpEr