From f9a7b18073b68aa46b9c58ecd82b62e5fadc40d8 Mon Sep 17 00:00:00 2001 From: Trung Dinh Date: Mon, 9 Dec 2024 22:36:07 +0700 Subject: [PATCH] Improve condition to pick up weak model --- pr_agent/algo/pr_processing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pr_agent/algo/pr_processing.py b/pr_agent/algo/pr_processing.py index bbe363e1..b37c3001 100644 --- a/pr_agent/algo/pr_processing.py +++ b/pr_agent/algo/pr_processing.py @@ -354,7 +354,7 @@ async def retry_with_fallback_models(f: Callable, model_type: ModelType = ModelT def _get_all_models(model_type: ModelType = ModelType.REGULAR) -> List[str]: - if model_type == ModelType.WEAK: + if get_settings().config.get('model_weak') and model_type == ModelType.WEAK: model = get_settings().config.model_weak else: model = get_settings().config.model