forked from phoenix/litellm-mirror
f252350881
* fix(ui_sso.py): fix faulty admin only check Fixes https://github.com/BerriAI/litellm/issues/6286 * refactor(sso_helper_utils.py): refactor /sso/callback to use helper utils, covered by unit testing Prevent future regressions * feat(prompt_factory): support 'ensure_alternating_roles' param Closes https://github.com/BerriAI/litellm/issues/6257 * fix(proxy/utils.py): add dailytagspend to expected views * feat(auth_utils.py): support setting regex for clientside auth credentials Fixes https://github.com/BerriAI/litellm/issues/6203 * build(cookbook): add tutorial for mlflow + langchain + litellm proxy tracing * feat(argilla.py): add argilla logging integration Closes https://github.com/BerriAI/litellm/issues/6201 * fix: fix linting errors * fix: fix ruff error * test: fix test * fix: update vertex ai assumption - parts not always guaranteed (#6296) * docs(configs.md): add argila env var to docs
38 lines
999 B
Python
38 lines
999 B
Python
# What is this?
|
|
## This tests the batch update spend logic on the proxy server
|
|
|
|
|
|
import asyncio
|
|
import os
|
|
import random
|
|
import sys
|
|
import time
|
|
import traceback
|
|
from datetime import datetime
|
|
|
|
from dotenv import load_dotenv
|
|
from fastapi import Request
|
|
|
|
load_dotenv()
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system path
|
|
|
|
import logging
|
|
from litellm.proxy.management_endpoints.sso_helper_utils import (
|
|
check_is_admin_only_access,
|
|
has_admin_ui_access,
|
|
)
|
|
from litellm.proxy._types import LitellmUserRoles
|
|
|
|
|
|
def test_check_is_admin_only_access():
|
|
assert check_is_admin_only_access("admin_only") is True
|
|
assert check_is_admin_only_access("user_only") is False
|
|
|
|
|
|
def test_has_admin_ui_access():
|
|
assert has_admin_ui_access(LitellmUserRoles.PROXY_ADMIN.value) is True
|
|
assert has_admin_ui_access(LitellmUserRoles.PROXY_ADMIN_VIEW_ONLY.value) is True
|
|
assert has_admin_ui_access(LitellmUserRoles.INTERNAL_USER_VIEW_ONLY.value) is False
|