fix: refine prompts and add additional package info (#1179)

* refine prompts and add additional package info

* refine prompts to be specific for GBDT models

* minor refine prompts

* use include to replace duplicate info

* refine prompts

* refactor: import DSTrace from base and remove exp_gen __init__

* lint

---------

Co-authored-by: Young <afe.young@gmail.com>
This commit is contained in:
amstrongzyf
2025-08-13 23:00:23 +08:00
committed by GitHub
parent e0f713c7d3
commit 5353bd31f2
11 changed files with 143 additions and 43 deletions
@@ -13,9 +13,9 @@ pipeline_coder:
{{ runtime_environment }}
{% if package_info is not none %}
To help you write the runnable code, the user has provided the package information which contains the package names and versions.
You should be careful about the package versions, as the code will be executed in the environment with the specified version and the api might be different from the latest version.
The user might provide the packages the environment doesn't have, you should avoid using any of them.
- To help you write the runnable code, the user has provided the package information which contains the package names and versions.
- You should be careful about the package versions, as the code will be executed in the environment with the specified version and the api might be different from the latest version.
- While the environment is fixed, you should not limit yourself to only the provided packages - feel free to explore other libraries that might better suit the task. However, prioritize using the available packages first, and only suggest alternatives when they would provide significant improvements or are more appropriate for the specific problem.
## Package Information
{{ package_info }}
{% endif %}