Skip to content

Commit

Permalink
Fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
Hialus committed Sep 25, 2023
1 parent 6b2fdee commit fd72d01
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 2 deletions.
4 changes: 3 additions & 1 deletion app/llms/strategy_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,9 @@ def set_correct_session(
"No viable LLMs found! Using LLM with longest context length."
)
llm_configs = {
llm_key: config for llm_key, config in self.llm.llm_configs.items() if llm_key not in exclude_llms
llm_key: config
for llm_key, config in self.llm.llm_configs.items()
if llm_key not in exclude_llms
}
if llm_configs.__len__() == 0:
raise ValueError("All LLMs are excluded!")
Expand Down
3 changes: 2 additions & 1 deletion app/services/circuit_breaker.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,8 @@ def protected_call(
async def protected_call_async(
cls, func, cache_key: str, accepted_exceptions: tuple = ()
):
"""Wrap function call to avoid too many failures in a row. Async version.
"""Wrap function call to avoid too many failures in a row.
Async version.
Params:
func: function to be called
Expand Down

0 comments on commit fd72d01

Please sign in to comment.