_jddlmZmZddlmZddlmZmZddlm Z dede deee ee ffdZ d dede d ee d ee deee ee ff d Z d S))OptionalTuple) UserIdentity)DEFAULT_DATABASEDEFAULT_TENANT)ChromaAuthError user_identity4overwrite_singleton_tenant_database_access_from_authreturnc|sdSd}d}|j}|j}|r|dkr|}|r;t|}t|dkrd|vrt |d}||fS)a( If settings.chroma_overwrite_singleton_tenant_database_access_from_auth is False, this function always returns (None, None). If settings.chroma_overwrite_singleton_tenant_database_access_from_auth is True, follows the following logic: - If the user only has access to a single tenant, this function will return that tenant as its first return value. - If the user only has access to a single database, this function will return that database as its second return value. If the user has access to multiple tenants and/or databases, including "*", this function will return None for the corresponding value(s). - If the user has access to multiple tenants and/or databases this function will return None for the corresponding value(s). NNN*r)tenant databasessetlenlist)r r rdatabase user_tenantuser_databasesuser_databases_sets Y/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/auth/utils/__init__.py(_singleton_tenant_database_if_applicablers& @z FH&K",N{c))3 00 ! " "a ' 'C7I,I,I.//2H 8 Nuser_provided_tenantuser_provided_databasec"t||\}}|r)|tkr|r||krtd|d|d|r)|tkr|r||krtd|d|d|r |tkr|r|}|r |tkr|r|}||fS)N)r r zTenant z does not match z5 from the server. Are you sure the tenant is correct?z Database z7 from the server. Are you sure the database is correct?)rrrr)r r rr new_tenant new_databases rmaybe_set_tenant_and_databaser!*sJ 1#=q    Q N 2 2  3 . . .P(<PPjPPPQQ QY "&6 6 6  7 2 2 2X*@XXR^XXXYY Y !*$8N$J$J %K) ".&<@P&P&P 'Q!- !7 77rr )typingrr chromadb.authrchromadb.configrrchromadb.errorsrboolstrrr!rrr)s """"""""&&&&&&<<<<<<<<++++++:> 8C=(3- '(J+/,0 ,8,8,8:>,8#3-,8%SM ,8  8C=(3- '( ,8,8,8,8,8,8r