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

refactor spend endpoints

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-15 12:17:33 -07:00
parent 23e5f36522
commit b4057dc022
5 changed files with 1847 additions and 2036 deletions
Download patch file Download diff file Expand all files Collapse all files

2030
litellm/proxy/proxy_server.py
View file

File diff suppressed because it is too large Load diff

1832
litellm/proxy/spend_management_endpoints.py Normal file
View file

File diff suppressed because it is too large Load diff

6
litellm/tests/test_blocked_user_list.py
View file

@ -37,11 +37,13 @@ from litellm.proxy.proxy_server import (
update_key_fn, update_key_fn,
generate_key_fn, generate_key_fn,
generate_key_helper_fn, generate_key_helper_fn,
user_info,
block_user,
)
from litellm.proxy.spend_management_endpoints import (
spend_user_fn, spend_user_fn,
spend_key_fn, spend_key_fn,
view_spend_logs, view_spend_logs,
user_info,
block_user,
) )
from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger

9
litellm/tests/test_key_generate_prisma.py
View file

@ -46,9 +46,6 @@ from litellm.proxy.proxy_server import (
update_key_fn, update_key_fn,
generate_key_fn, generate_key_fn,
generate_key_helper_fn, generate_key_helper_fn,
spend_user_fn,
spend_key_fn,
view_spend_logs,
user_info, user_info,
info_key_fn, info_key_fn,
chat_completion, chat_completion,
@ -60,8 +57,12 @@ from litellm.proxy.proxy_server import (
model_list, model_list,
LitellmUserRoles, LitellmUserRoles,
) )
from litellm.proxy.team_endpoints import team_info, new_team, update_team from litellm.proxy.team_endpoints import team_info, new_team, update_team
from litellm.proxy.spend_management_endpoints import (
spend_user_fn,
spend_key_fn,
view_spend_logs,
)
from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token, update_spend from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token, update_spend
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger

6
litellm/tests/test_update_spend.py
View file

@ -34,11 +34,13 @@ from litellm.proxy.proxy_server import (
update_key_fn, update_key_fn,
generate_key_fn, generate_key_fn,
generate_key_helper_fn, generate_key_helper_fn,
user_info,
block_user,
)
from litellm.proxy.spend_management_endpoints import (
spend_user_fn, spend_user_fn,
spend_key_fn, spend_key_fn,
view_spend_logs, view_spend_logs,
user_info,
block_user,
) )
from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token, update_spend from litellm.proxy.utils import PrismaClient, ProxyLogging, hash_token, update_spend
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger