ug,ddlmZmZddlZddlmZddlmZddl m Z ddl m Z ddlZddlmZmZmZddlmZmZmZdd lmZdd lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%gd Z&ejNe(Z)ea*d Z+d e,fdZ-e-Z.dZ/ ddl0m1Z1e1Z/e/swddl3Z3e3jhdkrde.rZddl5Z5ddl6Z6e5jne6jpddddge9de6jtjwde6jtd<nedeefdeede?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 efd3Zy#e2$rdZ/YJwxYw)6)DictOptionalN)Client) AdminClient) AsyncClient)TokenTransportHeader)DEFAULT_DATABASEDEFAULT_TENANTSettings)AdminAPIAsyncClientAPI ClientAPI) Collection) CollectionMetadata DocumentsEmbeddingFunction EmbeddingsIDsIncludeMetadataWhere QueryResult GetResult WhereDocumentUpdateCollectionMetadata)rrrrrrrrrrrrrrz0.6.3returnc, ddl}y#t$rYywxYw)NrTF) google.colab ImportError)googles F/var/www/openai/venv/lib/python3.12/site-packages/chromadb/__init__.py is_in_colabr"3s s  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 Btjjdi|ay)zGOverride Chroma's default settings, environment variables or .env filesN)chromadbconfigr __settings)kwargss r! configurer0^s))3F3JctS)N)r.r+r1r! get_settingsr3ds r1settingstenantdatabasecp| 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. Fr4r5r6)r is_persistentstr ClientCreatorr8s r!EphemeralClientr<hs<:"H[F8}H (6H 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. Tr5r6r4)r persist_directoryr9r:r;)r=r4r5r6s r!PersistentClientrAsD:!%H!H[F8}H H MMr1 localhosti@hostportsslheadersc| t}t|}t|}t|}t|}t|}d|_|j r+|j |k7rt d|j d|d||_|jr+|j|k7rt 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. chromadb.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! HttpClientrVs"0: t9D t9D s)C [F8}H=H""x'B'Bd'J6x7R7R6STGHLGMMN O  #'H''H,L,LPT,T;H<\<\;]^QRVQWWX Y  (,H$),H&%,H" H MMr1cK| t}t|}t|}t|}t|}t|}d|_|j r+|j |k7rt d|j d|d||_|jr+|j|k7rt d|jd|d||_||_||_ tj|||d{S7w)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!AsyncHttpClientrZs;0: t9D t9D s)C [F8}HHH""x'B'Bd'J6x7R7R6STGHLGMMN O  #'H''H,L,LPT,T;H<\<\;]^QRVQWWX Y  (,H$),H&%,H"#**8  sC0C92C73C9zapi.trychroma.comT) cloud_host cloud_port enable_sslapi_keyr[r\r]c|ddl}|jjd}|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(**..!12 X <=:[F8}H'lGZJZJj!J=H",H'1H$)3H& < (/6H+2F2U2UH/ H MMr1cJt|}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 MMr1ct|S)zU Creates an admin client that can be used to create tenants and databases. r4)AdminClientCreatorrms r!rrPs x 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!7AG:FF<<5      "   8 $ Z  T =  6I""Z/    !J ! !ui9KL  { #%([[__[%ACKK "( 4 h $( $Nx N NN N2#' $ N Nx N N N  N8(,#' $0N 0N 0N 0Nd38n % 0N x 0N  0N0N0Nh(,#' $2 2 2 2d38n % 2 x 2  222p"#' 7N *7N 7N7Nc]7Nx 7N  7N7N7N7Nv$ $NN NN N&&.Z1(1H1[IsGG! G!