o ɭRhE1@s>ddlmZmZddlZddlmZddlmZddl m Z ddl m Z ddlZddlmZmZmZddlmZmZmZdd lmZdd lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)gd Z*e+e,Z-ea.d Z/d e0fddZ1e1Z2dZ3z ddl4m5Z5e5Z3Wn e6ydZ3Ynwe3sddl7Z7e7j8dkre2rddl9Z9ddl:Z:e9;e:j?de:j>d<ne@dd@ddZAd efddZBdeefdeedeCd eCd efd!d"ZDd#deefd$eCdeedeCd eCd ef d%d&ZEddeefd$eeCdeedeCd eCd ef d'd(ZFd)d*dddeefd+eCd,eGd-e0d.eeeCeCfdeedeCd eCd efd/d0ZHd)d*dddeefd+eCd,eGd-e0d.eeeCeCfdeedeCd eCd efd1d2ZI  dAd3d*d4d5deCd eCd6eeCdeed7eCd8eGd9e0d efd:d;ZJt.eefdedeCd eCd efdd?ZdS)B)DictOptionalN)Client) AdminClient) AsyncClient)TokenTransportHeader)DEFAULT_DATABASEDEFAULT_TENANTSettings)AdminAPIAsyncClientAPI ClientAPI) Collection)CollectionMetadataUpdateMetadata DocumentsEmbeddingFunction EmbeddingsURIURIsIDsIncludeMetadata MetadatasWhere QueryResult GetResult WhereDocumentUpdateCollectionMetadata)rrrrrrrrrrrrrrrrrrz1.0.12returncCs$zddl}WdStyYdSw)NrTF) google.colab ImportError)googler#S/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/chromadb/__init__.py is_in_colab<s  r%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.cKstjjdi|adS)zGOverride Chroma's default settings, environment variables or .env filesNr#)chromadbconfigr __settings)kwargsr#r#r$ configuregsr1cCstS)N)r/r#r#r#r$ get_settingsmsr2settingstenantdatabasecCs2|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. NFr3r4r5)r is_persistentstr ClientCreatorr6r#r#r$EphemeralClientqs  r:z./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. NTr4r5r3)r persist_directoryr7r8r9r;r3r4r5r#r#r$PersistentClientsr?cCsF|durt}d|_|du|_|pd|_t|}t|}t|||dS)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. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. Nz!chromadb.api.rust.RustBindingsAPIr<)r chroma_api_implr7r=r8r9r>r#r#r$ RustClients  rB 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[r<) r r8intboolrAchroma_server_host ValueErrorchroma_server_http_portchroma_server_ssl_enabledchroma_server_headersr9rDrErFrGr3r4r5r#r#r$ HttpClients*rUcs|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.AsyncFastAPIrIrJrKrLr<) r r8rMrNrArOrPrQrRrSAsyncClientCreatorcreaterTr#r#r$AsyncHttpClients0 rXzapi.trychroma.comT) cloud_host cloud_port enable_sslapi_keyrYrZr[cCs|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: rHz1chromadb.auth.token_authn.TokenAuthClientProviderr<)osenvirongetprintinputr r8rMrNrArOrQrRchroma_client_auth_providerchroma_client_auth_credentialsrX_CHROMA_TOKEN"chroma_auth_token_transport_headerr9)r4r5r\r3rYrZr[r^r#r#r$ CloudClient)s2 rgcCst|}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. r<)r8r9r6r#r#r$rcs rcCs t|dS)zU Creates an admin client that can be used to create tenants and databases. r3)AdminClientCreatorrhr#r#r$rvs r)rN)NN)Ktypingrrloggingchromadb.api.clientrr9rrichromadb.api.async_clientrrVchromadb.auth.token_authnrchromadb.configr-rr r chromadb.apir r r chromadb.api.models.Collectionrchromadb.api.typesrrrrrrrrrrrrrrrr__all__ getLogger__name__loggerr/ __version__rNr%IN_COLAB is_clientchromadb.is_thin_clientr&r!r,sqlite_version_info subprocesssys check_call executable __import__modulespop RuntimeErrorr1r2r8r:r?rBrMrUrXrgr#r#r#r$sL     H          4 8 ;