Improve variable consistency in prompt rendering for pr generate labels flow

This commit is contained in:
trung.dinh@zoi.tech
2024-08-18 20:30:10 +07:00
parent bd5c19ee05
commit f9de8f283b

View File

@ -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,