phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

refactor(test_proxy_utils.py): place at root level test folder

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-28 13:15:19 -07:00
parent c1036001fa
commit 5403c5828c
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/tests/test_proxy_utils.py → tests/proxy_util_tests/test_proxy_utils.py
View file

@ -6,11 +6,12 @@ from unittest.mock import Mock
import pytest import pytest
from fastapi import Request from fastapi import Request
import litellm
sys.path.insert( sys.path.insert(
0, os.path.abspath("../..") 0, os.path.abspath("../..")
) # Adds the parent directory to the system path ) # Adds the parent directory to the system path
import litellm
from litellm.proxy._types import LitellmUserRoles, UserAPIKeyAuth from litellm.proxy._types import LitellmUserRoles, UserAPIKeyAuth
from litellm.proxy.auth.auth_utils import is_request_body_safe from litellm.proxy.auth.auth_utils import is_request_body_safe
from litellm.proxy.litellm_pre_call_utils import ( from litellm.proxy.litellm_pre_call_utils import (
@ -367,3 +368,12 @@ def test_is_request_body_safe_model_enabled(
error_raised = True error_raised = True
assert expect_error == error_raised assert expect_error == error_raised
def test_get_remaining_rate_limit():
"""
Ensure accurate remaining rate limit is returned to the client.
Relevant Issue: https://github.com/BerriAI/litellm/issues/5957
"""
pass