§
    ˜ŠtjŠ	  ã                   óv   — d dl mZ d dlmZmZ d dlmZ d dlmZ	m
Z
 d dlZ e¦   «         deddfd„¦   «         ZdS )	é    )ÚSequence)ÚresetÚskip_if_not_cluster)Ú	ClientAPI)ÚclientÚconfigNr   Úreturnc                 óŠ  — t          | ¦  «         |                      dddddœ¬¦  «        }d„ t          d¦  «        D ¦   «         }d„ t          d¦  «        D ¦   «         }|                     ||¬¦  «         |                     |d	         g¬
¦  «         t          j        ¦   «          t          j        ¦   «         }| 	                    dd¬¦  «        }t          |j        ¦  «        d	k    sJ ‚|j        }t          ¦   «         }|D ]O}|                     |j        j        ¦  «         |j        j        }	|j        j        }
|                     |	|
¦  «         ŒPd}t%          j        ¦   «         }	 | 	                    dd¬¦  «        }|j        }t          d„ |D ¦   «         ¦  «        }t          |¦  «        t          |¦  «        k    r't          |                     |¦  «        ¦  «        d	k    rn9t%          j        ¦   «         |z
  |k    r
J d¦   «         ‚t%          j        d¦  «         Œ·	 | 	                    dd¬¦  «        }|j        }d}|D ]}|j        j        dk    rd} nŒ|rn9t%          j        ¦   «         |z
  |k    r
J d¦   «         ‚t%          j        d¦  «         Œut%          j        d¦  «         |                     |d	         g¬
¦  «         d S )NÚtesté€   )zhnsw:construction_efzhnsw:search_efzhnsw:M)ÚnameÚmetadatac                 ó,   — g | ]}t          |¦  «        ‘ŒS © )Ústr©Ú.0Úis     úd/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/chromadb/test/distributed/test_reroute.pyú
<listcomp>z test_reroute.<locals>.<listcomp>   s   € Ð
%Ð
%Ð
%�a�3ˆq‰6Œ6Ð
%Ð
%Ð
%ó    é
   c                 óf   — g | ].}t          |¦  «        t          |¦  «        t          |¦  «        g‘Œ/S r   )Úfloatr   s     r   r   z test_reroute.<locals>.<listcomp>   s>   € ð )ð )ð )Ø+,�ˆq‰Œ•5˜‘8”8�U 1™XœXÐ&ð)ð )ð )r   )ÚidsÚ
embeddingsr   )Úquery_embeddingsÚchromazapp=query-service)Úlabel_selectorTc                 ó&   — g | ]}|j         j        ‘ŒS r   )r   Úuid)r   Úitems     r   r   z test_reroute.<locals>.<listcomp>3   s   € Ð;Ð;Ð;¨T�t”}Ô(Ð;Ð;Ð;r   Fz'Timed out waiting for new pods to starté   ÚRunningz*Timed out waiting for new pods to be ready)r   Úcreate_collectionÚrangeÚaddÚqueryr   Úload_kube_configÚ
k8s_clientÚ	CoreV1ApiÚlist_namespaced_podÚlenÚitemsÚsetr   r!   r   Ú	namespaceÚdelete_namespaced_podÚtimeÚintersectionÚsleepÚstatusÚphase)r   Ú
collectionr   r   Úv1Úresr.   Úseen_idsr"   r   r0   Útimeout_secsÚ
start_timeÚnew_idsÚreadys                  r   Útest_rerouter?      sõ  € õ 
ˆ&�M„M€MØ×)Ò)ØØ*-ÀÐPSÐTÐTð *ñ ô €Jð
 &Ð
%�5 ™9œ9Ð
%Ñ
%Ô
%€Cð)ð )Ý05°b±	´	ð)ñ )ô )€Jð ‡N‚N�s z€NÑ2Ô2Ð2Ø×Ò z°!¤} oÐÑ6Ô6Ð6õ ÔÑÔÐÝ	Ô	Ñ	Ô	€Bà
×
 Ò
  Ð:MÐ
 Ñ
NÔ
N€CÝˆsŒy‰>Œ>˜AÒÐÐÐØŒI€EÝ‰uŒu€Hð ð 2ð 2ˆØ�Š�T”]Ô&Ñ'Ô'Ð'ØŒ}Ô!ˆØ”MÔ+ˆ	Ø
× Ò   yÑ1Ô1Ð1Ð1ð €LÝ”‘”€JðØ×$Ò$ XÐ>QÐ$ÑRÔRˆØ”	ˆÝÐ;Ð;°UÐ;Ñ;Ô;Ñ<Ô<ˆÝˆw‰<Œ<�3˜x™=œ=Ò(Ð(­S°×1EÒ1EÀhÑ1OÔ1OÑ-PÔ-PÐTUÒ-UÐ-UØÝŒ9‰;Œ;˜Ñ# lÒ2Ð2ØCÐCÑCÔC�5ÝŒ
�1‰ŒˆððØ×$Ò$ XÐ>QÐ$ÑRÔRˆØ”	ˆØˆØð 	ð 	ˆDØŒ{Ô  IÒ-Ð-Ø�Ø�ð .ð ð 	ØÝŒ9‰;Œ;˜Ñ# lÒ2Ð2ØFÐFÑFÔF�5ÝŒ
�1‰Œˆðõ 	„Jˆq�M„M€MØ×Ò z°!¤} oÐÑ6Ô6Ð6Ð6Ð6r   )Útypingr   Úchromadb.test.conftestr   r   Úchromadb.apir   Ú
kubernetesr   r*   r   r2   r?   r   r   r   ú<module>rD      s¸   ðØ Ð Ð Ð Ð Ð ðð ð ð ð ð ð ð ð #Ð "Ð "Ð "Ð "Ð "Ø 3Ð 3Ð 3Ð 3Ð 3Ð 3Ð 3Ð 3Ø €€€ð ÐÑÔð>7Øð>7à	ð>7ð >7ð >7ñ Ôð>7ð >7ð >7r   