eidddlmZmZmZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZej"eZGdd Zy) )ClassVarDictOptionalN) ServerAPI)BaseHTTPClient)SettingsSystem)ProductTelemetryClient)ClientStartEventc eZdZUiZeeeefed<eed<e fde ddfdZ e dede defdZ e de defd Ze d edefd Ze d eddfd Ze dd ZedefdZddZe deefdZy)SharedSystemClient_identifier_to_system _identifiersettingsreturnNcxtj||_tj|j|yN)r _get_identifier_from_settingsr_create_system_if_not_exists)selfrs k/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/chromadb/api/shared_system_client.py__init__zSharedSystemClient.__init__s-.KKHU778H8H(S identifierc@||jvrUt|}||j|<|jt|jt|j n-|j|}|j |k7rtd|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 S66 6)J4>C % %j 1    6 7    *    !77 CO''83 ? |Kcd((44rcd}|j}| td|dvr|jr|j}|Sd}|S|dvrt t j }|Std|)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_settings/s ++  PQ Q   %%%77    TZZ\*JEhZPQ Qrsystemcjtj|j}|tj|<|Sr)r rrr)r,rs r_populate_data_from_systemz-SharedSystemClient._populate_data_from_systemKs-'EEfooV ?E00<rcTtj|||j}|S)zRCreate a client from an existing system. This is useful for testing and debugging.)r r.r)rr,rs r from_systemzSharedSystemClient.from_systemQs% 55f=v'rcit_yr)r rrrclear_system_cachez%SharedSystemClient.clear_system_cacheYs 350rc<tj|jSr)r rr)rs r_systemzSharedSystemClient._system]s!778H8HIIrct|jjt}|jt yr)r5rr capturer )rtelemetry_clients r_submit_client_start_eventz-SharedSystemClient._submit_client_start_eventas*<<001GH  !1!34rcg}ttjj}|D]} |j t }t |ts*|j}d|vrd|vrC|j}d}|jD]\}}|jdk(s|}n|r|j||syt|dkDrtj!d|dddd |dS#t$rYwxYw) 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-tokenzAMultiple 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_clientses+! 2 H H O O QR& F #__Y7 !*n=%002'w6+7:$88:")--/JCyy{&66"' OOG,1 : x=1  KKST\]^T_`babTcSddgh {  s#%C:C:/;C:+C:: DD)rN)__name__ __module__ __qualname__rrrr(r __annotations__rr classmethodr staticmethodrr.r0r3propertyr5r9rrQr2rrr r s+9;8Df$56;&ZTT T55(05 55,S66c ,@66JJJ558C=55rr )typingrrrloggingr) chromadb.apirchromadb.api.base_http_clientrchromadb.configrr chromadb.telemetry.productr !chromadb.telemetry.product.eventsr getLoggerrRrGr r2rrras>++ "8,=>   8 $NNr