o g,@sddlmZmZddlZddlmZddlmZddl m Z ddl m Z ddlZddlmZmZmZddlmZmZmZdd lmZdd lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%gd Z&e'e(Z)ea*d Z+d e,fddZ-e-Z.dZ/z ddl0m1Z1e1Z/Wn e2ydZ/Ynwe/sddl3Z3e3j4dkre.rddl5Z5ddl6Z6e57e6j8ddddge9de6j:;de6j:d<neddZ=d efddZ>deefdeede?d e?d efd!d"Z@d#deefd$e?deede?d e?d ef d%d&ZAd'd(dddeefd)e?d*eBd+e,d,eee?e?fdeede?d e?d efd-d.ZCd'd(dddeefd)e?d*eBd+e,d,eee?e?fdeede?d e?d efd/d0ZD  d?d1d(d2d3de?d e?d4ee?deed5e?d6eBd7e,d efd8d9ZEt*eefdede?d e?d efd:d;Zefded efd= 3.35.0. Please visit https://docs.trychroma.com/troubleshooting#sqlite to learn how to upgrade.cKstjjdi|adS)zGOverride Chroma's default settings, environment variables or .env filesNr)chromadbconfigr __settings)kwargsrrr configure^sr-cCstS)N)r+rrrr get_settingsdsr.settingstenantdatabasecCs2|durt}d|_t|}t|}t|||dS)aA Creates an in-memory instance of Chroma. This is useful for testing and development, but not recommended for production use. Args: tenant: The tenant to use for this client. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. NFr/r0r1)r is_persistentstr ClientCreatorr2rrr EphemeralClienths  r6z./chromapathcCs8|durt}||_d|_t|}t|}t|||dS)a Creates a persistent instance of Chroma that saves to disk. This is useful for testing and development, but not recommended for production use. Args: path: The directory to save Chroma's data to. Defaults to "./chroma". tenant: The tenant to use for this client. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. NTr0r1r/)r persist_directoryr3r4r5)r7r/r0r1rrr PersistentClientsr: localhosti@hostportsslheaderscCs|durt}t|}t|}t|}t|}t|}d|_|jr2|j|kr2td|jd|d||_|jrI|j|krItd|jd|d||_||_||_ t |||dS)a Creates a client that connects to a remote Chroma server. This supports many clients connecting to the same server, and is the recommended way to use Chroma in production. Args: host: The hostname of the Chroma server. Defaults to "localhost". port: The port of the Chroma server. Defaults to "8000". ssl: Whether to use SSL to connect to the Chroma server. Defaults to False. headers: A dictionary of headers to send to the Chroma server. Defaults to {}. settings: A dictionary of settings to communicate with the chroma server. tenant: The tenant to use for this client. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. Nchromadb.api.fastapi.FastAPI(Chroma server host provided in settings[3] is different to the one provided in HttpClient: []-Chroma server http port provided in settings[r8) r r4intboolchroma_api_implchroma_server_host ValueErrorchroma_server_http_portchroma_server_ssl_enabledchroma_server_headersr5r<r=r>r?r/r0r1rrr HttpClients*rNcs|durt}t|}t|}t|}t|}t|}d|_|jr3|j|kr3td|jd|d||_|jrJ|j|krJtd|jd|d||_||_||_ t j |||dIdHS)a Creates an async client that connects to a remote Chroma server. This supports many clients connecting to the same server, and is the recommended way to use Chroma in production. Args: host: The hostname of the Chroma server. Defaults to "localhost". port: The port of the Chroma server. Defaults to "8000". ssl: Whether to use SSL to connect to the Chroma server. Defaults to False. headers: A dictionary of headers to send to the Chroma server. Defaults to {}. settings: A dictionary of settings to communicate with the chroma server. tenant: The tenant to use for this client. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. Nz'chromadb.api.async_fastapi.AsyncFastAPIrArBrCrDr8) r r4rErFrGrHrIrJrKrLAsyncClientCreatorcreaterMrrr AsyncHttpClients0 rQzapi.trychroma.comT) cloud_host cloud_port enable_sslapi_keyrRrSrTcCs|durddl}|jd}|durtdtd}|dur!t}t|}t|}t|}t|}t|}t|}d|_ ||_ ||_ ||_ d|_ ||_tj|_t|||dS) z Creates a client to connect to a tennant and database on the Chroma cloud. Args: tenant: The tenant to use for this client. database: The database to use for this client. api_key: The api key to use for this client. NrCHROMA_API_KEYzDDon't have an API key? Get one at https://app.trychroma.comz"Please enter your Chroma API key: r@z1chromadb.auth.token_authn.TokenAuthClientProviderr8)osenvirongetprintinputr r4rErFrGrHrJrKchroma_client_auth_providerchroma_client_auth_credentialsrX_CHROMA_TOKEN"chroma_auth_token_transport_headerr5)r0r1rUr/rRrSrTrWrrr CloudClients2 r`cCst|}t|}t|||dS)z Return a running chroma.API instance tenant: The tenant to use for this client. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. r8)r4r5r2rrr r=s rcCs t|dS)zU Creates an admin client that can be used to create tenants and databases. r/)AdminClientCreatorrarrr rPs r)rN)NN)Ftypingrrloggingchromadb.api.clientrr5rrbchromadb.api.async_clientrrOchromadb.auth.token_authnrchromadb.configr)rr r chromadb.apir r r chromadb.api.models.Collectionrchromadb.api.typesrrrrrrrrrrrr__all__ getLogger__name__loggerr+ __version__rFr!IN_COLAB is_clientchromadb.is_thin_clientr"rr(sqlite_version_info subprocesssys check_call executable __import__modulespop RuntimeErrorr-r.r4r6r:rErNrQr`rrrr s.     8         4 8 ;