
    q9i                         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y)    )OptionalTupleList)BaseAPI)	Documents
EmbeddingsIDs	MetadatasNapiids
embeddings	metadatas	documentsreturnc                    g }t        |      | j                         kD  rt        dt        |      | j                               D ]s  }|j                  |||| j                         z    ||||| j                         z    nd |r|||| j                         z    nd |r|||| j                         z    nd f       u |S |j                  ||||f       |S )Nr   )lenget_max_batch_sizerangeappend)r   r   r   r   r   _batchesis          c/var/www/html/leadgen/airagagent/rag_env/lib/python3.12/site-packages/chromadb/utils/batch_utils.pycreate_batchesr      s     	  3x#((**q#c(C$:$:$<= 
	AOOA 6 6 889!- q1s'='='?#?@CLIa!c&<&<&>">?RVCLIa!c&<&<&>">?RV	
	 O 	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   