diff --git a/bugbug/tools/code_review/agent.py b/bugbug/tools/code_review/agent.py index 01c448ee29..0520f6b1a8 100644 --- a/bugbug/tools/code_review/agent.py +++ b/bugbug/tools/code_review/agent.py @@ -7,7 +7,7 @@ import json import os -from datetime import datetime +from datetime import UTC, datetime from logging import getLogger from typing import Optional @@ -61,6 +61,10 @@ logger = getLogger(__name__) +def current_date_for_prompt() -> str: + return datetime.now(UTC).date().isoformat() + + class CodeReviewTool(GenerativeModelTool): version = 2 @@ -209,6 +213,7 @@ def generate_initial_prompt( created_before = patch.date_created if self.is_experiment_env else None return FIRST_MESSAGE_TEMPLATE.format( + current_date=current_date_for_prompt(), patch=format_patch_set(patch.patch_set), patch_summarization=patch_summary, external_context=external_context, diff --git a/bugbug/tools/code_review/prompts.py b/bugbug/tools/code_review/prompts.py index 3b91556f7e..59adc4bdf2 100644 --- a/bugbug/tools/code_review/prompts.py +++ b/bugbug/tools/code_review/prompts.py @@ -89,7 +89,9 @@ """ -FIRST_MESSAGE_TEMPLATE = """Here is a summary of the patch: +FIRST_MESSAGE_TEMPLATE = """Current date: {current_date} + +Here is a summary of the patch: {patch_summarization}