mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
9b7ebb6a7d
* build(pyproject.toml): add new dev dependencies - for type checking * build: reformat files to fit black * ci: reformat to fit black * ci(test-litellm.yml): make tests run clear * build(pyproject.toml): add ruff * fix: fix ruff checks * build(mypy/): fix mypy linting errors * fix(hashicorp_secret_manager.py): fix passing cert for tls auth * build(mypy/): resolve all mypy errors * test: update test * fix: fix black formatting * build(pre-commit-config.yaml): use poetry run black * fix(proxy_server.py): fix linting error * fix: fix ruff safe representation error
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from typing import Any, Union
|
|
|
|
from litellm.proxy._types import (
|
|
GenerateKeyRequest,
|
|
LiteLLM_ManagementEndpoint_MetadataFields_Premium,
|
|
LiteLLM_TeamTable,
|
|
UserAPIKeyAuth,
|
|
)
|
|
from litellm.proxy.utils import _premium_user_check
|
|
|
|
|
|
def _is_user_team_admin(
|
|
user_api_key_dict: UserAPIKeyAuth, team_obj: LiteLLM_TeamTable
|
|
) -> bool:
|
|
for member in team_obj.members_with_roles:
|
|
if (
|
|
member.user_id is not None and member.user_id == user_api_key_dict.user_id
|
|
) and member.role == "admin":
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def _set_object_metadata_field(
|
|
object_data: Union[LiteLLM_TeamTable, GenerateKeyRequest],
|
|
field_name: str,
|
|
value: Any,
|
|
) -> None:
|
|
"""
|
|
Helper function to set metadata fields that require premium user checks
|
|
|
|
Args:
|
|
object_data: The team data object to modify
|
|
field_name: Name of the metadata field to set
|
|
value: Value to set for the field
|
|
"""
|
|
if field_name in LiteLLM_ManagementEndpoint_MetadataFields_Premium:
|
|
_premium_user_check()
|
|
object_data.metadata = object_data.metadata or {}
|
|
object_data.metadata[field_name] = value
|