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

fix: replace deprecated method and add missing decorator (#2072)

Browse Source 
Co-authored-by: songguodong <songguodong@songguodongdeMacBook-Air.local>
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
Guodong
2024-10-18 16:23:28 +08:00
committed by GitHub
parent c398fb64c4
commit 253c367ceb
3 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dbgpt/agent/core/action/base.py
View File

@@ -72,7 +72,7 @@ class ActionOutput(BaseModel):
"""Convert dict to ActionOutput object."""
if not param:
return None
return cls.parse_obj(param)
return cls.model_validate(param)
def to_dict(self) -> Dict[str, Any]:
"""Convert the object to a dictionary."""
@@ -181,10 +181,12 @@ class Action(ABC, Generic[T]):
if get_origin(cls) == list:
inner_type = get_args(cls)[0]
typed_cls = cast(Type[BaseModel], inner_type)
return [typed_cls.parse_obj(item) for item in json_result] # type: ignore
return [
typed_cls.model_validate(item) for item in json_result
] # type: ignore
else:
typed_cls = cast(Type[BaseModel], cls)
return typed_cls.parse_obj(json_result)
return typed_cls.model_validate(json_result)
@abstractmethod
async def run(