feat: add package query in draft.py (not yet enabled) #1083
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
peteryang1
commented
Jul 17, 2025
| # query their versions dynamically. | ||
| # --------------------------------------------------------------------- | ||
|
|
||
| def _package_query(self, scenario_desc: str) -> list[str]: |
Collaborator
Author
There was a problem hiding this comment.
move this function to rdagent/scenarios/data_science/proposal/exp_gen/utils.py
Collaborator
There was a problem hiding this comment.
I have now moved the functionality of this function to rdagent/scenarios/data_science/proposal/exp_gen/proposal.py, achieving it through one interaction with the Backend API. Therefore, I deem it unnecessary to implement it separately.
rdagent/scenarios/data_science/proposal/exp_gen/draft/prompts_draft.yaml
Outdated
Show resolved
Hide resolved
| ) | ||
|
|
||
| def get_runtime_environment(self) -> str: | ||
| def get_runtime_environment(self, pkgs: list[str] | None = None) -> str: |
Collaborator
Author
There was a problem hiding this comment.
make this function callable from docker
Collaborator
There was a problem hiding this comment.
This function wasn’t docker-callable before my changes.
Do we really need Docker support here, or can we just keep it simple for now?
…ment - Remove pkg_query modifications from draft components - Add package declaration requirement in task_gen prompts - Add optional packages field to CodingSketch model - Cache runtime_environment in scenario object for loop-wide reuse - Parse packages from LLM response and generate runtime environment dynamically
peteryang1
commented
Jul 17, 2025
| "torch", | ||
| "tensorflow", | ||
| "pandas", | ||
| "numpy", |
Collaborator
Author
There was a problem hiding this comment.
list(set(A) + set(B))
licong01-cloud
pushed a commit
to licong01-cloud/RD-Agent
that referenced
this pull request
Dec 13, 2025
* feat: add package query in draft.py (not yet enabled) * feat: integrate package query into task_gen and cache runtime environment - Remove pkg_query modifications from draft components - Add package declaration requirement in task_gen prompts - Add optional packages field to CodingSketch model - Cache runtime_environment in scenario object for loop-wide reuse - Parse packages from LLM response and generate runtime environment dynamically * some refinement * feat: merge default packages with CLI args in package_info.py * fix: code style --------- Co-authored-by: Qizheng Li <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
Motivation and Context
How Has This Been Tested?
Screenshots of Test Results (if appropriate):
Types of changes
📚 Documentation preview 📚: https://RDAgent--1083.org.readthedocs.build/en/1083/