Team Ai
Datasetpublic

codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face
1likes14kdownloads
async_client.cpython-313.pyc306 linesDownload Raw Back to __pycache__
1�

2��jGI��L�SSKrSSKJrJr SSKJr SSKJr SSKJ	r	 SSK3Jr SSKJ
r
JrJr SSKJrJrJrJr SS	KJr SS4KJr SSKJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+ SSK,J-r-J.r.J/r/J0r0 SS
K1J2r2 SSK3J4r4J5r5J6r6J7r7 "SS\\5r8"SS\\
5r9g)�N)�Optional�Sequence)�UUID)�override)�UserIdentity)�maybe_set_tenant_and_database)�
AsyncAdminAPI�AsyncClientAPI�AsyncServerAPI)�CreateCollectionConfiguration�UpdateCollectionConfiguration�.validate_embedding_function_conflict_on_create�+validate_embedding_function_conflict_on_get)�AsyncCollection)�SharedSystemClient)�CollectionMetadata�5DataLoader�	Documents�6Embeddable�EmbeddingFunction�7Embeddings�	GetResult�IDs�Include�IncludeMetadataDocuments�!IncludeMetadataDocumentsDistances�Loadable�	Metadatas�QueryResult�Schema�URIs�DefaultEmbeddingFunction�DeleteResult)�DEFAULT_DATABASE�DEFAULT_TENANT�Settings�System)�ChromaError)�Database�Tenant�Where�
WhereDocumentc��\rSrSr%Sr\\S'\r\	\S'\8r\	\S'\\S'\
\\9\"54S\	S\	S\SS4S	jj5r\
\\104S11\S\	S\	SS4Sjj5r\
\S12\SS4S
j55r\S\4Sj5r\\134S\	S\	SS4Sjj5r\S\	SS4Sj5rS\	S\	SS4Sjr\S\4Sj5r\SDS\\S\\S\\4Sjj5r\S\4Sj5r\SSS\ "5SS4S\	S\\!S\\"S\\#S\\$\%S\\&\'S\(S\4S jj5r)\\ "5S4S\	S\\$\%S\\&\'S\4S!jj5r*\\ "5S4S"\+S\\$\%S\\&\'S\4S#jj5r,\SSS\ "5S4S\	S\\!S\\"S\\#S\\$\%S\\&\'S\4S$jj5r-\SES"\+S%\\	S&\\#S'\\.SS414S(jj5r/\S\	SS4S)j5r0\SES*\1S+\+S,\2S-\\3S.\\4S/\\5S\(4S0jj5r6\SFS+\+S*\1S,\\2S-\\3S.\\4S/\\5S\(4S1jj5r7\SES+\+S*\1S,\2S-\\3S.\\4S/\\5S\(4S2jj5r8\S+\+S\4S3j5r9\SGS+\+S5\S\:4S6jj5r;\SSSSS\<4S+\+S*\\1S7\\=S\\S\\S8\\>S9\?S\:4S:jj5r@SES+\+S*\\1S7\\=S8\\>S\\S\A4S;jjrB\SS4SS\C4S+\+S<\2S*\\1S=\S7\\=S8\\>S9\?S\D4S>jj5rE\S\(4S?j5rF\S\	4S@j5rG\S\4SAj5rH\S\4SBj5rISCrJg)H�AsyncClient�*a�A client for Chroma. This is the main entrypoint for interacting with Chroma.15A client internally stores its tenant and database and proxies calls to a16Server API instance of Chroma. It treats the Server API and corresponding System17as a singleton, so multiple clients connecting to the same resource will share the18same API instance.19 20Client implementations should be implement their own API-caching strategies.21�
_admin_client�tenant�database�_server�settings�returnc��# �U"US9n[R"UR5 XlX$lURR[5UlUR5IShv�Nn[UURUUS9upgU(aXdlU(aXtl[RUR5Ul
URURURS9IShv�N UR5 U$N�N7f)N�r4)�4overwrite_singleton_tenant_database_access_from_auth�user_provided_tenant�user_provided_database�r1r2)r�_populate_data_from_system�_systemr1r2�instancerr3�get_user_identityr�;chroma_overwrite_singleton_tenant_database_access_from_auth�AsyncAdminClient�from_systemr0�_validate_tenant_database�_submit_client_start_event)�clsr1r2r4�self�
user_identity�maybe_tenant�maybe_databases        �_D:\code\apps\devtools\python\user_packages\Python313\site-packages\chromadb/api/async_client.py�create�AsyncClient.create<s�����H�%���5�5�d�l�l�C��� �
��|�|�,�,�^�<���"�4�4�6�6�
�'D��AI�BF�BF�!'�#+�	(22�$���&�K��*�M�-�9�9�$�,�,�G����,�,�D�K�K�$�-�-�,�X�X�X��'�'�)���%7�	Y�s%�A*D	�,D�-BD	�.D�/D	�D	�systemc��^# �[RX#UR5IShv�N$N7f)zRCreate a client from an existing system. This is useful for testing and debugging.N)r.rKr4)rErMr1r2s    rJ�from_system_async�AsyncClient.from_system_asyncas$���!�'�'��&�/�/�J�J�J�J�s�$-�+�-rc��[S5e)zNAsyncClient cannot be created synchronously. Use .from_system_async() instead.)�NotImplementedError)rErMs  rJrB�AsyncClient.from_systemls��"�\�23�	24�c��R# �URR5IShv�N$N7f�N)r3r?�rFs rJr?�AsyncClient.get_user_identityws����\�\�3�3�5�5�5�5���'�%�'Nc��V# �URXS9IShv�N XlX lgN7f�Nr;�rCr1r2)rFr1r2s   rJ�25set_tenant�AsyncClient.set_tenant{s,����,�,�F�,�N�N�N��� �
�	O�s�)�'�)c��`# �URURUS9IShv�N XlgN7fr[r\)rFr2s  rJ�set_database�AsyncClient.set_database�s-����,�,�D�K�K�(�,�S�S�S� �
�	T�s�.�,�.c��v# �URRUS9IShv�N URRX!S9IShv�N gN'![Ra [	S5e[26anUeSnAf[a [	SUS35ef=fNQ![Ra [	S5ef=f7f)N��namezACould not connect to a Chroma server. Are you sure it is running?zCould not connect to tenant z. Are you sure it exists?�rdr1)r0�27get_tenant�httpx�ConnectError�28ValueErrorr(�	Exception�get_database)rFr1r2�es    rJrC�%AsyncClient._validate_tenant_database�s����	��$�$�/�/�V�/�<�<�<�	��$�$�1�1�x�1�O�O�O�
=���!�!�	��S��
��	��G���	��.�v�h�6O�P��
�	��
P���!�!�	��S��
�	�s[�B9�A29�A�A30�B�B�B�B9�A31�32(B�2A4�4B�B9�B� B6�6B9c��R# �URR5IShv�N$N7frV)r3�	heartbeatrWs rJro�AsyncClient.heartbeat�s����\�\�+�+�-�-�-�-�rY�limit�offsetc���# �URRXURURS9IShv�NnUVs/sHn[	URUS9PM sn$N)s snf7f)Nr;)�client�model)r3�list_collectionsr1r2r)rFrqrr�modelsrus     rJrv�AsyncClient.list_collections�sc����|�|�4�4��$�+�+��
�
�5�33�34��PV�V�v�e��t�|�|�5�A�v�V�V�35��W�s �3A&�A�A&�A!�A&�!A&c��z# �URRURURS9IShv�N$N7fr[)r3�count_collectionsr1r2rWs rJrz�AsyncClient.count_collections�s8����\�\�3�3��;�;����4�36�37�	38�39�s�2;�9�;Frd�schema�
configuration�metadata�embedding_function�data_loader�
get_or_createc40��# �Uc0nURS5n[XX5 UbUcXSS'URRUUUUURUR41US9IShv�Nn	[
URU	UUS9$N7f)Nr)rdr|r}r~r1r2r��rtrurr�)�getrr3�create_collectionr1r2r)42rFrdr|r}r~rr�r��configuration_efrus43          rJr��AsyncClient.create_collection�s����� ��M�(�,�,�-A�B��6��	44��)�.>�.F�2D�.�/��l�l�4�4���'���;�;��]�]�'�5�45�46����<�<��1�#�	47�	48�49�s�A"B�$A?�%Bc���# �URRUURURS9IShv�NnURRS5n[
X%5 [URUUUS9$N@7f)N�rdr1r2rr�)r3�get_collectionr1r2�configuration_jsonr�rr)rFrdrr�ru�persisted_ef_configs      rJr��AsyncClient.get_collection�s����l�l�1�1���;�;��]�]�2�50�51��52$�6�6�:�:�;O�P��3��	53���<�<��1�#�	54�	55�56���3A8�A6�AA8�idc���# �URRUURURS9IShv�NnURRS5n[
X%5 [URUUUS9$N@7f)aXGet a collection by its ID.57 58Args:59    id: The UUID of the collection.60    embedding_function: Optional embedding function for the collection.61    data_loader: Optional data loader for documents with URIs.62 63Returns:64    AsyncCollection: The requested collection.65 66Raises:67    ValueError: If the embedding function conflicts with configuration.68)�
collection_idr1r2Nrr�)r3�get_collection_by_idr1r2r�r�rr)rFr�rr�rur�s      rJr�� AsyncClient.get_collection_by_id�s���,�l�l�7�7���;�;��]�]�8�69�70��71$�6�6�:�:�;O�P��3��	72���<�<��1�#�	73�	74�75�r�c	��P# �Uc0nURS5n[XW5 UbUcXSS'URRUUUUURUR76S9IShv�NnURRS5n	[XY5 [URUUUS9$N@7f)Nr)rdr|r}r~r1r2r�)	r�rr3�get_or_create_collectionr1r2r�rr)77rFrdr|r}r~rr�r�rur�s78          rJr��$AsyncClient.get_or_create_collections����� ��M�(�,�,�-A�B��6��	79��)�.>�.F�2D�.�/��l�l�;�;���'���;�;��]�]�
<�80�81��$�6�6�:�:�;O�P��3��	82���<�<��1�#�	83�	84�85�s�A!B&�#B$�$AB&�new_name�new_metadata�new_configurationc	��# �URRUUUUURURS9IShv�N$N7f)N)r�r�r�r�r1r2)r3�_modifyr1r2)rFr�r�r�r�s     rJr��AsyncClient._modifyLsG����\�\�)�)���%�/��;�;��]�]�
*�86�87�	88�89�s�6?�=�?c��|# �URRUURURS9IShv�N$N7f)Nr�)r3�delete_collectionr1r2�rFrds  rJr��AsyncClient.delete_collection]s>���90�\�\�3�3���;�;��]�]�4�91�92�	93�94�s�3<�:�<�idsr��95embeddings�	metadatas�	documents�urisc��# �URRUUUUUUURURS9IShv�N$N7f)N)r�r�r�r�r�r�r1r2)r3�_addr1r2)rFr�r�r�r�r�r�s       rJr��AsyncClient._addlsM����\�\�&�&��'�!�����;�;��]�]�'�	96�	97�		98�	99���8A�?�Ac��# �URRUUUUUUURURS9IShv�N$N7f�N)r�r�r�r�r�r�r1r2)r3�_updater1r2�rFr�r�r�r�r�r�s       rJr��AsyncClient._update��M����\�\�)�)�'��!�����;�;��]�]�*�	100�	101�		102�	103�r�c��# �URRUUUUUUURURS9IShv�N$N7fr�)r3�_upsertr1r2r�s       rJr��AsyncClient._upsert�r�r�c��P# �URRUS9IShv�N$N7f)N)r�)r3�_count)rFr�s  rJr��AsyncClient._count�s.����\�\�(�(�'�)�104�105�	106�107���&�$�&�108�nc��R# �URRUUS9IShv�N$N7f)N)r�r�)r3�_peek)rFr�r�s   rJr��AsyncClient._peek�s1����\�\�'�'�'��(�109�110�	111�112�rY�where�where_document�includec��# �URRUUUUUUUURURS9	IShv�N$N7f)N)	r�r�r�rqrrr�r�r1r2)r3�_getr1r2)rFr�r�r�rqrrr�r�s        rJr��AsyncClient._get�sP����\�\�&�&�'�����)���;�;��]�]�'�113114�115116�117	118�119120���9A�A�Ac121��# �URRUUUUUURURS9IShv�N$N7f)N)r�r�r�r�rqr1r2)r3�_deleter1r2)rFr�r�r�r�rqs      rJr��AsyncClient._delete�sJ����\�\�)�)�'���)���;�;��]�]�*�122�123�	124�125�s�7A�>�A�query_embeddings�	n_resultsc��# �URRUUUUUUUURURS9	IShv�N$N7f)N)	r�r�r�r�r�r�r�r1r2)r3�_queryr1r2)rFr�r�r�r�r�r�r�s        rJr��AsyncClient._query�sP����\�\�(�(�'�-����)���;�;��]�]�)�126127�128129�130	131�132133�r�c��R# �URR5IShv�N$N7frV)r3�resetrWs rJr��AsyncClient.reset�s����\�\�'�'�)�)�)�)�rYc��R# �URR5IShv�N$N7frV)r3�get_versionrWs rJr��AsyncClient.get_version�s����\�\�-�-�/�/�/�/�rYc�6�URR5$rV)r3�get_settingsrWs rJr��AsyncClient.get_settingss���|�|�(�(�*�*rTc��R# �URR5IShv�N$N7frV)r3�get_max_batch_sizerWs rJr��AsyncClient.get_max_batch_sizes����\�\�4�4�6�6�6�6�rY)r0r3r2r1)NN)NNN)NNNN)r�)K�__name__�134__module__�__qualname__�__firstlineno__�__doc__r	�__annotations__r%r1�strr$r2r�classmethodr&rKr'rOrrBrr?r]r`rC�introrrrrvrzr"r rrrrrr�boolr�r�rr�r�r
r�r�rrrrr!r�r�r�r�rr�rr+r,rr�r#r�rrr�r�r�r�r��__static_attributes__�rTrJr.r.*s����!� � �F�C� �$�H�c�$�
���%�(�%�Z�	"��"��"��	"�135136�"��"�H�137%�(�	K��K��K��	K�138139�K��K��
�140��141�142�143���144��6��6��6��<L�!�s�!�c�!�QU�!��!�145�!�3�!�4�!��!��c��S��T��0�.��.��.��CG�W��c�]�W�3;�C�=�W�	�/�	"�W��W��146��147��148�149�$(�AE�15�
%�&�6:�#�(150��(151��� �(152� � =�>�	(153�154�-�.�(155�%��j�)�156�
(157��j��2�3�(158��(159�160�(161��(162�T�
%�&�6:�
163��164�%��j�)�165�166��j��2�3�
167�168�169��170�4�
%�&�6:�
%171��%172�%��j�)�173�%174��j��2�3�
%175�176�%177��%178�N�$(�AE�15�
%�&�6:�*179��*180��� �*181� � =�>�	*182�183�-�.�*184�%��j�)�185�
*186��j��2�3�*187�188�*189��*190�X�#'�59�EI�191��192��3�-�193��1�2�	194�195$�$A�B�196�197�
198��199� �200��201�202�203��204��*.�)-�#�205�
�206��207��	208�209�I�&�210��I�&�
211��t�n�212�213�214��215�(�216,0�)-�)-�#�217��218��219��Z�(�	220�221�I�&�222��I�&�
223��t�n�224�225�226��227�(�*.�)-�#�228��229��230��	231�232�I�&�233��I�&�
234��t�n�235�236�237��238�(�239�$�240�3�241��242�243�244��245�#�246�y�247��248��"�!%�#� $�26�3�249��250��c�]�251����	252�253��}�254���
�
255�!��/�256��257�258�259��260�4"&�26�#�
261��262��c�]�263����	264�265!��/�266���}�
267�268�269�$�270"��!%�26�<�271��272�%�273��c�]�	274�275�276����
277�!��/�278��279�280�281��282�,�*�T�*��*��0�3�0��0��+�h�+��+��7�#�7��7rTr.c�`^�\rSrSr%\\S'\"54S\SS4U4Sjjjr\\	4S\283S\284SS4S	jj5r\\	4S\285S\286S\4S287jj5r
\\	4S\288S\289SS4Sjj5r\SS\	4S\\S
\\S\290S\\4Sjj5r\S\291SS4Sj5r\S\292S\4Sj5r\\S\SS4Sj55rSrU=r$)rAir3r4r5Nc�l>�[TU]U5 URR[5UlgrV)�super�__init__r=r>rr3)rFr4�	__class__s  �rJr��AsyncAdminClient.__init__s&���
����"��|�|�,�,�^�<��rTrdr1c��P# �URRXS9IShv�N$N7f�Nre)r3�create_database�rFrdr1s   rJr�� AsyncAdminClient.create_database�#����\�\�1�1�t�1�K�K�K�K�r�c��P# �URRXS9IShv�N$N7fr�)r3rkr�s   rJrk�AsyncAdminClient.get_databases#����\�\�.�.�D�.�H�H�H�H�r�c��P# �URRXS9IShv�N$N7fr�)r3�delete_databaser�s   rJr�� AsyncAdminClient.delete_databaser�r�rqrrc��R# �URRXUS9IShv�N$N7f)N)rqrrr1)r3�list_databases)rFrqrrr1s    rJr��AsyncAdminClient.list_databasess2����\�\�0�0��v�1�293�294�	295�296�rYc��P# �URRUS9IShv�N$N7f�Nrc)r3�
create_tenantr�s  rJr��AsyncAdminClient.create_tenant)s#����\�\�/�/�T�/�:�:�:�:�r�c��P# �URRUS9IShv�N$N7fr�)r3rfr�s  rJrf�AsyncAdminClient.get_tenant-s#����\�\�,�,�$�,�7�7�7�7�r�rMc�R�[R"U5 U"URS9nU$)Nr7)rr<r4)rErMr>s   rJrB�AsyncAdminClient.from_system1s%��	�5�5�f�=�����0���rT)r3)r�r�r�r�rr�r&r�rr%r�r�r)rkr�rr�rr�r�r*rfr�r'rBr��
__classcell__)r�s@rJrArAs���
��,4�J�=��=�4�=�=��=K�L�#�L�s�L�PT�L��L��:H�I�s�I�C�I�X�I��I��=K�L�#�L�s�L�PT�L��L�� $� $�$�	297���}�298���
�299��	300�301302�(�	�303��304��;��;��;��;��8�S�8�V�8��8��
����305����rTrA):rg�typingrr�uuidr�	overridesr�
chromadb.authr�chromadb.auth.utilsr�chromadb.apir	r306r�%chromadb.api.collection_configurationrr
rr�#chromadb.api.models.AsyncCollectionr�!chromadb.api.shared_system_clientr�chromadb.api.typesrrrrrrrrrrrrrrr r!r"r#�chromadb.configr$r%r&r'�chromadb.errorsr(�chromadb.typesr)r*r+r,r.rAr�rTrJ�<module>r
s|���%���&�=�F�F���@�@������(O�N�'�A�A�\7�$�n�\7�B.�)�=�.rT
codekingpro/portable-devtools · Team Ai