github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-17 15:10:14 +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

4
dbgpt/core/awel/flow/base.py
View File

@@ -151,6 +151,7 @@ _OPERATOR_CATEGORY_DETAIL = {
"database": _CategoryDetail("Database", "Interact with the database"),
"type_converter": _CategoryDetail("Type Converter", "Convert the type"),
"example": _CategoryDetail("Example", "Example operator"),
"code": _CategoryDetail("Code", "Code operator"),
}
@@ -169,6 +170,7 @@ class OperatorCategory(str, Enum):
DATABASE = "database"
TYPE_CONVERTER = "type_converter"
EXAMPLE = "example"
CODE = "code"
def label(self) -> str:
"""Get the label of the category."""
@@ -1361,6 +1363,8 @@ def _register_operator(view_cls: Optional[Type[T]]):
"""Register the operator."""
if not view_cls or not view_cls.metadata:
return
if "metadata" not in view_cls.__dict__:
return # Skip the base class
metadata = view_cls.metadata
metadata.type_name = view_cls.__qualname__
metadata.type_cls = _get_type_name(view_cls)