github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-06 19:40:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(runtime): Execute codes in a sandbox environment (#2119)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-11-10 22:36:53 +08:00
committed by GitHub
parent 8593f10ae9
commit a134257e5c
22 changed files with 867 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
setup.py
View File

@@ -517,13 +517,15 @@ def code_execution_requires():
"""
pip install "dbgpt[code]"
Code execution dependencies. For building a docker image.
Code execution dependencies.
"""
setup_spec.extras["code"] = setup_spec.extras["core"] + [
"pyzmq",
"msgpack",
# for AWEL operator serialization
"cloudpickle",
"lyric-py>=0.1.4",
"lyric-py-worker>=0.1.4",
"lyric-js-worker>=0.1.4",
]
@@ -723,6 +725,7 @@ def default_requires():
setup_spec.extras["default"] += setup_spec.extras["datasource"]
setup_spec.extras["default"] += setup_spec.extras["torch"]
setup_spec.extras["default"] += setup_spec.extras["cache"]
setup_spec.extras["default"] += setup_spec.extras["code"]
if INCLUDE_QUANTIZATION:
# Add quantization extra to default, default is True
setup_spec.extras["default"] += setup_spec.extras["quantization"]