diff --git a/pr_agent/tools/pr_description.py b/pr_agent/tools/pr_description.py index 0b4e19cd..357e9470 100644 --- a/pr_agent/tools/pr_description.py +++ b/pr_agent/tools/pr_description.py @@ -99,7 +99,7 @@ class PRDescription: # ticket extraction if exists await extract_and_cache_pr_tickets(self.git_provider, self.vars) - await retry_with_fallback_models(self._prepare_prediction, ModelType.TURBO) + await retry_with_fallback_models(self._prepare_prediction, ModelType.WEAK) if self.prediction: self._prepare_data() diff --git a/pr_agent/tools/pr_help_message.py b/pr_agent/tools/pr_help_message.py index 3b0ed5a1..c3ec2c1d 100644 --- a/pr_agent/tools/pr_help_message.py +++ b/pr_agent/tools/pr_help_message.py @@ -114,7 +114,7 @@ class PRHelpMessage: self.vars['snippets'] = docs_prompt.strip() # run the AI model - response = await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.REGULAR) + response = await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.WEAK) response_yaml = load_yaml(response) response_str = response_yaml.get('response') relevant_sections = response_yaml.get('relevant_sections') diff --git a/pr_agent/tools/pr_line_questions.py b/pr_agent/tools/pr_line_questions.py index 50bf4183..4ef5b9a7 100644 --- a/pr_agent/tools/pr_line_questions.py +++ b/pr_agent/tools/pr_line_questions.py @@ -79,7 +79,7 @@ class PR_LineQuestions: line_end=line_end, side=side) if self.patch_with_lines: - response = await retry_with_fallback_models(self._get_prediction, model_type=ModelType.TURBO) + response = await retry_with_fallback_models(self._get_prediction, model_type=ModelType.WEAK) get_logger().info('Preparing answer...') if comment_id: diff --git a/pr_agent/tools/pr_questions.py b/pr_agent/tools/pr_questions.py index 8112510e..1ab496dc 100644 --- a/pr_agent/tools/pr_questions.py +++ b/pr_agent/tools/pr_questions.py @@ -63,7 +63,7 @@ class PRQuestions: if img_path: get_logger().debug(f"Image path identified", artifact=img_path) - await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.TURBO) + await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.WEAK) pr_comment = self._prepare_pr_answer() get_logger().debug(f"PR output", artifact=pr_comment) diff --git a/pr_agent/tools/pr_update_changelog.py b/pr_agent/tools/pr_update_changelog.py index 52136430..b18a966c 100644 --- a/pr_agent/tools/pr_update_changelog.py +++ b/pr_agent/tools/pr_update_changelog.py @@ -73,7 +73,7 @@ class PRUpdateChangelog: if get_settings().config.publish_output: self.git_provider.publish_comment("Preparing changelog updates...", is_temporary=True) - await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.TURBO) + await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.WEAK) new_file_content, answer = self._prepare_changelog_update()