o gu@s>ddlmZmZmZmZddlmZmZmZGdddZ dS))DictListSetcast) LogRecord OperationVectorc@seZdZUdZeeefed<eeed<eeed<eeed<e ed<e ed<d!d d Z de fd d Z de efddZ de efddZde ede efddZdedefddZdedefddZede fddZd"dededd fdd Zd 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_countreturnNcCs.i|_t|_t|_t|_d|_d|_dS)Nr)r setr r r rrselfrY/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/segment/impl/vector/batch.py__init__s  zBatch.__init__cCst|jt|jS)z'Get the number of changes in this batch)lenr r rrrr__len__sz Batch.__len__cC t|jS)z0Get the list of deleted embeddings in this batch)listr rrrrget_deleted_ids zBatch.get_deleted_idscCr)z0Get the list of written embeddings in this batch)rr rrrrget_written_idsrzBatch.get_written_idsidscsfdd|DS)z.Get the list of vectors to write in this batchcs$g|]}ttj|ddqS)record embedding)rrr ).0idrrr %sz-Batch.get_written_vectors..r)rrrrrget_written_vectors#s zBatch.get_written_vectorsr"cCs |j|S)zGet the record for a given ID)r rr"rrr get_record)rzBatch.get_recordcCs ||jvS)zCheck if a given ID is deleted)r r%rrr is_deleted-rzBatch.is_deletedcCr)N)rr rrrr delete_count1rzBatch.delete_countFrexists_alreadycCs|dd}|ddtjkr||jvru|j||j|ddtjkr.|jd8_nR|j|ddtjkrH|jd8_|j |n8|j|ddtj krt||j vrg|jd8_|j |n|jd8_|j |n ||j vr|j |||jvr|j|=dSdS||j|<|j |||j vr|j ||ddtj kr|s|jd7_|j |dS|jd7_dS|ddtjkr|jd7_dS|ddtjkr|jd7_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. rr" operationN) rDELETEr remover ADDrUPDATErr addUPSERTr )rrr)r"rrrapply5sF             z Batch.apply)rN)F)__name__ __module__ __qualname____doc__rstrr__annotations__rintrrrrrrr$r&boolr'propertyr(r2rrrrr s$     r N) typingrrrrchromadb.typesrrrr rrrrs