chore: format

This commit is contained in:
Timothy Jaeryang Baek
2026-03-25 16:43:06 -05:00
parent 05252e19b5
commit 350d52f515
73 changed files with 766 additions and 55 deletions
+1 -3
View File
@@ -682,9 +682,7 @@ def load_oauth_providers():
**({'timeout': int(OAUTH_TIMEOUT.value)} if OAUTH_TIMEOUT.value else {}),
},
redirect_uri=GOOGLE_REDIRECT_URI.value,
**({
'authorize_params': GOOGLE_OAUTH_AUTHORIZE_PARAMS
} if GOOGLE_OAUTH_AUTHORIZE_PARAMS else {}),
**({'authorize_params': GOOGLE_OAUTH_AUTHORIZE_PARAMS} if GOOGLE_OAUTH_AUTHORIZE_PARAMS else {}),
)
return client
+1 -3
View File
@@ -840,9 +840,7 @@ def get_embedding_function(
embeddings = []
for i, batch_embeddings in enumerate(batch_results):
if batch_embeddings is None:
raise Exception(
f'Embedding generation failed for batch {i + 1}/{len(batches)}'
)
raise Exception(f'Embedding generation failed for batch {i + 1}/{len(batches)}')
embeddings.extend(batch_embeddings)
log.debug(
+12 -8
View File
@@ -73,15 +73,17 @@ log = logging.getLogger(__name__)
async def send_get_request(
request: Request = None, url=None, key=None, user: UserModel = None, config=None,
request: Request = None,
url=None,
key=None,
user: UserModel = None,
config=None,
):
timeout = aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT_MODEL_LIST)
try:
async with aiohttp.ClientSession(timeout=timeout, trust_env=True) as session:
if request and config:
headers, cookies = await get_headers_and_cookies(
request, url, key, config, user=user
)
headers, cookies = await get_headers_and_cookies(request, url, key, config, user=user)
else:
headers = {
**({'Authorization': f'Bearer {key}'} if key else {}),
@@ -105,7 +107,11 @@ async def send_get_request(
async def get_models_request(
request: Request = None, url=None, key=None, user: UserModel = None, config=None,
request: Request = None,
url=None,
key=None,
user: UserModel = None,
config=None,
):
if is_anthropic_url(url):
return await get_anthropic_models(url, key, user=user)
@@ -383,9 +389,7 @@ async def get_all_models_responses(request: Request, user: UserModel) -> list:
if enable:
if len(model_ids) == 0:
request_tasks.append(
get_models_request(request, url, api_keys[idx], user=user, config=api_config)
)
request_tasks.append(get_models_request(request, url, api_keys[idx], user=user, config=api_config))
else:
model_list = {
'object': 'list',
+3 -3
View File
@@ -3692,9 +3692,9 @@ async def streaming_chat_response_handler(response, ctx):
current_response_tool_call['function']['name'] = delta_name
if delta_arguments:
current_response_tool_call['function'][
'arguments'
] += delta_arguments
current_response_tool_call['function']['arguments'] += (
delta_arguments
)
# Emit pending tool calls in real-time
if response_tool_calls: