codekingpro/portable-devtools
114k
1�
2 ��jV � �4 � S SK r S SKr S SKrS SKrS SKJr S SKrS SKJr S SK J3r4 S SK J r S SK
Jr S SKJrJr S SKJr S S KJr S S5KJr S SKrS S KJr S S \ S
\ S\ S\ S\ 46S jjr S\!\ SS4S jr"S S jr#S\SS4S jr$S\S\S\SS4S jr%S\SS4S jr&g) � N)�Event)�Client)�7Collection)�cli)�build_cli_args)�Settings�System)�get_sql)�SqliteDB)�Table)�8invariants�host�port� max_retries�
initial_delay�returnc � � Un[ U5 HV n [ R " XS9nUR 5 nUS:� a g XRS-9 : d M; [ R " U5 US- nMX g! [ a [ S5 NAf = f)a Wait for server to be ready using exponential backoff.10Args:11 client: ChromaDB client instance12 max_retries: Maximum number of retry attempts13 initial_delay: Initial delay in seconds before first retry14Returns:15 bool: True if server is ready, False if max retries exceeded16�r r r Tz!Heartbeat failed, trying again...� � F)�range�chromadb�17HttpClient� heartbeat� Exception�print�time�sleep)r r r r �delay�attempt�clientr s �\D:\code\apps\devtools\python\user_packages\Python313\site-packages\chromadb/test/test_cli.py�wait_for_serverr# s� � �
�E���%�� � �(�(�d�>�F� �(�(�*�I��1�}�� � �1�_� $� �J�J�u� � �Q�J�E� &� �� � � �5� 6� � �s �*A)�)B� B�argsc �D � U [ l [ R " 5 g )N)�sys�argvr �app)r$ s r" � start_appr) 2 s � ��C�H��G�G�I� c �L � SSS.n SS/nUR [ S 0 U D65 [ U5 [ R " [18 U4S9nUR
5 [ R " S5 [ SSS 9( d S195 eUR 5 UR 5 g ) N�chroma_test_dataiA )�pathr �chroma�run��targetr$ � � localhostr z4Server failed to start within maximum retry attempts� ) �extendr r �multiprocessing�Processr) �startr r r# � terminate�join)�kwargsr$ �server_processs r" �test_appr= 6 s� � �(�$�
7�F� �e� �D��K�K��(��(�)� �$�K�$�,�,�I�T�G�L�N������J�J�q�M� � �$� 7�o�9o�o� 7�� � � ����r* �sqlite_persistentc � � U nUR [ 5 nUR nUR SS5 UR U5 [20 R " U5 nUR S5 nUR S5 nS[ S[ SS 4S jnU" US 5 U" US215 UR 5 n[ S 5 n UR 5 R U 5 R S 5 n22[ U235 u p�UR! X�5 UR# 5 / :X d e S S S 5 S
SSUR$ R& S/[( l [, R. " 5 UR 5 n[ S 5 n UR 5 R U 5 R S 5 n24[ U255 u p�UR! X�5 UR# 5 n
[1 U
5 S:X d eU
S S S:X d e S S S 5 [3 US5 ( a U?UR R5 S5 R6 ( d e[8 R: " XU/S5 g ! , ( d f GNA= f! , ( d f Ny= f)N�automatically_purgeF� collection1� collection2�26collection�numr c � � [ U5 Vs/ s H n[ U5 PM nn[ R R US5 nU R X4S9 g s snf )Nr )�ids�27embeddings)r �str�np�random�rand�add)rC rD �irF rG s r" � add_records� test_vacuum.<locals>.add_recordsS sE � � $�S�z�*�z�!�s�1�v�z��*��Y�Y�^�^�C��+�28����3��6�� +s �A�d i� �maintenance_log�*r. �vacuum�--path�--forcer r r �configT) �instancer rV �
set_parameter�29set_configr � from_system�create_collectionr �int�txr � querybuilder�from_�selectr30 �execute�fetchall�settings�persist_directoryr& r' r r( �len�hasattr�
get_parameter�valuer
�log_size_below_max)r> �system�sqliterV r! rA rB rN �cur�t�q�sql�params�rowss r" � test_vacuumrr E s � �
�F�
�_�_�X�
&�F� �]�]�F�31���.��6�32���f� � �
�
��
'�F��*�*�=�9�K��*�*�=�9�K�7�33� 7�� 7�� 7�34 � �S�!�� �U�#� 35����� �#� $�� � � � !� '� '�� *� 1� 1�#� 6�� �a�j� �� � � �C� ��|�|�~��#�#�#� 36� �(�H�f�o�o�.O�.O�QZ�[�C�H��G�G�I� 37����� �#� $�� � � � !� '� '�� *� 1� 1�#� 6�� �a�j� �� � � �C� ��|�|�~���4�y�A�~� �~��A�w�q�z�X�%�%�%� 38� �v�x� � � �M� �=�=� &� &�'<� =� C� C�C� C� �!�!�&� �*D�d�K�9 39��� 40��s �4A.I �4BI/� 41I,�/42I=rc � ready_event�shutdown_eventc �6 � [ U S9nUR 5 UR [ 5 nUR 5 nUR S5 UR
5 UR 5 S S S 5 UR 5 g ! , ( d f N = f)N)rc z"INSERT INTO tenants DEFAULT VALUES) r r8 rW r r] ra �set�wait�stop)rc rs rt rj rk rl s r" � simulate_transactional_writery { sk � � �X�
&�F�43�L�L�N�
�_�_�X�
&�F� ����� � � �8�9�������� � 44�45 �K�K�M� 46��s �2B47�48 49Bc �T � [ R " S5 nUR 5 nUR 5 nUR [ U R50 X44S9nUR
5 UR 5 SSSU R51 R SSS/[ l52 [ R " 5 UR 5 nS UR R 5 ; d e UR! 5 UR# 5 g53! UR! 5 UR# 5 f = f) zVVacuum command should fail with details if there is a long-lived lock on the database.�spawnr0 r. rS rT rU z --timeout�10zFailed to vacuum ChromaN)r6 � get_contextr r7 ry rc r8 rw rd r& r' r r( �54readouterr�err�striprv r: )r> �capfd�ctxrs rt �process�captureds r" � test_vacuum_errors_if_lockedr� � s� � �55�56%�57%�g�58.�C��)�)�+�K��Y�Y�[�N��k�k�+� �(�(�+�
F� � �G� �M�M�O������ �h��2C�2L�2L�2^�2^�`i�kv�x|�}��� ��� ��#�#�%��(�H�L�L�,>�,>�,@�@�@�@���� �� � ��� ��� �� � ��s
�8A+D �"D')r2 g �?)r N)'r6 �multiprocessing.contextr& r �multiprocessing.synchronizer r �chromadb.api.clientr � chromadb.api.models.Collectionr � chromadb.clir �chromadb.cli.clir �chromadb.configr r �chromadb.db.baser59 �chromadb.db.impl.sqliter �pypikar �numpyrI �chromadb.test.propertyr
rH r\ �float�boolr# �listr) r= rr ry r� r4 r* r" �<module>r� s� �� � � 60� � -� � &� 5� � +� ,� $� ,� � � -�61 25���� ����).�� 62��:�D��I� �$� � � 3L�6� 3L�d� 3L�l �� �%*� �<A� � � � �F� �d� r*