(code quality) run ruff rule to ban unused imports (#7313)

* remove unused imports

* fix AmazonConverseConfig

* fix test

* fix import

* ruff check fixes

* test fixes

* fix testing

* fix imports
This commit is contained in:
Ishaan Jaff 2024-12-19 12:33:42 -08:00 committed by GitHub
parent 5e344497ce
commit c7f14e936a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
347 changed files with 5473 additions and 7207 deletions

View file

@ -12,7 +12,6 @@ All /key management endpoints
import asyncio
import copy
import json
import re
import secrets
import traceback
import uuid
@ -281,11 +280,8 @@ async def generate_key_fn( # noqa: PLR0915
"""
try:
from litellm.proxy.proxy_server import (
create_audit_log_for_update,
general_settings,
litellm_proxy_admin_name,
prisma_client,
proxy_logging_obj,
user_api_key_cache,
user_custom_key_generate,
)
@ -590,8 +586,6 @@ async def update_key_fn(
```
"""
from litellm.proxy.proxy_server import (
create_audit_log_for_update,
litellm_proxy_admin_name,
prisma_client,
proxy_logging_obj,
user_api_key_cache,
@ -709,15 +703,7 @@ async def delete_key_fn(
HTTPException: If an error occurs during key deletion.
"""
try:
from litellm.proxy.proxy_server import (
create_audit_log_for_update,
general_settings,
litellm_proxy_admin_name,
prisma_client,
proxy_logging_obj,
user_api_key_cache,
user_custom_key_generate,
)
from litellm.proxy.proxy_server import prisma_client, user_api_key_cache
if prisma_client is None:
raise Exception("Not connected to DB!")
@ -818,14 +804,7 @@ async def info_key_fn_v2(
-d {"keys": ["sk-1", "sk-2", "sk-3"]}
```
"""
from litellm.proxy.proxy_server import (
create_audit_log_for_update,
general_settings,
litellm_proxy_admin_name,
prisma_client,
proxy_logging_obj,
user_custom_key_generate,
)
from litellm.proxy.proxy_server import prisma_client
try:
if prisma_client is None:
@ -889,14 +868,7 @@ async def info_key_fn(
-H "Authorization: Bearer sk-02Wr4IAlN3NvPXvL5JVvDA"
```
"""
from litellm.proxy.proxy_server import (
create_audit_log_for_update,
general_settings,
litellm_proxy_admin_name,
prisma_client,
proxy_logging_obj,
user_custom_key_generate,
)
from litellm.proxy.proxy_server import prisma_client
try:
if prisma_client is None: