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

refactor(agent): Refactor resource of agents (#1518)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-15 09:57:19 +08:00
committed by GitHub
parent db4d318a5f
commit 559affe87d
102 changed files with 2633 additions and 2549 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -27,8 +27,7 @@ from dbgpt._private.pydantic import (
from dbgpt.util.json_utils import find_json_objects
from dbgpt.vis.base import Vis
from ...resource.resource_api import AgentResource, ResourceType
from ...resource.resource_loader import ResourceLoader
from ...resource.base import AgentResource, Resource, ResourceType
T = TypeVar("T", bound=Union[BaseModel, List[BaseModel], None])
@@ -77,11 +76,11 @@ class Action(ABC, Generic[T]):
def __init__(self):
"""Create an action."""
self.resource_loader: Optional[ResourceLoader] = None
self.resource: Optional[Resource] = None
def init_resource_loader(self, resource_loader: Optional[ResourceLoader]):
"""Initialize the resource loader."""
self.resource_loader = resource_loader
def init_resource(self, resource: Optional[Resource]):
"""Initialize the resource."""
self.resource = resource
@property
def resource_need(self) -> Optional[ResourceType]:

2
dbgpt/agent/core/action/blank_action.py
View File

@@ -3,7 +3,7 @@
import logging
from typing import Optional
from ...resource.resource_api import AgentResource
from ...resource.base import AgentResource
from .base import Action, ActionOutput
logger = logging.getLogger(__name__)