o \hw1@sVddlmZmZmZddlZddlmZddlmZ 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*dd l+m,Z,gd Z-e.e/Z0ea1d Z2de3fddZ4e4Z5dZ6z ddl7m8Z8e8Z6Wn e9ydZ6Ynwe6sddl:Z:e:j;dkre5rddle=j?ddddge@de=jABde=jAd<neCddAddZDdefddZEdeefdeed eFd!eFdefd"d#ZGd$deefd%eeFe,fdeed eFd!eFdef d&d'ZHddeefd%eeFdeed eFd!eFdef d(d)ZId*d+dddeefd,eFd-eJd.e3d/eeeFeFfdeed eFd!eFdefd0d1ZKd*d+dddeefd,eFd-eJd.e3d/eeeFeFfdeed eFd!eFdefd2d3ZL  dBd4d+d5d6d eFd!eFd7eeFdeed8eFd9eJd:e3defd;d<ZMt1eefded eFd!eFdefd=d>Zefdedefd?d@ZdS)C)DictOptionalUnionN)Client) AdminClient) AsyncClient)TokenTransportHeader)DEFAULT_DATABASEDEFAULT_TENANTSettings)AdminAPIAsyncClientAPI ClientAPI) Collection)CollectionMetadataUpdateMetadata DocumentsEmbeddingFunction EmbeddingsURIURIsIDsIncludeMetadata MetadatasWhere QueryResult GetResult WhereDocumentUpdateCollectionMetadata)Path)rrrrrrrrrrrrrrrrrrz1.0.13returncCs$zddl}WdStyYdSw)NrTF) google.colab ImportError)googler%T/home/air/segue/gemini/backup/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& configurehsr3cCstS)N)r1r%r%r%r& get_settingsnsr4settingstenantdatabasecCs2|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. NFr5r6r7)r is_persistentstr ClientCreatorr8r%r%r&EphemeralClientrs  r<z./chromapathcCs<|durt}t||_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. NTr6r7r5)r r:persist_directoryr9r;r=r5r6r7r%r%r&PersistentClients rAcCsF|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_implr9r?r:r;r@r%r%r& RustClients  rD 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 r:intboolrCchroma_server_host ValueErrorchroma_server_http_portchroma_server_ssl_enabledchroma_server_headersr;rFrGrHrIr5r6r7r%r%r& HttpClients*rWcs|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.AsyncFastAPIrKrLrMrNr>) r r:rOrPrCrQrRrSrTrUAsyncClientCreatorcreaterVr%r%r&AsyncHttpClients0 rZzapi.trychroma.comT) cloud_host cloud_port enable_sslapi_keyr[r\r]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: rJz1chromadb.auth.token_authn.TokenAuthClientProviderr>)osenvirongetprintinputr r:rOrPrCrQrSrTchroma_client_auth_providerchroma_client_auth_credentialsrX_CHROMA_TOKEN"chroma_auth_token_transport_headerr;)r6r7r^r5r[r\r]r`r%r%r& CloudClient*s2 ricCst|}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>)r:r;r8r%r%r&rds rcCs t|dS)zU Creates an admin client that can be used to create tenants and databases. r5)AdminClientCreatorrjr%r%r&rws r)r!N)NN)Ntypingrrrloggingchromadb.api.clientrr;rrkchromadb.api.async_clientrrXchromadb.auth.token_authnrchromadb.configr/r r r chromadb.apir r rchromadb.api.models.Collectionrchromadb.api.typesrrrrrrrrrrrrrrrrpathlibr __all__ getLogger__name__loggerr1 __version__rPr'IN_COLAB is_clientchromadb.is_thin_clientr(r#r.sqlite_version_info subprocesssys check_call executable __import__modulespop RuntimeErrorr3r4r:r<rArDrOrWrZrir%r%r%r&sN     H            4 8 ;