ti,ddlmZmZddlZddlmZddlmZddl m Z ddl m Z ddlZddlmZmZmZddlmZmZmZdd lmZdd lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%gd Z&ej'e(Z)ea*d Z+d e,fdZ-e-Z.dZ/ ddl0m1Z1e1Z/n #e2$rdZ/YnwxYwe/slddl3Z3e3j4dkr]e.rPddl5Z5ddl6Z6e5j7e6j8ddddge9de6j:;de6j:d<n edeefdeede?de?d efdZ@ddeefd e?deede?de?d ef d!ZAd"d#dddeefd$e?d%eBd&e,d'eee?e?fdeede?de?d efd(ZCd"d#dddeefd$e?d%eBd&e,d'eee?e?fdeede?de?d efd)ZD d5d*d#d+d,de?de?d-ee?deed.e?d/eBd0e,d efd1ZEtTeefdede?de?d efd2Zefded efd3ZdS)6)DictOptionalN)Client) AdminClient) AsyncClient)TokenTransportHeader)DEFAULT_DATABASEDEFAULT_TENANTSettings)AdminAPIAsyncClientAPI ClientAPI) Collection) CollectionMetadata DocumentsEmbeddingFunction EmbeddingsIDsIncludeMetadataWhere QueryResult GetResult WhereDocumentUpdateCollectionMetadata)rrrrrrrrrrrrrrz0.5.3returnc2 ddl}dS#t$rYdSwxYw)NrTF) google.colab ImportError)googles D/var/www/icac/venv/lib/python3.11/site-packages/chromadb/__init__.py is_in_colabr"4s:t uus  F)is_thin_client)#rz-mpipinstallzpysqlite3-binary pysqlite3sqlite3zYour system has an unsupported version of sqlite3. Chroma requires sqlite3 >= 3.35.0. Please visit https://docs.trychroma.com/troubleshooting#sqlite to learn how to upgrade.c 2tjjdi|adS)zGOverride Chroma's default settings, environment variables or .env filesN)chromadbconfigr __settings)kwargss r! configurer0_s )33F33JJJctS)N)r.r+r1r! get_settingsr3es r1settingstenantdatabasec|t}d|_t|}t|}t|||S)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. NFr4r5r6)r is_persistentstr ClientCreatorr8s r!EphemeralClientr<isH::"H[[F8}}H (6H M M MMr1z./chromapathc|t}||_d|_t|}t|}t |||S)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. NTr5r6r4)r persist_directoryr9r:r;)r=r4r5r6s r!PersistentClientrAsP::!%H!H[[F8}}H H M M MMr1 localhosti@hostportsslheadersc|t}t|}t|}t|}t|}t|}d|_|jr&|j|krt d|jd|d||_|jr&|j|krt d|jd|d||_||_||_ t|||S)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[r?) r r:intboolchroma_api_implchroma_server_host ValueErrorchroma_server_http_portchroma_server_ssl_enabledchroma_server_headersr;rCrDrErFr4r5r6s r! HttpClientrVsB0:: t99D t99D s))C [[F8}}H=H" x'Bd'J'J Ox7R O OHL O O O   #'H' H,LPT,T,T YH<\ Y YRV Y Y Y   (,H$),H&%,H" H M M MMr1cK|t}t|}t|}t|}t|}t|}d|_|jr&|j|krt d|jd|d||_|jr&|j|krt d|jd|d||_||_||_ tj |||d{VS)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.AsyncFastAPIrIrJrKrLr?) r r:rMrNrOrPrQrRrSrTAsyncClientCreatorcreaterUs r!AsyncHttpClientrZsp0:: t99D t99D s))C [[F8}}HHH" x'Bd'J'J Ox7R O OHL O O O   #'H' H,LPT,T,T YH<\ Y YRV Y Y Y   (,H$),H&%,H"#*8      r1zapi.trychroma.comT) cloud_host cloud_port enable_sslapi_keyr[r\r]c|ddl}|jd}|tdt d}|t }t |}t |}t |}t |}t|}t|}d|_ ||_ ||_ ||_ d|_ ||_tj|_t%|||S) 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: rHz1chromadb.auth.token_authn.TokenAuthClientProviderr?)osenvirongetprintinputr r:rMrNrOrPrRrSchroma_client_auth_providerchroma_client_auth_credentialsrX_CHROMA_TOKEN"chroma_auth_token_transport_headerr;)r5r6r^r4r[r\r]ras r! CloudClientrjs( *..!122 X   <==::[[F8}}H'llGZJZJj!!J=H",H'1H$)3H& < (/6H+2F2UH/ H M M MMr1cbt|}t|}t|||S)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. r?)r:r;r8s r!rr>s/[[F8}}H H M M MMr1c"t|S)zU Creates an admin client that can be used to create tenants and databases. r4)AdminClientCreatorrms r!rrQs x 0 0 00r1)rN)NN)Ftypingrrloggingchromadb.api.clientrr;rrnchromadb.api.async_clientrrXchromadb.auth.token_authnrchromadb.configr,r r r chromadb.apir r rchromadb.api.models.Collectionrchromadb.api.typesrrrrrrrrrrrr__all__ getLogger__name__loggerr. __version__rNr"IN_COLAB is_clientchromadb.is_thin_clientr#rr)sqlite_version_info subprocesssys check_call executable __import__modulespop RuntimeErrorr0r3r:r<rArMrVrZrjr+r1r!rs!!!!!!!!777777AAAAAAGGGGGG::::::FFFFFFFFFF<<<<<<<<<<555555                                "  8 $ $ XZZ  T ;==  666666IIIIINNN"Z//       JJJ !J !ui9KL    J{ # # #%([__[%A%ACK " ",( 4444 h $( $NNx N NN NNNN2#' $ NN Nx N N N  NNNN8(,#' $0N0N 0N 0N 0Nd38n % 0N x 0N  0N0N0N0N0N0Nh(,#' $22 2 2 2d38n % 2 x 2  222222p"#' 7N *7N7N7N 7N7Nc]7Nx 7N  7N7N7N7N7N7N7Nv$ $NNN NN NNNN&&.XZZ11(1H111111sBB%$B%