refactor: add advanced model parameter to regression test nodes for enhanced analysis capabilities

Author: dcloud347

prometheus/lang_graph/nodes/get_pass_regression_test_patch_subgraph_node.py

@@ -16,7 +16,8 @@
 class GetPassRegressionTestPatchSubgraphNode:
     def __init__(
         self,
-        model: BaseChatModel,
+        base_model: BaseChatModel,
+        advanced_model: BaseChatModel,
         container: BaseContainer,
         git_repo: GitRepository,
         testing_patch_key: str,
@@ -26,7 +27,8 @@ def __init__(
     ):
         self._logger = logging.getLogger(f"thread-{threading.get_ident()}.{__name__}")
         self.subgraph = GetPassRegressionTestPatchSubgraph(
-            base_model=model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
             git_repo=git_repo,
         )

prometheus/lang_graph/nodes/run_regression_tests_subgraph_node.py

@@ -11,11 +11,16 @@
 
 class RunRegressionTestsSubgraphNode:
     def __init__(
-        self, model: BaseChatModel, container: BaseContainer, passed_regression_tests_key: str
+        self,
+        base_model: BaseChatModel,
+        advanced_model: BaseChatModel,
+        container: BaseContainer,
+        passed_regression_tests_key: str,
     ):
         self._logger = logging.getLogger(f"thread-{threading.get_ident()}.{__name__}")
         self.subgraph = RunRegressionTestsSubgraph(
-            base_model=model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
         )
         self.passed_regression_tests_key = passed_regression_tests_key

prometheus/lang_graph/subgraphs/bug_get_regression_tests_subgraph.py

@@ -68,7 +68,8 @@ def __init__(
         )
         # Step 4: Run the selected regression tests before the fix and store results
         run_regression_tests_subgraph_node = RunRegressionTestsSubgraphNode(
-            model=base_model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
             passed_regression_tests_key="selected_regression_tests",
         )

prometheus/lang_graph/subgraphs/get_pass_regression_test_patch_subgraph.py

@@ -31,6 +31,7 @@ class GetPassRegressionTestPatchSubgraph:
     def __init__(
         self,
         base_model: BaseChatModel,
+        advanced_model: BaseChatModel,
         container: BaseContainer,
         git_repo: GitRepository,
     ):
@@ -39,7 +40,9 @@ def __init__(
 
         Args:
             base_model: Lighter LLM for simpler tasks (e.g., file selection).
+            advanced_model: More capable LLM for complex analysis tasks.
             container: Docker-based sandbox for running code.
+            git_repo: Git repository interface for codebase manipulation.
         """
         noop_nodes = NoopNode()
         # Step 1: Update the Git repository with the current testing patch
@@ -50,7 +53,8 @@ def __init__(
         update_container_node = UpdateContainerNode(container=container, git_repo=git_repo)
         # Step 3: Run regression tests on the current testing patch
         run_regression_tests_subgraph_node = RunRegressionTestsSubgraphNode(
-            model=base_model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
             passed_regression_tests_key="current_passed_tests",
         )

prometheus/lang_graph/subgraphs/issue_feature_subgraph.py

@@ -127,7 +127,8 @@ def __init__(
 
         # Phase 5: Optional regression test validation
         get_pass_regression_test_patch_subgraph_node = GetPassRegressionTestPatchSubgraphNode(
-            model=base_model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
             git_repo=git_repo,
             testing_patch_key="final_candidate_patches",

prometheus/lang_graph/subgraphs/issue_not_verified_bug_subgraph.py

@@ -74,7 +74,8 @@ def __init__(
 
         # Get pass regression test patch subgraph node
         get_pass_regression_test_patch_subgraph_node = GetPassRegressionTestPatchSubgraphNode(
-            model=base_model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
             git_repo=git_repo,
             testing_patch_key="final_candidate_patches",

prometheus/lang_graph/subgraphs/issue_verified_bug_subgraph.py

@@ -109,7 +109,8 @@ def __init__(
         # Phase 5: Run Regression Tests if available
         get_pass_regression_test_patch_branch_node = NoopNode()
         get_pass_regression_test_patch_subgraph_node = GetPassRegressionTestPatchSubgraphNode(
-            model=base_model,
+            base_model=base_model,
+            advanced_model=advanced_model,
             container=container,
             git_repo=git_repo,
             testing_patch_key="edit_patch",

prometheus/lang_graph/subgraphs/run_regression_tests_subgraph.py

@@ -22,13 +22,15 @@ class RunRegressionTestsSubgraph:
     def __init__(
         self,
         base_model: BaseChatModel,
+        advanced_model: BaseChatModel,
         container: BaseContainer,
     ):
         """
         Initialize the run regression tests pipeline with all necessary parts.
 
         Args:
             base_model: Lighter LLM for simpler tasks (e.g., file selection).
+            advanced_model: More capable LLM for complex analysis tasks.
             container: Docker-based sandbox for running code.
         """
         # Run regression tests node
@@ -38,7 +40,9 @@ def __init__(
             name="run_regression_tests_tools",
             messages_key="run_regression_tests_messages",
         )
-        run_regression_tests_structured_node = RunRegressionTestsStructuredNode(model=base_model)
+        run_regression_tests_structured_node = RunRegressionTestsStructuredNode(
+            model=advanced_model
+        )
         # Define the state machine
         workflow = StateGraph(RunRegressionTestsState)
         workflow.add_node("run_regression_tests_node", run_regression_tests_node)