ZiWnddlmZmZddlmZddlmZddlmZddl Z ddl m Z m Z m Z ddlmZmZmZmZddlmZdd lmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,m-Z-dd l+m.Z.m/Z/ddl0m1Z1ddl2m3Z3m4Z4ddl5m6Z6m7Z7m8Z8m9Z9Gddee Z:Gddee Z;dS))OptionalSequence) TracebackType)UUID)overrideN)AdminAPI ClientAPI ServerAPI)CreateCollectionConfigurationUpdateCollectionConfiguration.validate_embedding_function_conflict_on_create+validate_embedding_function_conflict_on_get)SharedSystemClient)CollectionMetadata DataLoader Documents EmbeddableEmbeddingFunction Embeddings GetResultIDsIncludeLoadable Metadatas QueryResultSchemaURIsIncludeMetadataDocuments!IncludeMetadataDocumentsDistancesDefaultEmbeddingFunction) UserIdentity)maybe_set_tenant_and_database)SettingsSystem)DEFAULT_TENANTDEFAULT_DATABASE) Collection)ChromaAuthError ChromaError)DatabaseTenantWhere WhereDocumentc~eZdZUdZeZeed<eZ eed<e ed<e ed<dZ e ed<eeefdeedeeded d ffd Zeeeefd ededed dfd Zed efdZed efdZe dFdeedeed eefdZed efdZed d d ed dfdedeedeedee dee!e"dee#e$de d efdZ%eed fdedee!e"dee#e$d efdZ&ed d d ed fdedeedeedee dee!e"dee#e$d efdZ'e dGde(deed ee d!ee)d d f d"Z*eded d fd#Z+e dGd$e,d%e(d&e-d'ee.d(ee/d)ee0d e fd*Z1e dHd%e(d$e,d&ee-d'ee.d(ee/d)ee0d e fd+Z2e dGd%e(d$e,d&e-d'ee.d(ee/d)ee0d e fd,Z3ed%e(d efd-Z4edId%e(d/ed e5fd0Z6ed d d d d e7fd%e(d$ee,d1ee8deedeed2ee9d3e:d e5fd4Z; dFd%e(d$ee,d1ee8d2ee9d d f d5Zfd8Z?ed e fd9Z@ed efd:ZAed efd;ZBed efd<ZCeefdeded d fd=ZDeded d fd>ZEdJd?ZFdKd@ZGdAeeHeIdBeeIdCeeJd d fdDZKdeded d fdEZLxZMS)LClientaA client for Chroma. This is the main entrypoint for interacting with Chroma. A client internally stores its tenant and database and proxies calls to a Server API instance of Chroma. It treats the Server API and corresponding System as a singleton, so multiple clients connecting to the same resource will share the same API instance. Client implementations should be implement their own API-caching strategies. tenantdatabase_server _admin_clientF_closedsettingsreturnNct| |||_|||_|jt |_|}t||j ||\}}||td||td|r||_|r||_t |j|_||j|j|dS#t"$rIt%|drt'j|jjt'j|jwxYw)Nr5)4overwrite_singleton_tenant_database_access_from_authuser_provided_tenantuser_provided_databasez]Could not determine a tenant from the current authentication method. Please provide a tenant.zkCould not determine a database name from the current authentication method. Please provide a database name.r0r1r3)super__init__r0r1_systeminstancer r2get_user_identityr";chroma_overwrite_singleton_tenant_database_access_from_authr( AdminClient from_systemr3_validate_tenant_database_submit_client_start_event Exceptionhasattrr_release_system _identifier)selfr0r1r5 user_identity maybe_tenantmaybe_database __class__s P/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/api/client.pyr>zClient.__init__>s (+++- !$ # (  <00;;DL 2244M+HEMFJ%+'/ ,,, (L.#%s%(*:%B +*  / . "-!8!8!F!FD   * *$+ * V V V  + + - - - - -   t_-- S"243E3QRRR  .t/? @ @ @  s C6DAE0systemcTtj|||||j}|S)N)r0r1r5r_populate_data_from_systemr5)clsrQr0r1r@s rPrDzClient.from_systemts2 5f===3fx&/RRRc |jS#tj$rt dt $r}|d}~wt $r!}t t|d}~wwxYw)NACould not connect to a Chroma server. Are you sure it is running?)r2rAhttpx ConnectError ValueErrorr)rGstr)rKes rPrAzClient.get_user_identitys %<1133 3!   S    G % % %SVV$$ $ %s!'A2A A2A--A2c4|jS)z2Return the server time in nanoseconds since epoch.)r2 heartbeatrKs rPr_zClient.heartbeats|%%'''rVlimitoffsetclfdj||jjDS)zList collections for the current tenant and database, with pagination. Returns: Sequence[Collection]: Collection objects for the current tenant. c<g|]}tj|S))clientmodel)r'r2).0rfrKs rP z+Client.list_collections..s8    dl% 8 8 8   rVr<)r2list_collectionsr0r1)rKrarbs` rPrizClient.list_collectionssR    66vdkDM7    rVcN|j|j|jS)z9Return the number of collections in the current database.r<)r2count_collectionsr0r1r`s rPrkzClient.count_collectionss,|--;.   rVnameschema configurationmetadataembedding_function data_loader get_or_createc |i}|d}t||||||d<|j||||j|j||} t |j| ||S)aICreate a collection with optional configuration and metadata. If using a schema, do not provide `embedding_function`. Instead, provide the `embedding_function` as part of the schema. Args: name: Collection name. schema: Optional collection schema for indexes and encryption. configuration: Optional collection configuration. metadata: Optional collection metadata. embedding_function: Optional embedding function for the collection. data_loader: Optional data loader for documents with URIs. get_or_create: Whether to return an existing collection if present. Returns: Collection: The created collection. Raises: ValueError: If the embedding function conflicts with configuration. Nrp)rlrmror0r1rrrnrerfrprq)getr r2create_collectionr0r1r') rKrlrmrnrorprqrrconfiguration_efrfs rPrvzClient.create_collectionsB  M(,,-ABB6  0    ).>.F2DM. / ..;]''/  <1#     rVc|j||j|j}|jd}t ||t|j|||S)aGet a collection by name. Args: name: Collection name. embedding_function: Optional embedding function for the collection. data_loader: Optional data loader for documents with URIs. Returns: Collection: The requested collection. Raises: ValueError: If the embedding function conflicts with configuration. rlr0r1rprt)r2get_collectionr0r1configuration_jsonrurr')rKrlrprqrfpersisted_ef_configs rPrzzClient.get_collections, ++;],   $6::;OPP3  3   <1#     rVc>|i}|d}t||||||d<|j||||j|j|}|jd} t|| t|j|||S)a9Get an existing collection or create a new one. If the collection does not exist, it will be created. If the collection already exists, the schema, configuration, and metadata arguments will be ignored. Args: name: Collection name. schema: Optional collection schema for indexes and encryption. configuration: Optional collection configuration. metadata: Optional collection metadata. embedding_function: Optional embedding function for the collection. data_loader: Optional data loader for URI-backed data. Returns: Collection: The existing or newly created collection. Raises: ValueError: If the embedding function does not match the collection's embedding function. Nrp)rlrmror0r1rnrt) rur r2get_or_create_collectionr0r1r{rr') rKrlrmrnrorprqrwrfr|s rPr~zClient.get_or_create_collections@  M(,,-ABB6  0     ).>.F2DM. / 55;]' 6  $6::;OPP3  3   <1#     rVidnew_name new_metadatanew_configurationcV|j||j|j|||S)N)rr0r1rrr)r2_modifyr0r1)rKrrrrs rPrzClient._modifyVs9|##;]%/ $   rVcP|j||j|jS)Nry)r2delete_collectionr0r1rKrls rPrzClient.delete_collectiongs0 |--;].   rVids collection_id embeddings metadatas documentsurisc Z|j||j|j|||||S)N)rr0r1rrrrr)r2_addr0r1)rKrrrrrrs rPrz Client._addvs?|  ;]'!!   rVc Z|j||j|j|||||SN)rr0r1rrrrr)r2_updater0r1rKrrrrrrs rPrzClient._update?|##';]!$   rVc Z|j||j|j|||||Sr)r2_upsertr0r1rs rPrzClient._upsertrrVcP|j||j|jS)N)rr0r1)r2_countr0r1)rKrs rPrz Client._counts.|""';]#   rV ncR|j|||j|jS)N)rrr0r1)r2_peekr0r1)rKrrs rPrz Client._peeks1|!!';] "   rVwherewhere_documentincludec \|j||j|j|||||| S)N) rr0r1rrrarbrr)r2_getr0r1)rKrrrrarbrrs rPrz Client._getsB|  ';])!   rVcZ|j||j|j|||dS)N)rr0r1rrr)r2_deleter0r1)rKrrrrs rPrzClient._deletesC ';])      rVquery_embeddings n_resultsc \|j|||j|j||||| S)N) rrr0r1rrrrr)r2_queryr0r1)rKrrrrrrrs rPrz Client._querysB|""';]-)#   rVc4|jSN)r2resetr`s rPrz Client.resets|!!###rVc4|jSr)r2 get_versionr`s rPrzClient.get_versions|'')))rVc4|jSr)r2 get_settingsr`s rPrzClient.get_settings s|((***rVc4|jSr)r2get_max_batch_sizer`s rPrzClient.get_max_batch_sizes|..000rVcP|||||_||_dSNr<rEr0r1)rKr0r1s rP set_tenantzClient.set_tenants- &&fx&HHH   rVcL||j|||_dSrr)rKr1s rP set_databasezClient.set_databases( &&dkH&MMM  rVc|jrdSd|_t|drtj|jjtj|jdS)aClose the client and release all resources. This method decrements the reference count for the underlying System. When the last client using a shared System calls close(), the System is stopped and all resources (database connections, etc.) are released. This is particularly important for PersistentClient to avoid SQLite file locking issues. Note: If multiple clients share the same System (e.g., multiple PersistentClient instances with the same path), the System will only be stopped when the last client is closed. This allows safe use of context managers with multiple clients. Example: >>> client = chromadb.PersistentClient(path="./chroma_db") >>> # ... use client ... >>> client.close() Or using context manager: >>> with chromadb.PersistentClient(path="./chroma_db") as client: ... # ... use client ... NTr3)r4rHrrIr3rJr`s rPclosez Client.close#sb0 <  F  4 ) ) O  .t/A/M N N N *4+;<<<< classmethodrr$rDr!rAintr_rr'rirkr rr rrrrrrvrzr~rr rrrrrrrrrrrrrrr,r-rrrrrrrrrrrrrrtype BaseExceptionrrrE __classcell__rOs@rPr/r/+s !FC   $Hc$$$ GT !/"2%XZZ 44 43-4 4  444444l %(    X[ %< % % %X %(3(((X(CG   c]  3;C=  *     X   3   X $(AE15 % $ & &6:#< < <  <  => < -. < % j ) < j23< <  < < < X< | % $ & &6: % % % % j ) % j23 %  % % % X% N$(AE15 % $ & &6:> > >  >  => > -. > % j ) > j23>  > > > X> @#'59EI    3- 12  $$AB      X       X  *.)-#      I&  I&  tn     X ( ,0)-)-#    Z(  I&  I&  tn     X ( *.)-#      I&  I&  tn     X ( D S   X   4 C    X "!%# $263   c]   }    !/      X 4"&26    c]   !/        "!%26<   % c]      !/      X ,$t$$$X$*S***X*+h+++X+1C111X16F!!!!4!!!X! !S!T!!!X!"="="="=H4 ./-('   strVr/c eZdZUdZeed<efdeddffd Zee fde de ddfd Z ee fde de de fd Z ee fde de ddfd Zedde fd eed eede dee fdZede ddfdZede defdZeededdfdZxZS)rCz0Admin client for managing tenants and databases.r2r5r6Nct||jt|_dSr)r=r>r?r@r r2)rKr5rOs rPr>zAdminClient.__init__rs6 """|,,Y77 rVrlr0c:|j||S)zCreate a database in a tenant. Args: name: Database name. tenant: Tenant that owns the database. r)r2create_databaserKrlr0s rPrzAdminClient.create_databasev|++f+EEErVc:|j||S)zGet a database by name. Args: name: Database name. tenant: Tenant that owns the database. Returns: Database: The database record. r)r2rrs rPrzAdminClient.get_databases|((d6(BBBrVc:|j||S)zDelete a database by name. Args: name: Database name. tenant: Tenant that owns the database. r)r2delete_databasers rPrzAdminClient.delete_databaserrVrarbc<|j|||S)N)r0)r2list_databases)rKrarbr0s rPrzAdminClient.list_databasess!|**5&*HHHrVc8|j|SNr)r2 create_tenantrs rPrzAdminClient.create_tenants|))t)444rVc8|j|Sr)r2rrs rPrzAdminClient.get_tenants|&&D&111rVrQcPtj|||j}|S)Nr8rS)rUrQr@s rPrDzAdminClient.from_systems. 5f===3000rV)rrrrr rr#r>rr%r\rr*rrrrrrrr+rrr$rDrrs@rPrCrCms@:: ,4HJJ88848888887EFFCFF$FFFXF4B C C Cc Cx C C CX C7EFFCFF$FFFXF $ $$ II}I I I (  IIIXI5#5$555X52s2v222X2  X[rVrC)<typingrrtypesruuidr overridesrrY chromadb.apirr r %chromadb.api.collection_configurationr r r r!chromadb.api.shared_system_clientrchromadb.api.typesrrrrrrrrrrrrrrrrr chromadb.authr!chromadb.auth.utilsr"chromadb.configr#r$r%r&chromadb.api.models.Collectionr'chromadb.errorsr(r)chromadb.typesr*r+r,r-r/rCrrVrPrs%%%%%%%% 7777777777 A@@@@@&'&&&&&======,,,,,,,,<<<<<<<<55555588888888AAAAAAAAAAAA}}}}} }}}DCCCCC$hCCCCCrV