From bd5c19ee05eb2102cea1e9be0c384aed4fba2d55 Mon Sep 17 00:00:00 2001 From: "trung.dinh@zoi.tech" Date: Sun, 18 Aug 2024 20:29:59 +0700 Subject: [PATCH 1/3] Improve variable consistency in prompt rendering for pr description flow --- pr_agent/tools/pr_description.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pr_agent/tools/pr_description.py b/pr_agent/tools/pr_description.py index 2bcbd89e..1b3f33c1 100644 --- a/pr_agent/tools/pr_description.py +++ b/pr_agent/tools/pr_description.py @@ -348,8 +348,8 @@ extra_file_yaml = set_custom_labels(variables, self.git_provider) self.variables = variables - system_prompt = environment.from_string(get_settings().get(prompt, {}).get("system", "")).render(variables) - user_prompt = environment.from_string(get_settings().get(prompt, {}).get("user", "")).render(variables) + system_prompt = environment.from_string(get_settings().get(prompt, {}).get("system", "")).render(self.variables) + user_prompt = environment.from_string(get_settings().get(prompt, {}).get("user", "")).render(self.variables) response, finish_reason = await self.ai_handler.chat_completion( model=model, From f9de8f283bbe7a8889cd5d4b159ce0f60840c4b8 Mon Sep 17 00:00:00 2001 From: "trung.dinh@zoi.tech" Date: Sun, 18 Aug 2024 20:30:10 +0700 Subject: [PATCH 2/3] Improve variable consistency in prompt rendering for pr generate labels flow --- pr_agent/tools/pr_generate_labels.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pr_agent/tools/pr_generate_labels.py b/pr_agent/tools/pr_generate_labels.py index 2657a857..47b6dd36 100644 --- a/pr_agent/tools/pr_generate_labels.py +++ b/pr_agent/tools/pr_generate_labels.py @@ -137,8 +137,8 @@ class PRGenerateLabels: 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, From dc8a4be2d456081dc989ff366c9a5243ffb5f37a Mon Sep 17 00:00:00 2001 From: "trung.dinh@zoi.tech" Date: Sun, 18 Aug 2024 20:38:00 +0700 Subject: [PATCH 3/3] Add new line for a better visually --- pr_agent/tools/pr_generate_labels.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pr_agent/tools/pr_generate_labels.py b/pr_agent/tools/pr_generate_labels.py index 47b6dd36..1911e0cd 100644 --- a/pr_agent/tools/pr_generate_labels.py +++ b/pr_agent/tools/pr_generate_labels.py @@ -137,6 +137,7 @@ class PRGenerateLabels: 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(self.variables) user_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.user).render(self.variables)