Zi)9ddlmZmZmZddlZddlmZddlmZ ddl m Z ddl mZddlZddlmZmZmZddlmZmZmZdd lmZdd lmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9dd l:m;Z;dd lZ>m?Z?m@Z@dd lAmBZBddlCZCgdZDddlEmFZFejGeHZIeaJdZKdeLfdZMeMZNdZO ddlPmQZQeQZOn #eR$rdZOYnwxYweOslddlSZSeSjTdkr]eNrPddlUZUddlVZVeUjWeVjXddddgeYdeVjZ[deVjZd<n e\dd:dZ]defdZ^deefdeed e_d!e_defd"Z`d#deefd$ee_eBfdeed e_d!e_def d%Zaddeefd$ee_deed e_d!e_def d&Zbd'd(dddeefd)e_d*ecd+eLd,eee_e_fdeed e_d!e_defd-Zdd'd(dddeefd)e_d*ecd+eLd,eee_e_fdeed e_d!e_defd.Ze d;d/d0d1d2d ee_d!ee_d3ee_deed4e_d5ecd6eLdefd7Zfteefded e_d!e_defd8Zefdedefd9ZdS)<)DictOptionalUnionN)Client) AdminClient) AsyncClient)TokenTransportHeader)DEFAULT_DATABASEDEFAULT_TENANTSettings)AdminAPIAsyncClientAPI ClientAPI) Collection)CmekCollectionMetadataUpdateMetadata DocumentsEmbeddingFunction EmbeddingsURIURIsIDsIncludeMetadata Metadatas ReadLevelWhere QueryResult GetResult WhereDocumentUpdateCollectionMetadata SparseVector SparseVectorsSparseEmbeddingFunctionSchemaVectorIndexConfigHnswIndexConfigSpannIndexConfigFtsIndexConfigSparseVectorIndexConfigStringInvertedIndexConfigIntInvertedIndexConfigFloatInvertedIndexConfigBoolInvertedIndexConfig)Search)KeyKKnnRrf)Path)%rrrrrr!rrrrrrrrrr"rr r r0r1r2r3r4r#r$r%r&r'r(r)r*r+r,r-r.r/)CloudClientArgz1.5.2returnc2 ddl}dS#t$rYdSwxYw)NrTF) google.colab ImportError)googles N/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/__init__.py is_in_colabr=ss: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< configurerKs )33F33JJJctSN)rIrFrLr< get_settingsrOs rLsettingstenantdatabasec|t}d|_t|}t|}t|||S)aCreate an in-memory client for local use. This client stores all data in memory and does not persist to disk. It is intended for testing and development. Args: settings: Optional settings to override defaults. tenant: Tenant name to use for requests. Defaults to the default tenant. database: Database name to use for requests. Defaults to the default database. Returns: ClientAPI: A configured client instance. NFrPrQrR)r is_persistentstr ClientCreatorrTs r<EphemeralClientrXsH$::"H[[F8}}H (6H M M MMrLz./chromapathc|t}t||_d|_t|}t|}t |||S)aCreate a persistent client that stores data on disk. This client is intended for local development and testing. For production, prefer a server-backed Chroma instance. Args: path: Directory to store persisted data. settings: Optional settings to override defaults. tenant: Tenant name to use for requests. database: Database name to use for requests. Returns: ClientAPI: A configured client instance. NTrQrRrP)r rVpersist_directoryrUrWrYrPrQrRs r<PersistentClientr^sV(::!$TH!H[[F8}}H H M M MMrLc|t}d|_|du|_|pd|_t |}t |}t |||S)a Creates an ephemeral or persistance instance of Chroma that saves to disk. This is useful for testing and development, but not recommended for production use. Args: path: An optional directory to save Chroma's data to. The client is ephemeral if a None value is provided. Defaults to None. tenant: The tenant to use for this client. database: The database to use for this client. Nz!chromadb.api.rust.RustBindingsAPIr[)r chroma_api_implrUr\rVrWr]s r< RustClientrbsa::BH!-H!%H[[F8}}H H M M MMrL localhosti@hostportsslheadersc|t}t|}t|}t|}t|}t|}d|_|jr&|j|krt d|jd|d||_|jr&|j|krt d|jd|d||_||_||_ t|||S)a>Create a client that connects to a Chroma server. Args: host: Hostname of the Chroma server. port: HTTP port of the Chroma server. ssl: Whether to enable SSL for the connection. headers: Optional headers to send with each request. settings: Optional settings to override defaults. tenant: Tenant name to use for requests. database: Database name to use for requests. Returns: ClientAPI: A configured client instance. Raises: ValueError: If settings specify a different host or port. 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 rVintboolrachroma_server_host ValueErrorchroma_server_http_portchroma_server_ssl_enabledchroma_server_headersrWrdrerfrgrPrQrRs r< HttpClientrvsB6:: 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 MMrLcK|t}t|}t|}t|}t|}t|}d|_|jr&|j|krt d|jd|d||_|jr&|j|krt d|jd|d||_||_||_ tj |||d{VS)aCreate an async client that connects to a Chroma HTTP server. This supports multiple clients connecting to the same server and is the recommended production configuration. Args: host: Hostname of the Chroma server. port: HTTP port of the Chroma server. ssl: Whether to enable SSL for the connection. headers: Optional headers to send with each request. settings: Optional settings to override defaults. tenant: Tenant name to use for requests. database: Database name to use for requests. Returns: AsyncClientAPI: A configured async client instance. Raises: ValueError: If settings specify a different host or port. Nz'chromadb.api.async_fastapi.AsyncFastAPIrjrkrlrmr[) r rVrnrorarprqrrrsrtAsyncClientCreatorcreaterus r<AsyncHttpClientrz8sp<:: 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      rLzapi.trychroma.comiT) cloud_host cloud_port enable_sslapi_keyr{r|r}c tdd|g}td|Ds5|D]2}|jp#tj|j|_3d|D} | rOtddd| Dd dd | D|t}|ptjd }|t|}|ptjd }|t|}t|}t|}t|}t|}d|_ ||_||_||_d|_||_t&j|_d|_t/|||S)agCreate a client for Chroma Cloud. If not provided, `tenant`, `database`, and `api_key` will be inferred from the environment variables `CHROMA_TENANT`, `CHROMA_DATABASE`, and `CHROMA_API_KEY`. Args: tenant: Tenant name to use, or None to infer from credentials. database: Database name to use, or None to infer from credentials. api_key: API key for Chroma Cloud. settings: Optional settings to override defaults. Returns: ClientAPI: A configured client instance. Raises: ValueError: If no API key is provided or available in the environment. r~CHROMA_API_KEY)nameenv_varvaluecg|] }|j SrFr.0args r< zCloudClient..s333c 333rLc g|] }|j | SrNrrs r<rzCloudClient..sFFFCCI4EC4E4E4ErLzMissing required arguments: z, cg|] }|j SrF)rrs r<rzCloudClient..s5W5W5W3ch5W5W5WrLz8. Please provide them or set the environment variables: cg|] }|j SrF)rrs r<rzCloudClient..sOtOtOt`cPSP[OtOtOtrLN CHROMA_TENANTCHROMA_DATABASEriz1chromadb.auth.token_authn.TokenAuthClientProviderTr[)r6allrosenvirongetrrqjoinr rVrnrorarprrrschroma_client_auth_providerchroma_client_auth_credentialsr X_CHROMA_TOKEN"chroma_auth_token_transport_header;chroma_overwrite_singleton_tenant_database_access_from_authrW) rQrRr~rPr{r|r} required_argsr missing_argss r< CloudClientrss8 I/?wOOOM 33]333 4 4A  A AC @RZ^^CK%@%@CIIFF=FFFL  x4995W5W,5W5W5W+X+X x xEIYYOtOtgsOtOtOtEuEu x x   :: 6rz~~o66F V<2:>>*;<r:rDsqlite_version_info subprocesssys check_call executable __import__modulespop RuntimeErrorrKrOrVrXr^rbrnrvrzrrFrLr<rs((((((((((777777HGGGGG::::::FFFFFFFFFF<<<<<<<<<<555555                                                                  F655555 ) ) ) V*)))))  8 $ $ XZZ  T ;==  666666IIIIINNN"Z//       JJJ !J !ui9KL    J{ # # #%([__[%A%ACK " ",( 4444 h $( $NNx N NN NNNN<(#' $ NN T Nx N N N  NNNNB#' $ NN 3-Nx N N N  NNNN<(,#' $3N3N 3N 3N 3Nd38n % 3N x 3N  3N3N3N3N3N3Nn(,#' $88 8 8 8d38n % 8 x 8  888888x!"!#' FN *FNFNFN SMFNsmFNc]FNx FN  FNFNFNFNFNFNFNT$ $NNN NN NNNN&&.XZZ11(1H111111sC%%C/.C/