q 9iv9ddlmZmZmZddlZddlmZddlmZ 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7dd l8m9Z9dd l:m;Z;mZ>dd l?m@Z@ddlAZAgdZBddlCmDZDejeFZGeaHdZIdeJfdZKeKZLdZM ddlNmOZOeOZMeMswddlQZQeQjdkrdeLrZddlSZSddlTZTeSjeTjddddgeWdeTjjdeTjd<neZdd:dZ[defdZ\deefdeed e]d!e]defd"Z^d#deefd$ee]e@fdeed e]d!e]def d%Z_ddeefd$ee]deed e]d!e]def d&Z`d'd(dddeefd)e]d*ead+eJd,eee]e]fdeed e]d!e]defd-Zbd'd(dddeefd)e]d*ead+eJd,eee]e]fdeed e]d!e]defd.Zc d;d/d0d1d2d ee]d!ee]d3ee]deed4e]d5ead6eJdefd7Zdteefded e]d!e]defd8Zefdedefd9Zy#eP$rdZMYpwxYw)<)DictOptionalUnionN)Client) AdminClient) AsyncClient)TokenTransportHeader)DEFAULT_DATABASEDEFAULT_TENANTSettings)AdminAPIAsyncClientAPI ClientAPI) Collection)CollectionMetadataUpdateMetadata DocumentsEmbeddingFunction EmbeddingsURIURIsIDsIncludeMetadata MetadatasWhere QueryResult GetResult WhereDocumentUpdateCollectionMetadata SparseVector SparseVectorsSparseEmbeddingFunctionSchemaVectorIndexConfigHnswIndexConfigSpannIndexConfigFtsIndexConfigSparseVectorIndexConfigStringInvertedIndexConfigIntInvertedIndexConfigFloatInvertedIndexConfigBoolInvertedIndexConfig)Search)KeyKKnnRrf)Path)$rrrrrrrrrrrrrrr rrr r.r/r0r1r2r!r"r#r$r%r&r'r(r)r*r+r,r-)CloudClientArgz1.3.6returnc, ddl}y#t$rYywxYw)NrTF) google.colab ImportError)googles Z/var/www/html/leadgen/airagagent/rag_env/lib/python3.12/site-packages/chromadb/__init__.py is_in_colabr;ps 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: configurerIs))3F3JctS)N)rGrDrJr: get_settingsrLs rJsettingstenantdatabasecp| 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. FrMrNrO)r is_persistentstr ClientCreatorrQs r:EphemeralClientrUs<:"H[F8}H (6H MMrJz./chromapathc| t}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. TrNrOrM)r rSpersist_directoryrRrTrVrMrNrOs r:PersistentClientr[sH:!$TH!H[F8}H H MMrJc| t}d|_|du|_|xsd|_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. Defaults to the default tenant. database: The database to use for this client. Defaults to the default database. Nz!chromadb.api.rust.RustBindingsAPIrX)r chroma_api_implrRrYrSrTrZs r: RustClientr_sU:BH!-H!%H[F8}H H MMrJ 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[rX) r rSintboolr^chroma_server_host ValueErrorchroma_server_http_portchroma_server_ssl_enabledchroma_server_headersrTrarbrcrdrMrNrOs r: HttpClientrss"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 MMrJcK| 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.AsyncFastAPIrgrhrirjrX) r rSrkrlr^rmrnrorprqAsyncClientCreatorcreaterrs r:AsyncHttpClientrw(s;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.comiT) cloud_host cloud_port enable_sslapi_keyrxryrzc tdd|g}t|Dcgc]}|jc}sC|D]>}|jxs)tjj |j |_@|Dcgc]}|j|} }| r_tddj| Dcgc]}|jc}ddj| Dcgc]}|j c}| t}|xstjj d}| t|}|xstjj d}| t|}t|}t|}t|}t|}d |_||_||_||_d |_||_t(j*|_d |_t1||| Scc}wcc}wcc}wcc}w) aK Creates a client to connect to a tenant and database on Chroma cloud. Args: tenant: The tenant to use for this client. Optional. If not provided, it will be inferred from the API key if the key is scoped to a single tenant. If provided, it will be validated against the API key's scope. database: The database to use for this client. Optional. If not provided, it will be inferred from the API key if the key is scoped to a single database. If provided, it will be validated against the API key's scope. api_key: The api key to use for this client. r{CHROMA_API_KEY)nameenv_varvaluezMissing required arguments: z, z8. Please provide them or set the environment variables: CHROMA_TENANTCHROMA_DATABASErfz1chromadb.auth.token_authn.TokenAuthClientProviderTrX)r4allrosenvirongetrrnjoinr~r rSrkrlr^rmrorpchroma_client_auth_providerchroma_client_auth_credentialsr X_CHROMA_TOKEN"chroma_auth_token_transport_header;chroma_overwrite_singleton_tenant_database_access_from_authrT) rNrOr{rMrxryrz required_argsarg missing_argss r: CloudClientr]s( I/?wOM ]3c 3 4  AC @RZZ^^CKK%@CI A$1FCCII4ECFLF*499,5W3chh5W+X*YZEEIYYgsOt`cPSP[P[OtEuDv x  : 6rzz~~o6F V<2::>>*;rs$((7H:FF<<5B6 ( T*   8 $ Z  T =  6I""Z/    !J ! !ui9KL  { #%([[__[%ACKK "( 4 h $( $Nx N NN N2(#' $ N T Nx N N N  N8#' $ N 3-Nx N N N  N<(,#' $0N 0N 0N 0Nd38n % 0N x 0N  0N0N0Nh(,#' $2 2 2 2d38n % 2 x 2  222l!"!#' >N *>N SM>Nsm>Nc]>Nx >N  >N>N>N>ND$ $NN NN N&&.Z1(1H1c IsII  I