feat: improve prompt

Author: jungs1

src/mutahunter/core/prompts/system.py

@@ -1,4 +1,3 @@
 SYSTEM_PROMPT = """
-# Context:
 You are an AI Agent named Mutanthunter, part of the Software Quality Assurance Team. Your task is to mutate the {{language}} code provided to you. You will be provided with the Abstract Syntax Tree (AST) of the source code for contextual understanding. This AST will help you understand the entire source code. Make sure to read the AST before proceeding with the mutation. 
 """

src/mutahunter/core/prompts/user.py

@@ -5,20 +5,35 @@
 ```
 
 ## Response
-The output must be in JSON format, wrapped in triple backticks (json...), and adhere to the following Pydantic definitions.
+The output must be a JSON object equivalent to type $Mutants, according to the following Pydantic definitions:
 ```
 class SingleMutant(BaseModel):
-    type: str = Field(description="The type of the mutation operator.(e.g., Off-by-One Error, Boundary Condition, Arithmetic, Block removal, Relational Operator, etc.)")
+    function_name: str = Field(description="The name of the function where the mutation was applied.")
+    type: str = Field(description="The type of the mutation operator.")
     description: str = Field(description="Description of the mutation.")
-    context_before: str = Field(description="Line of code context before the mutation.")
     original_line: str = Field(description="The original line of code before mutation.")
     mutated_line: str = Field(description="The line of code after mutation, including a comment with the mutation description.")
-    context_after: str = Field(description="Line of code context after the mutation.")
 
 class Mutants(BaseModel):
     changes: List[Change] = Field(description="A list of changes representing the mutants.")
 ```
 
+## Output
+The output must be in JSON format, wrapped in triple backticks (json...), and adhere to the following Pydantic definitions.
+```
+{
+    "changes": [
+        {
+            'function_name': "divide",
+            "type": "DivisionByZero",
+            'description': "Added division by zero check to prevent division by zero error.",
+            'original_line': "    return a / b",
+            'mutated_line': "    if (b == 0) throw new ArithmeticException("Division by zero"); return a / b; // Mutant: Added division by zero check"
+        }
+    ]
+}
+```
+
 ## Function Block to Mutate
 Lines Covered: {{covered_lines}}. Only mutate lines that are covered by execution.
 Note that we have manually added line numbers for each line of code. Do not include line numbers in your mutation. Make sure indentation is preserverd when generating mutants.