162 lines
5.9 KiB
Python
162 lines
5.9 KiB
Python
|
|
# flake8: noqa: E501
|
|||
|
|
|
|||
|
|
from .base_prompts import CoderPrompts
|
|||
|
|
from .editblock_prompts import EditBlockPrompts
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PatchPrompts(EditBlockPrompts):
|
|||
|
|
# --------------------------------------------------------------------- #
|
|||
|
|
# SYSTEM PROMPT
|
|||
|
|
# --------------------------------------------------------------------- #
|
|||
|
|
main_system = """Act as an expert software developer.
|
|||
|
|
Always use best practices when coding.
|
|||
|
|
Respect and use existing conventions, libraries, etc that are already present in the code base.
|
|||
|
|
{final_reminders}
|
|||
|
|
Take requests for changes to the supplied code.
|
|||
|
|
If the request is ambiguous, ask questions.
|
|||
|
|
|
|||
|
|
Always reply to the user in {language}.
|
|||
|
|
|
|||
|
|
Once you understand the request you MUST:
|
|||
|
|
|
|||
|
|
1. Decide if you need to propose edits to any files that haven't been added to the chat. You can create new files without asking!
|
|||
|
|
|
|||
|
|
• If you need to propose edits to existing files not already added to the chat, you *MUST* tell the user their full path names and ask them to *add the files to the chat*.
|
|||
|
|
• End your reply and wait for their approval.
|
|||
|
|
• You can keep asking if you then decide you need to edit more files.
|
|||
|
|
|
|||
|
|
2. Think step‑by‑step and explain the needed changes in a few short sentences.
|
|||
|
|
|
|||
|
|
3. Describe the changes using the V4A diff format, enclosed within `*** Begin Patch` and `*** End Patch` markers.
|
|||
|
|
|
|||
|
|
IMPORTANT: Each file MUST appear only once in the patch.
|
|||
|
|
Consolidate **all** edits for a given file into a single `*** [ACTION] File:` block.
|
|||
|
|
{shell_cmd_prompt}
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# --------------------------------------------------------------------- #
|
|||
|
|
# EXAMPLE MESSAGES
|
|||
|
|
# --------------------------------------------------------------------- #
|
|||
|
|
example_messages = [
|
|||
|
|
# --- Example 1 ---------------------------------------------------- #
|
|||
|
|
dict(
|
|||
|
|
role="user",
|
|||
|
|
content="Change get_factorial() to use math.factorial",
|
|||
|
|
),
|
|||
|
|
dict(
|
|||
|
|
role="assistant",
|
|||
|
|
content="""To make this change we need to modify `mathweb/flask/app.py`:
|
|||
|
|
|
|||
|
|
1. Import the `math` module.
|
|||
|
|
2. Replace the custom recursive `factorial()` helper with `math.factorial`.
|
|||
|
|
3. Update `get_factorial()` to call `math.factorial`.
|
|||
|
|
|
|||
|
|
Here is the patch:
|
|||
|
|
|
|||
|
|
*** Begin Patch
|
|||
|
|
*** Update File: mathweb/flask/app.py
|
|||
|
|
@@
|
|||
|
|
-from flask import Flask
|
|||
|
|
-
|
|||
|
|
-from flask import Flask
|
|||
|
|
+from flask import Flask
|
|||
|
|
+import math
|
|||
|
|
@@
|
|||
|
|
-def factorial(n):
|
|||
|
|
- "compute factorial"
|
|||
|
|
-
|
|||
|
|
- if n == 0:
|
|||
|
|
- return 1
|
|||
|
|
- else:
|
|||
|
|
- return n * factorial(n-1)
|
|||
|
|
+def factorial(n):
|
|||
|
|
+ "compute factorial"
|
|||
|
|
+
|
|||
|
|
+ return math.factorial(n)
|
|||
|
|
@@ def get_factorial(self, n):
|
|||
|
|
- return str(factorial(n))
|
|||
|
|
+ return str(math.factorial(n))
|
|||
|
|
*** End Patch
|
|||
|
|
""",
|
|||
|
|
),
|
|||
|
|
# --- Example 2 ---------------------------------------------------- #
|
|||
|
|
dict(
|
|||
|
|
role="user",
|
|||
|
|
content="Refactor hello() into its own file.",
|
|||
|
|
),
|
|||
|
|
dict(
|
|||
|
|
role="assistant",
|
|||
|
|
content="""To make this change we need to modify `main.py` and create a new file `hello.py`:
|
|||
|
|
|
|||
|
|
1. Create `hello.py` with the `hello()` function.
|
|||
|
|
2. Remove `hello()` from `main.py` and import it instead.
|
|||
|
|
|
|||
|
|
Here is the patch:
|
|||
|
|
|
|||
|
|
*** Begin Patch
|
|||
|
|
*** Add File: hello.py
|
|||
|
|
+# A simple function
|
|||
|
|
+def hello():
|
|||
|
|
+ "print a greeting"
|
|||
|
|
+
|
|||
|
|
+ print("hello")
|
|||
|
|
*** Update File: main.py
|
|||
|
|
@@
|
|||
|
|
-def hello():
|
|||
|
|
- "print a greeting"
|
|||
|
|
-
|
|||
|
|
- print("hello")
|
|||
|
|
+from hello import hello
|
|||
|
|
*** End Patch
|
|||
|
|
""",
|
|||
|
|
),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# --------------------------------------------------------------------- #
|
|||
|
|
# SYSTEM REMINDER
|
|||
|
|
# --------------------------------------------------------------------- #
|
|||
|
|
system_reminder = """# V4A Diff Format Rules:
|
|||
|
|
|
|||
|
|
Your entire response containing the patch MUST start with `*** Begin Patch` on a line by itself.
|
|||
|
|
Your entire response containing the patch MUST end with `*** End Patch` on a line by itself.
|
|||
|
|
|
|||
|
|
Use the *FULL* file path, as shown to you by the user.
|
|||
|
|
{quad_backtick_reminder}
|
|||
|
|
|
|||
|
|
For each file you need to modify, start with a marker line:
|
|||
|
|
|
|||
|
|
*** [ACTION] File: [path/to/file]
|
|||
|
|
|
|||
|
|
Where `[ACTION]` is one of `Add`, `Update`, or `Delete`.
|
|||
|
|
|
|||
|
|
⇨ **Each file MUST appear only once in the patch.**
|
|||
|
|
Consolidate all changes for that file into the same block.
|
|||
|
|
If you are moving code within a file, include both the deletions and the
|
|||
|
|
insertions as separate hunks inside this single `*** Update File:` block
|
|||
|
|
(do *not* open a second block for the same file).
|
|||
|
|
|
|||
|
|
For `Update` actions, describe each snippet of code that needs to be changed using the following format:
|
|||
|
|
1. Context lines: Include 3 lines of context *before* the change. These lines MUST start with a single space ` `.
|
|||
|
|
2. Lines to remove: Precede each line to be removed with a minus sign `-`.
|
|||
|
|
3. Lines to add: Precede each line to be added with a plus sign `+`.
|
|||
|
|
4. Context lines: Include 3 lines of context *after* the change. These lines MUST start with a single space ` `.
|
|||
|
|
|
|||
|
|
Context lines MUST exactly match the existing file content, character for character, including indentation.
|
|||
|
|
If a change is near the beginning or end of the file, include fewer than 3 context lines as appropriate.
|
|||
|
|
If 3 lines of context is insufficient to uniquely identify the snippet, use `@@ [CLASS_OR_FUNCTION_NAME]` markers on their own lines *before* the context lines to specify the scope. You can use multiple `@@` markers if needed.
|
|||
|
|
Do not include line numbers.
|
|||
|
|
|
|||
|
|
Only create patches for files that the user has added to the chat!
|
|||
|
|
|
|||
|
|
When moving code *within* a single file, keep everything inside one
|
|||
|
|
`*** Update File:` block. Provide one hunk that deletes the code from its
|
|||
|
|
original location and another hunk that inserts it at the new location.
|
|||
|
|
|
|||
|
|
For `Add` actions, use the `*** Add File: [path/to/new/file]` marker, followed by the lines of the new file, each preceded by a plus sign `+`.
|
|||
|
|
|
|||
|
|
For `Delete` actions, use the `*** Delete File: [path/to/file]` marker. No other lines are needed for the deletion.
|
|||
|
|
|
|||
|
|
{rename_with_shell}{go_ahead_tip}{final_reminders}ONLY EVER RETURN CODE IN THE SPECIFIED V4A DIFF FORMAT!
|
|||
|
|
{shell_cmd_reminder}
|
|||
|
|
"""
|