ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-13 11:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1154 from KennyDizi/main

Browse source 
Enhance variable consistency in prompt rendering for PR tools
This commit is contained in:
Tal 2024-08-19 08:48:10 +03:00 committed by GitHub
commit 2e186ebae8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pr_agent/tools/pr_description.py
View file

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

5
pr_agent/tools/pr_generate_labels.py
View file

@ -137,8 +137,9 @@ 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,