improve regression test selection node

Author: dcloud347

prometheus/lang_graph/nodes/bug_get_regression_tests_selection_node.py

@@ -107,6 +107,10 @@ def format_human_message(self, state: BugGetRegressionTestsState):
         )
 
     def __call__(self, state: BugGetRegressionTestsState):
+        # If no context gathered return empty tests
+        if not state["select_regression_context"]:
+            return {"selected_regression_tests": []}
+
         human_prompt = self.format_human_message(state)
         response = self.model.invoke({"human_prompt": human_prompt})
         self._logger.debug(f"Model response: {response}")

prometheus/lang_graph/subgraphs/bug_get_regression_tests_subgraph.py

@@ -96,8 +96,10 @@ def __init__(
         workflow.add_edge(
             "context_retrieval_subgraph_node", "bug_get_regression_tests_selection_node"
         )
-        workflow.add_edge(
-            "bug_get_regression_tests_selection_node", "run_regression_tests_subgraph_node"
+        workflow.add_conditional_edges(
+            "bug_get_regression_tests_selection_node",
+            lambda state: len(state["selected_regression_tests"]) > 0,
+            {True: "run_regression_tests_subgraph_node", False: END},
         )
         workflow.add_edge("run_regression_tests_subgraph_node", END)