_juLddlmZmZmZmZddlmZmZmZGddZ dS))DictListSetcast) LogRecord OperationVectorcBeZdZUdZeeefed<eeed<eeed<eeed<e ed<e ed<dd Z de fd Z de efd Z de efd Zde ede efdZdedefdZdedefdZede fdZddededd 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_countreturnNci|_t|_t|_t|_d|_d|_dS)Nr)r setr rrrrselfs _/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/segment/impl/vector/batch.py__init__zBatch.__init__sC!EEEE"uucTt|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 ridsc fd|DS)z.Get the list of vectors to write in this batchchg|].}ttj|dd/S)record embedding)rr r ).0idrs r z-Batch.get_written_vectors..%sB   NPD-b1(;KH I I   r)rr#s` rget_written_vectorszBatch.get_written_vectors#s.    TW    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)r r.s r is_deletedzBatch.is_deleted-sT&&&rc*t|jS)N)rr rs r delete_countzBatch.delete_count1s4$%%%rFr&exists_alreadyc2|dd}|ddtjkrl||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|=dSdS||j|<|j |||j vr|j ||ddtj kr@|s,|xjdz c_|j |dS|xjdz c_dS|ddtjkr|xjdz c_dS|ddtjkr|xjdz c_dSdS)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&r4r)s rapplyz Batch.apply5s H d # ( K (I,< < <T&&&!((,,,'+H5kBimSSNNa'NNN(,X6{CyGWWW%%*%%%))"----(,X6{CyGWWWT111!+,33B7777))Q.)))--b1114,,,!%%b)))T)))(,,,*)(.D  $   ! !" % % %T&&&!((,,,h , 0@@@%+NNa'NN(,,R00000%%*%%%%!+.)-??!#!+.)2BBB!!Q&!!!!CBr)rN)F)__name__ __module__ __qualname____doc__rstrr__annotations__rintrrrrr"r r,r/boolr1propertyr3r>r+rrr r sAA#y.))))c(c(XNNN?????'c'''''c'''' tCy T&\    (S(Y(((('S'T''''&c&&&X&5'5'I5't5'5'5'5'5'5'5'rr N) typingrrrrchromadb.typesrrr r r+rrrJs((((((((((((7777777777e'e'e'e'e'e'e'e'e'e'r