Use regular model for pr review and code suggestion flows

This commit is contained in:
Trung Dinh
2024-12-08 21:51:09 +07:00
parent dec2859fc4
commit 936894e4d1
2 changed files with 3 additions and 3 deletions

View File

@ -111,9 +111,9 @@ class PRCodeSuggestions:
self.git_provider.publish_comment("Preparing suggestions...", is_temporary=True)
if not self.is_extended:
data = await retry_with_fallback_models(self._prepare_prediction)
data = await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.REGULAR)
else:
data = await retry_with_fallback_models(self._prepare_prediction_extended)
data = await retry_with_fallback_models(self._prepare_prediction_extended, model_type=ModelType.REGULAR)
if not data:
data = {"code_suggestions": []}