diff --git a/pr_agent/tools/pr_generate_labels.py b/pr_agent/tools/pr_generate_labels.py index 2657a857b..47b6dd36e 100644 --- a/pr_agent/tools/pr_generate_labels.py +++ b/pr_agent/tools/pr_generate_labels.py @@ -137,8 +137,8 @@ async def _get_prediction(self, model: str) -> str: environment = Environment(undefined=StrictUndefined) set_custom_labels(variables, self.git_provider) self.variables = variables - system_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.system).render(variables) - user_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.user).render(variables) + system_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.system).render(self.variables) + user_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.user).render(self.variables) response, finish_reason = await self.ai_handler.chat_completion( model=model,