
    Zi                         d dl mZmZmZ d dlmZ d dlmZmZm	Z	m
Z
 	 	 	 ddede	dee         dee
         d	ee         d
eee	ee         ee
         ee         f                  fdZdS )    )OptionalTupleList)BaseAPI)	Documents
EmbeddingsIDs	MetadatasNapiids
embeddings	metadatas	documentsreturnc                     g }t          |          |                                 k    rt          dt          |          |                                           D ]}|                    ||||                                 z            |||||                                 z            nd |r||||                                 z            nd |r||||                                 z            nd f           n|                    ||||f           |S )Nr   )lenget_max_batch_sizerangeappend)r   r   r   r   r   _batchesis          W/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/utils/batch_utils.pycreate_batchesr      s;    	  3xx#((****q#c((C$:$:$<$<== 
	 
	AOOA 6 6 8 8889!- q1s'='='?'?#??@@CLVIa!c&<&<&>&>">>??RVCLVIa!c&<&<&>&>">>??RV	 	 	 	
	 	j)Y?@@@O    )NNN)typingr   r   r   chromadb.apir   chromadb.api.typesr   r   r	   r
   r    r   r   <module>r      s   ( ( ( ( ( ( ( ( ( (                        (,%)%) 		 $ 	"	
 	" 
%Xj)8I+>@SS
TU     r   