update pre-commit (#330)

This commit is contained in:
Shuchang Zheng
2024-05-16 17:11:49 -07:00
committed by GitHub
parent 482a8b8466
commit 9e1b07451f
19 changed files with 40 additions and 24 deletions

View File

@@ -37,9 +37,11 @@ class LLMAPIHandlerFactory:
redis_port=llm_config.redis_port,
redis_password=llm_config.redis_password,
routing_strategy=llm_config.routing_strategy,
fallbacks=[{llm_config.main_model_group: llm_config.fallback_model_group}]
if llm_config.fallback_model_group
else [],
fallbacks=(
[{llm_config.main_model_group: llm_config.fallback_model_group}]
if llm_config.fallback_model_group
else []
),
num_retries=llm_config.num_retries,
retry_after=llm_config.retry_delay_seconds,
set_verbose=False if SettingsManager.get_settings().is_cloud_environment() else llm_config.set_verbose,

View File

@@ -59,5 +59,4 @@ class LLMAPIHandler(Protocol):
step: Step | None = None,
screenshots: list[bytes] | None = None,
parameters: dict[str, Any] | None = None,
) -> Awaitable[dict[str, Any]]:
...
) -> Awaitable[dict[str, Any]]: ...