Add run_reproduce_test option to issue handling and update related descriptions

Author: dcloud347

prometheus/app/api/issue.py

@@ -47,6 +47,7 @@ def answer_issue(issue: IssueRequest, request: Request) -> Response[IssueRespons
         issue_type=issue.issue_type,
         run_build=issue.run_build,
         run_existing_test=issue.run_existing_test,
+        run_reproduce_test=issue.run_reproduce_test,
         number_of_candidate_patch=issue.number_of_candidate_patch,
         dockerfile_content=issue.dockerfile_content,
         image_name=issue.image_name,

prometheus/app/models/requests/issue.py

@@ -33,12 +33,17 @@ class IssueRequest(BaseModel):
     )
     run_build: Optional[bool] = Field(
         default=False,
-        description="When editing the code, whenver we should run the build to verify the fix",
+        description="When editing the code, whenever we should run the build to verify the fix",
         examples=[False],
     )
     run_existing_test: Optional[bool] = Field(
         default=False,
-        description="When editing the code, whenver we should run the existing test to verify the fix",
+        description="When editing the code, whenever we should run the existing test to verify the fix",
+        examples=[False],
+    )
+    run_reproduce_test: Optional[bool] = Field(
+        default=True,
+        description="When editing the code, whenever we should run the reproduce test to verify the fix",
         examples=[False],
     )
     number_of_candidate_patch: Optional[int] = Field(

prometheus/app/services/issue_service.py

@@ -43,6 +43,7 @@ def answer_issue(
         issue_type: IssueType,
         run_build: bool,
         run_existing_test: bool,
+        run_reproduce_test: bool,
         number_of_candidate_patch: int,
         dockerfile_content: Optional[str] = None,
         image_name: Optional[str] = None,
@@ -62,6 +63,7 @@ def answer_issue(
             issue_type (IssueType): The type of the issue (BUG or QUESTION).
             run_build (bool): Whether to run the build commands.
             run_existing_test (bool): Whether to run existing tests.
+            run_reproduce_test (bool): Whether to run reproduce tests.
             number_of_candidate_patch (int): Number of candidate patches to generate.
             dockerfile_content (Optional[str]): Content of the Dockerfile for user-defined environments.
             image_name (Optional[str]): Name of the Docker image.
@@ -118,6 +120,7 @@ def answer_issue(
                 issue_type,
                 run_build,
                 run_existing_test,
+                run_reproduce_test,
                 number_of_candidate_patch,
             )
 

prometheus/lang_graph/graphs/issue_graph.py

@@ -102,6 +102,7 @@ def invoke(
         issue_type: IssueType,
         run_build: bool,
         run_existing_test: bool,
+        run_reproduce_test: bool,
         number_of_candidate_patch: int,
     ):
         """
@@ -116,6 +117,7 @@ def invoke(
             "issue_type": issue_type,
             "run_build": run_build,
             "run_existing_test": run_existing_test,
+            "run_reproduce_test": run_reproduce_test,
             "number_of_candidate_patch": number_of_candidate_patch,
         }
 

prometheus/lang_graph/graphs/issue_state.py

@@ -18,6 +18,7 @@ class IssueState(TypedDict):
     issue_type: IssueType
     run_build: bool
     run_existing_test: bool
+    run_reproduce_test: bool
     number_of_candidate_patch: int
 
     edit_patch: str

prometheus/lang_graph/nodes/issue_bug_subgraph_node.py

@@ -57,6 +57,7 @@ def __call__(self, state: IssueState):
                 issue_comments=state["issue_comments"],
                 run_build=state["run_build"],
                 run_existing_test=state["run_existing_test"],
+                run_reproduce_test=state["run_reproduce_test"],
                 number_of_candidate_patch=state["number_of_candidate_patch"],
             )
 

prometheus/lang_graph/subgraphs/issue_bug_state.py

@@ -7,6 +7,7 @@ class IssueBugState(TypedDict):
     issue_comments: Sequence[Mapping[str, str]]
     run_build: bool
     run_existing_test: bool
+    run_reproduce_test: bool
     number_of_candidate_patch: int
 
     reproduced_bug: bool

prometheus/lang_graph/subgraphs/issue_bug_subgraph.py

@@ -77,8 +77,17 @@ def __init__(
         )
 
         workflow.add_node("issue_bug_responder_node", issue_bug_responder_node)
-        # Set the entry point for the workflow
-        workflow.set_entry_point("bug_reproduction_subgraph_node")
+
+        # Start with bug reproduction subgraph if reproducing bug is required,
+        # otherwise start with not verified bug subgraph
+        workflow.set_conditional_entry_point(
+            lambda state: state["run_reproduce_test"],
+            {
+                True: "bug_reproduction_subgraph_node",
+                False: "issue_not_verified_bug_subgraph_node",
+            },
+        )
+
         # Go to verified bug subgraph if the bug is verified, otherwise go to not verified bug subgraph
         workflow.add_conditional_edges(
             "bug_reproduction_subgraph_node",
@@ -109,6 +118,7 @@ def invoke(
         issue_comments: Sequence[Mapping[str, str]],
         run_build: bool,
         run_existing_test: bool,
+        run_reproduce_test: bool,
         number_of_candidate_patch: int,
         recursion_limit: int = 30,
     ):
@@ -120,6 +130,7 @@ def invoke(
             "issue_comments": issue_comments,
             "run_build": run_build,
             "run_existing_test": run_existing_test,
+            "run_reproduce_test": run_reproduce_test,
             "number_of_candidate_patch": number_of_candidate_patch,
         }