BvhY|ddlmZddlmZddlmZddlmZmZm Z m Z m Z ddl m Z Gdde ZGdd e Zy ) )Dict) HTTPException)override) AuthzAction AuthzResourceServerAuthenticationProviderServerAuthorizationProvider UserIdentity)Systemc4eZdZdZedeeefdefdZy)ExampleAuthenticationProviderzIn practice the tenant would likely be resolved from some other opaque value (e.g. key/token). Here, it's just passed directly as a header for simplicity.headersreturnc<td|jddS)Ntestzx-tenant)user_idtenant)r get)selfrs X/RAG/venv/lib/python3.12/site-packages/chromadb/test/client/test_database_tenant_auth.pyauthenticate_or_raisez3ExampleAuthenticationProvider.authenticate_or_raises ;;z40  N) __name__ __module__ __qualname____doc__rrstrr rrrr r s0e  T#s(^   rr c NeZdZdZdeddffd Zededede ddfd Z xZ S) ExampleAuthorizationProviderzUA simple authz provider that asserts the user's tenant matches the resource's tenant.systemrNcFt|||j|_y)N)super__init__settings _settings)rr! __class__s rr$z%ExampleAuthorizationProvider.__init__s  ruseractionresourcec|jy|tjk(ry|j|jk7r tddy)Ni Unauthorized) status_codedetail)rrRESETr)rr(r)r*s rauthorize_or_raisez/ExampleAuthorizationProvider.authorize_or_raise sD ;;   [&& &  ;;(// )CG G *r) rrrrr r$rr rrr0 __classcell__)r's@rr r sS_)v)$) H  H*5 HAN H  H Hrr N)typingrfastapir overridesr chromadb.authrrrr r chromadb.configr r r rrrr7s9!# $@ H#>Hr