File tree Expand file tree Collapse file tree 2 files changed +6
-3
lines changed
rdagent/scenarios/data_science/proposal/exp_gen Expand file tree Collapse file tree 2 files changed +6
-3
lines changed Original file line number Diff line number Diff line change 30
30
DSExperimentPlan ,
31
31
RD_Agent_TIMER_wrapper ,
32
32
)
33
- from rdagent .scenarios .data_science .proposal .exp_gen .utils import get_available_packages_prompt , get_packages
33
+ from rdagent .scenarios .data_science .proposal .exp_gen .utils import (
34
+ get_available_packages_prompt ,
35
+ get_packages ,
36
+ )
34
37
from rdagent .utils .agent .tpl import T
35
38
from rdagent .utils .repo .diff import generate_diff_from_dict
36
39
from rdagent .utils .workflow import wait_retry
Original file line number Diff line number Diff line change @@ -108,11 +108,11 @@ def get_packages(pkgs: list[str] | None = None) -> str:
108
108
def get_available_packages_prompt () -> str :
109
109
"""Generate prompt template for dynamically detected available packages."""
110
110
# Use the same approach as get_packages but call the packages prompt functionality
111
-
111
+
112
112
env = get_ds_env ()
113
113
implementation = FBWorkspace ()
114
114
fname = "package_info.py"
115
115
implementation .inject_files (** {fname : (Path (__file__ ).absolute ().resolve ().parent / "package_info.py" ).read_text ()})
116
-
116
+
117
117
stdout = implementation .execute (env = env , entry = f"python { fname } --packages-prompt" )
118
118
return stdout .strip ()
You can’t perform that action at this time.
0 commit comments