codekingpro/portable-devtools
114k
1�
2 ��j� � � � S SK JrJrJrJrJr S SKrS SKr S SK3J r J r S\\\R S\\\\\ \\R 4 4S jrS\\\4 S\ S \SS4S4 jrS\\\4 S \ SS4S jrS\ S\S
4S jrS \ SS
4S jrg)� )�List�Dict�Any�Optional�UnionN)� QueryResult� GetResult�5embeddings�returnc � � U c g[ U S [ R 5 ( a U Vs/ s H oR 5 PM sn$ U $ s snf )z�6Transform embeddings from numpy arrays to lists of floats.7This is a shared helper function to avoid duplicating the transformation logic.8Nr )�9isinstance�np�ndarray�tolist)r10 �embs �\D:\code\apps\devtools\python\user_packages\Python313\site-packages\chromadb/utils/results.py�_transform_embeddingsr sN � � ��� �j��m�R�Z�Z� 0� 0� ",�,��#��� ��,�� ���,s �A � data_dict� query_result� query_idxc �� � US Hg nUR U5 nUc M UR S5 nUS:X a [ U5 n[ U[ 5 ( a [ U5 S:� a XB nX@U' Mi g)a$ 11Helper function to add fields from a query result to a dictionary.12Handles the nested array structure specific to query results.13 14Args:15 data_dict: Dictionary to add the fields to16 query_result: QueryResult containing the data17 query_idx: Index of the current query being processed18�includedN�sr19 r )�get�rstripr r
�list�len)r r r �field�value�keys r �_add_query_fieldsr! so � � �j�)�� � � ��'�� � ��,�,�s�#�C�� �$�-�e�4���%��&�&�3�u�:��>� �(�� "�c�N� *� �20get_resultc � � US H? nUR U5 nUc M UR S5 nUS:X a [ U5 nX0U' MA g)z�21Helper function to add fields from a get result to a dictionary.22Handles the flat array structure specific to get results.23 24Args:25 data_dict: Dictionary to add the fields to26 get_result: GetResult containing the data27r Nr r28 )r r r )r r# r r r s r �_add_get_fieldsr% 0 sN � � �J�'�����u�%�� � ��,�,�s�#�C�� �$�-�e�4�� "�c�N�
(r" z pd.DataFramec � � SSK n/ n[ U S 5 n[ U5 HO n0 nU S U US' [ XPU5 UR29 " U5 nUR
SSS9 UR U5 MQ U$ ! [ a [ S5 ef = f)a 30Function to convert QueryResult to list of DataFrames.31Handles the nested array structure specific to query results.32Column order is defined by the order of the fields in the QueryResult.33 34Args:35 query_result: QueryResult to convert to DataFrames.36 37Returns:38 List of DataFrames.39r Nz:pandas is required to convert query results to DataFrames.�ids�idT��inplace)�pandas� ImportErrorr �ranger! � DataFrame� set_index�append)r �pd�dfs� num_queries�i� data_for_df�dfs r �query_result_to_dfsr7 E s� � �X��
�C��l�5�)�*�K�
�;�
��&(� � (�� /�� 2� �D���+�Q�7�
�\�\�+�
&��40� � �T�4� �(� �41�42�2�� � �J�� � X��V�W�W�X�s �A6 �6B c � � SSK n0 nU S US' [ X 5 UR " U5 nUR SSS9 U$ ! [ a [ S5 ef = f)a 43Function to convert GetResult to a DataFrame.44Handles the flat array structure specific to get results.45Column order is defined by the order of the fields in the GetResult.46 47Args:48 get_result: GetResult to convert to a DataFrame.49 50Returns:51 DataFrame.52r Nz9pandas is required to convert get results to a DataFrame.r' r( Tr) )r+ r, r% r. r/ )r# r1 r5 r6 s r �get_result_to_dfr9 e sk � �W�� #%�K�"�5�)�K����K�,� � � �k� "�B��L�L��t�L�$�
�I�� � W��U�V�V�W�s �? �A)�typingr r r r r �numpyr r+ r1 �chromadb.api.typesr r r �floatr �str�intr! r% r7 r9 � r" r �<module>rA s� �� 3� 3� � � 5�
���b�j�j�)�*�
�
�e�D��e��%�t�B�J�J�'7�7�8�9�
� #��C��H�~�#� �#� �#� 53� #�2#��C��H�~�#��#� 54�#�* �k� �d�>�6J� �@�� �~� r"