fix a small bug

Author: Hoder-zyf

rdagent/scenarios/data_science/proposal/exp_gen/package_info.py

@@ -154,5 +154,15 @@ def get_python_packages():
             print(pkg)
 
 
+def print_available_packages_prompt():
+    """Print the available packages prompt to stdout for external consumption"""
+    prompt = get_available_packages_prompt()
+    print(prompt)
+
+
 if __name__ == "__main__":
-    get_python_packages()
+    # Check if we should print available packages prompt
+    if len(sys.argv) > 1 and sys.argv[1] == "--packages-prompt":
+        print_available_packages_prompt()
+    else:
+        get_python_packages()

rdagent/scenarios/data_science/proposal/exp_gen/proposal.py

@@ -24,14 +24,13 @@
 )
 from rdagent.scenarios.data_science.proposal.exp_gen.idea_pool import DSIdea
 from rdagent.scenarios.data_science.proposal.exp_gen.package_info import (
-    get_available_packages_prompt,
     get_persistent_problem_guidelines,
 )
 from rdagent.scenarios.data_science.proposal.exp_gen.planner import (
     DSExperimentPlan,
     RD_Agent_TIMER_wrapper,
 )
-from rdagent.scenarios.data_science.proposal.exp_gen.utils import get_packages
+from rdagent.scenarios.data_science.proposal.exp_gen.utils import get_available_packages_prompt, get_packages
 from rdagent.utils.agent.tpl import T
 from rdagent.utils.repo.diff import generate_diff_from_dict
 from rdagent.utils.workflow import wait_retry

rdagent/scenarios/data_science/proposal/exp_gen/utils.py

@@ -103,3 +103,16 @@ def get_packages(pkgs: list[str] | None = None) -> str:
     pkg_args = " ".join(pkgs) if pkgs else ""
     stdout = implementation.execute(env=env, entry=f"python {fname} {pkg_args}")
     return stdout
+
+
+def get_available_packages_prompt() -> str:
+    """Generate prompt template for dynamically detected available packages."""
+    # Use the same approach as get_packages but call the packages prompt functionality
+    
+    env = get_ds_env()
+    implementation = FBWorkspace()
+    fname = "package_info.py"
+    implementation.inject_files(**{fname: (Path(__file__).absolute().resolve().parent / "package_info.py").read_text()})
+    
+    stdout = implementation.execute(env=env, entry=f"python {fname} --packages-prompt")
+    return stdout.strip()