chore: format
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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(
|
||||
|
||||
@@ -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',
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user