'Xhx ddlmZddlmZddlmZmZmZmZddl m Z ddl Z ddl mZddlmZddlmZmZmZmZdd lmZdd lmZ dd ed ed eeefdedeededefdZGddeZy))count)sleep)AnyIterableOptionalUnion)uuid4N)ResourceExhaustedResponse)SyncApis)Batch PointsList PointStructShardKeySelector) BaseUploader) show_warningopenapi_clientcollection_namebatch max_retriesshard_key_selectorwaitreturnc |\}}}|dtDn|}|dtDn|}t|||D cgc]B\} } } t| t| tj r| j n| xsi| D} } } } d} | |kr, |jj|t| || yycc} } } w#t$r?}td|jdtd t|jYd}~nAd}~wt$r2}td | d zd td | |d z k(r|| d z } Yd}~nd}~wwxYw| |kr)Nc3BK|]}ttywN)strr .0_s N/RAG/venv/lib/python3.12/site-packages/qdrant_client/uploader/rest_uploader.py zupload_batch..s/w!UWwsc3 K|]}dywrrs r r!zupload_batch..s+7aT7s )idvectorpayloadr)points shard_key)rpoint_insert_operationsrz3Batch upload failed due to rate limit. Waiting for z seconds before retrying...)messagecategory stacklevelzBatch upload failed z times. Retrying...T)rzipr isinstancenpndarraytolist points_api upsert_pointsr r r retry_after_s UserWarningr Exception)rrrrrr ids_batch vectors_batch payload_batchidxr%r&r'attemptexes r upload_batchr@s}/4+I}m3<3D/uw/)I/res11L'VV8= 222  2 2 !!12 2  2 2j% %rK