ZiddlmZmZmZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZejeZGdd ZdS) )ClassVarDictOptionalN) ServerAPI)BaseHTTPClient)SettingsSystem)ProductTelemetryClient)ClientStartEventceZdZUiZeeeefed<iZ eeee fed<e j Z ee j ed<eed<efdeddfdZed ededefd Zededefd Zed edefd Zed eddfdZed eddfdZed ede fdZed eddfdZeddZedefdZddZedeefdZdS)SharedSystemClient_identifier_to_system_identifier_to_refcount_refcount_lock _identifiersettingsreturnNct||_t|j|t|jdSN)r _get_identifier_from_settingsr_create_system_if_not_exists_increment_refcount)selfrs ^/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/api/shared_system_client.py__init__zSharedSystemClient.__init__sP.KKHUU778H(SSS..t/?@@@@@ identifiercH||jvrbt|}||j|<|t|t|n+|j|}|j|krtd|d|j|S)Nz)An instance of Chroma already exists for z with different settings)rr instancer rstartr ValueError)clsrr new_systemprevious_systems rrz/SharedSystemClient._create_system_if_not_existss S6 6 6))J4>C %j 1    6 7 7 7    * * *       !7 CO'833 d ddd(44rcd}|j}|td|dvr|jr|j}n:d}n7|dvr!t t j}ntd||S)Nz1Chroma API implementation must be set in settings)zchromadb.api.segment.SegmentAPIz!chromadb.api.rust.RustBindingsAPI ephemeral)zchromadb.api.fastapi.FastAPIz'chromadb.api.async_fastapi.AsyncFastAPIz&Unsupported Chroma API implementation )chroma_api_implr! is_persistentpersist_directorystruuiduuid4)rrapi_impls rrz0SharedSystemClient._get_identifier_from_settings3s +  PQQ Q    % %7      TZ\\**JJPhPPQQ Qrsystemcbt|j}|tj|<|Sr)r rrr)r/rs r_populate_data_from_systemz-SharedSystemClient._populate_data_from_systemOs+'EEfoVV ?E0<rcZt|||j}|S)zRCreate a client from an existing system. This is useful for testing and debugging.)r r1r)r"r/rs r from_systemzSharedSystemClient.from_systemUs. 55f===3v''rc|j5||jvr d|j|<|j|xxdz cc<ddddS#1swxYwYdS)z6Increment the reference count for a system identifier.rNrr)r"rs rrz&SharedSystemClient._increment_refcount]s  9 9!<<<:;+J7  ' 3 3 3q 8 3 3 3 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s)>AAc|j5||jvr>|j|xxdzcc<|j|}|dkr|j|=|cdddS ddddS#1swxYwYdS)zODecrement the reference count for a system identifier and return the new count.r5rNr6)r"rcounts r_decrement_refcountz&SharedSystemClient._decrement_refcountes   S888+J7771<7773J?A::3J?                           s;AAA"%A"c||}|dkr3|j|d}||dSdSdS)zDecrement refcount and stop the system if this was the last reference. This consolidates the "decrement + conditional stop" pattern used in both Client.close() and the Client.__init__ exception handler. rN)r9rpopstop)r"rrefcountr/s r_release_systemz"SharedSystemClient._release_systemqs\**:66 q==.22:tDDF!  =!!rc6it_it_dSr)r rrrrclear_system_cachez%SharedSystemClient.clear_system_cache~s35057222rc0tj|jSr)r rr)rs r_systemzSharedSystemClient._systems!78HIIrc|jt}|t dSr)rCrr capturer )rtelemetry_clients r_submit_client_start_eventz-SharedSystemClient._submit_client_start_events:<001GHH  !1!3!344444rc~g}ttj}|D]} |t }t |ts3|}d|vrd|vrP| }d}| D]!\}}| dkr|}n"|r| |#t$rYwxYw|sdSt|dkr,td|dddd |dS) a  Try to extract api key from existing client instances by checking httpx session headers. Requirements to pull api key: - must be a BaseHTTPClient instance (ignore RustBindingsAPI and SegmentAPI) - must have "api.trychroma.com" or "gcp.trychroma.com" in the _api_url (ignore local/self-hosted instances) - must have "x-chroma-token" or "X-Chroma-Token" in the headers Returns: The first api key found, or None if no client instances have api keys set. zapi.trychroma.comzgcp.trychroma.comNzx-chroma-tokenr5zAMultiple Chroma Cloud clients found, using API key starting with rz...)listr rvaluesrr isinstancer get_api_urlget_request_headersitemslowerappend Exceptionlenloggerinfo) api_keyssystems_snapshotr/ server_apiapi_urlheadersapi_keykeyvalues r%get_chroma_cloud_api_key_from_clientsz8SharedSystemClient.get_chroma_cloud_api_key_from_clientss! 2 H O O Q QRR&  F #__Y77 !*n==%0022'w66+7::$88::")--//JCyy{{&666"'7-OOG,,,     4 x==1   KKhT\]^T_`bab`bTchhh   {s/C$#C$A#C$$ C10C1)rN)__name__ __module__ __qualname__rrrr+r __annotations__rint threadingLockrrr classmethodr staticmethodrr1r3rr9r>rApropertyrCrGrr^r@rrr r s9;8Df$56;;;8:Xd38n5:::/=y~/?/?NHY^,???&XZZAAA AAAA55(05 555[5,S\66c\ ,@[9S9T999[9 S S   [      [ 888\8JJJJXJ555558C=555\555rr )typingrrrloggingrdr, chromadb.apirchromadb.api.base_http_clientrchromadb.configrr chromadb.telemetry.productr !chromadb.telemetry.product.eventsr getLoggerr_rTr r@rrrqs++++++++++ """"""888888,,,,,,,,======>>>>>>  8 $ $ssssssssssr