tPj)UdZddlZddlZddlmZmZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZddlmZmZdd lmZd Ze ed <d Ze ed <dZe ed<GddeZGddeZGddeZeiZe ee efed<Gdde Z!GddZ"e"Z#e ed<ee$edfZ%e ee$edfed<de&de$edffd Z'Gd!d"Z(e(Z)e ed#<d$e!d%ed&e*d'e+d(e ddf d)Z,d*ee e&fde+fd+Z-d*ee e&fde fd,Z.e#fd-e d.d/ee efdzd0ee efdzd1e dzd2e dzd3ed4e"ddfd5Z/y)6aCustom validation of team metadata on team create/update. Operators point `general_settings.custom_team_metadata_validate` at an async Python function (loaded via `get_instance_fn`, like `custom_key_generate`). The function receives a `TeamMetadataValidationPayload` and returns a `TeamMetadataValidationResult`. The proxy awaits it before committing a team write and fails closed: a rejected value surfaces the function's own message (HTTP 400), while any raised exception or timeout blocks the write with a generic message (HTTP 503). N) AwaitableMapping)MappingProxyType)FinalLiteralProtocol) HTTPExceptionstatus) BaseModel JsonValue TypeAdapter)CommonProxyErrorsUserAPIKeyAuth)TeamMetadataFieldSchemag@0DEFAULT_TEAM_METADATA_VALIDATION_TIMEOUT_SECONDSzgTeam metadata validation is currently unavailable, so the team was not saved. Contact your proxy admin.4DEFAULT_TEAM_METADATA_VALIDATION_UNAVAILABLE_MESSAGEz Team metadata failed validation.1DEFAULT_TEAM_METADATA_VALIDATION_REJECTED_MESSAGEcJeZdZUdZedzed<dZedzed<dZedzed<y)TeamMetadataRequesterNuser_id user_email user_role)__name__ __module__ __qualname__rstr__annotations__rr~/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/management_helpers/team_metadata_validation.pyrr!s,GS4Z!Jd ! IsTz rrceZdZUeded<eeefed<dZeeefdzed<dZ edzed<dZ edzed<e ed<y) TeamMetadataValidationPayloadcreateupdate operationmetadataNexisting_metadatateam_id team_alias requester) rrrrrrrr r(r)r*rrrr r"r"'sX)**c9n%%8<wsI~.5<GS4Z!Jd !$$rr"c,eZdZUeed<dZedzed<y)TeamMetadataValidationResultvalidN error_message)rrrboolrr/rrrr r-r-0s K $M3:$rr-_EMPTY_METADATAc"eZdZdedeefdZy)TeamMetadataValidatorpayloadreturncyNr)selfr4s r __call__zTeamMetadataValidator.__call__9srN)rrrr"rr-r9rrr r3r38sq =qYOkElqrr3c<eZdZddZdedzddfdZdedzfdZy)TeamMetadataValidatorRegistryr5Ncd|_yr7 _validatorr8s r __init__z&TeamMetadataValidatorRegistry.__init__=s 8<r validatorc||_yr7r=)r8rAs r setz!TeamMetadataValidatorRegistry.set@s #rc|jSr7r=r?s r getz!TeamMetadataValidatorRegistry.getCs rr5N)rrrr@r3rCrErrr r;r;<s2=$2T9$d$*T1rr; TEAM_METADATA_VALIDATOR_REGISTRY._TEAM_METADATA_SCHEMA_ADAPTER raw_schemar5c|ytj|}td|Dtt fdD}|rt ddj ||S)zhParse ``general_settings.team_metadata_schema``; raises on a malformed schema so config load fails fast.rc34K|]}|jywr7)key).0fields r z-parse_team_metadata_schema..Ss6e 6sc3LK|]}j|dkDs|yw)N)count)rMrLkeyss r rOz-parse_team_metadata_schema..Ts!,XSDJJsOVWDWS,Xs$$z.team_metadata_schema contains duplicate keys: z, )rHvalidate_pythontuplesorted frozenset ValueErrorjoin)rIfieldsduplicate_keysrSs @r parse_team_metadata_schemar\Nsg1AA*MF6v66D"9,XD,X#XYNI$))TbJcIdeff MrcDeZdZddZdeedfddfdZdeedffdZy) TeamMetadataSchemaRegistryr5Ncd|_y)Nr_fieldsr?s r r@z#TeamMetadataSchemaRegistry.__init__[s <> rrZ.c||_yr7r`)r8rZs r rCzTeamMetadataSchemaRegistry.set^s  rc|jSr7r`r?s r rEzTeamMetadataSchemaRegistry.getas ||rrF)rrrr@rUrrCrErrr r^r^Zs<?% 7 <=$U2C78rr^TEAM_METADATA_SCHEMA_REGISTRYrAr4 premium_usertimeout_secondsunavailable_messageclK|dur8ttjddtjj it j|s=t jt|ddsttjddi tj|||d{}tj|}|j s/ttjd|j"xst$iy7U#t$rttjd|iwxYww)NTerrorz8custom_team_metadata_validate is an Enterprise feature. ) status_codedetailr9z7custom_team_metadata_validate must be an async function)timeout)r r HTTP_400_BAD_REQUESTrnot_premium_uservalueinspectiscoroutinefunctiongetattrHTTP_500_INTERNAL_SERVER_ERRORasynciowait_forr-model_validate ExceptionHTTP_503_SERVICE_UNAVAILABLEr.r/r)rAr4rerfrg raw_resultresults r run_team_metadata_validationr{hs,433STeTvTvT|T|S}~   ##I.'2M2MgV_akmqNr2s==R   ")"2"29W3E"__ 4CCJO <<33--b1b   `  ;;01   s0BD4 D 2D3D ?56FG  sBC?C)0__doc__rtrpcollections.abcrrtypesrtypingrrrfastapir r pydanticr r r litellm.proxy._typesrr7litellm.types.proxy.management_endpoints.team_endpointsrrrrrrr"r-r1rr3r;rGrUrHobjectr\r^rdr0rr{rrrrrr rs1 ."++)66B;>0%=m5e<^15]!I! %I%%9% 3C22FwsI~./FrHr+H*I %IYd !3 &'Zu[7NPS7S1T%UV 6 e