Ë
    lQêjŠ	  ã                   óh   — 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y)	é    )ÚSequence)ÚresetÚskip_if_not_cluster)Ú	ClientAPI)ÚclientÚconfigNr   Úreturnc                 ó  — t        | «       | j                  dddddœ¬«      }t        d«      D �cg c]  }t        |«      ‘Œ }}t        d«      D �cg c]"  }t	        |«      t	        |«      t	        |«      g‘Œ$ }}|j                  ||¬«       |j                  |d   g¬«       t        j                  «        t        j                  «       }|j                  d	d
¬«      }t        |j                  «      dkD  sJ ‚|j                  }t        «       }|D ]e  }	|j                  |	j                  j                   «       |	j                  j"                  }
|	j                  j$                  }|j'                  |
|«       Œg d}t)        j(                  «       }	 |j                  d	d
¬«      }|j                  }t        |D �	cg c]  }	|	j                  j                   ‘Œ c}	«      }t        |«      t        |«      k(  rt        |j+                  |«      «      dk(  rn7t)        j(                  «       |z
  |kD  rJ d«       ‚t)        j,                  d«       Œ¸	 |j                  d	d
¬«      }|j                  }d}|D ]  }	|	j.                  j0                  dk7  sŒd} n |rn7t)        j(                  «       |z
  |kD  rJ d«       ‚t)        j,                  d«       Œt)        j,                  d«       |j                  |d   g¬«       y c c}w c c}w c c}	w )NÚtesté€   )zhnsw:construction_efzhnsw:search_efzhnsw:M)ÚnameÚmetadataé
   )ÚidsÚ
embeddingsr   )Úquery_embeddingsÚchromazapp=query-service)Úlabel_selectorTFz'Timed out waiting for new pods to starté   ÚRunningz*Timed out waiting for new pods to be ready)r   Úcreate_collectionÚrangeÚstrÚfloatÚaddÚqueryr   Úload_kube_configÚ
k8s_clientÚ	CoreV1ApiÚlist_namespaced_podÚlenÚitemsÚsetr   Úuidr   Ú	namespaceÚdelete_namespaced_podÚtimeÚintersectionÚsleepÚstatusÚphase)r   Ú
collectionÚir   r   Úv1Úresr"   Úseen_idsÚitemr   r%   Útimeout_secsÚ
start_timeÚnew_idsÚreadys                   úk/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/test/distributed/test_reroute.pyÚtest_rerouter7      sº  € ô 
ˆ&„MØ×)Ñ)ØØ*-ÀÐPSÑTð *ó €Jô
 ! ›9Ö
%�aŒ3ˆq�6Ð
%€CÐ
%ä05°b³	ö)Ø+,Œˆq‹”5˜“8œU 1›XÒ&ð)€Jð )ð ‡N�N�s z€NÔ2Ø×Ñ z°!¡} oÐÔ6ô ×ÑÔÜ	×	Ñ	Ó	€Bà
×
 Ñ
  Ð:MÐ
 Ó
N€CÜˆs�y‰y‹>˜AÒÐÐØ�I‰I€EÜ‹u€Hð ò 2ˆØ�‰�T—]‘]×&Ñ&Ô'Ø�}‰}×!Ñ!ˆØ—M‘M×+Ñ+ˆ	Ø
× Ñ   yÕ1ð	2ð €LÜ—‘“€JØ
Ø×$Ñ$ XÐ>QÐ$ÓRˆØ—	‘	ˆÜ°UÖ;¨T�t—}‘}×(Ó(Ò;Ó<ˆÜˆw‹<œ3˜x›=Ò(¬S°×1EÑ1EÀhÓ1OÓ-PÐTUÒ-UØÜ�9‰9‹;˜Ñ# lÒ2ØCÐCÓC�5Ü�
‰
�1Œð ð Ø×$Ñ$ XÐ>QÐ$ÓRˆØ—	‘	ˆØˆØò 	ˆDØ�{‰{× Ñ  IÓ-Ø�Ùð	ñ ØÜ�9‰9‹;˜Ñ# lÒ2ØFÐFÓF�5Ü�
‰
�1Œð ô 	‡J�Jˆq„MØ×Ñ z°!¡} oÐÕ6ùòk &ùò)ùò: <s   °K3Á'K8Æ7K=)Útypingr   Úchromadb.test.conftestr   r   Úchromadb.apir   Ú
kubernetesr   r   r   r'   r7   © ó    r6   ú<module>r>      s?   ðÝ ÷õ #ß 3Û ñ Óð>7Øð>7à	ò>7ó ñ>7r=   