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

docs(agent): Add agents development guide (#1673)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-06-28 09:56:29 +08:00
committed by GitHub
parent 1796969bac
commit 45a960f331
43 changed files with 4010 additions and 146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/agent/resource/database.py
View File

@@ -3,16 +3,18 @@
import dataclasses
import logging
from concurrent.futures import Executor, ThreadPoolExecutor
from typing import Any, Generic, List, Optional, Tuple, Union
from typing import TYPE_CHECKING, Any, Generic, List, Optional, Tuple, Union
import cachetools
from dbgpt.datasource.rdbms.base import RDBMSConnector
from dbgpt.util.cache_utils import cached
from dbgpt.util.executor_utils import blocking_func_to_async
from .base import P, Resource, ResourceParameters, ResourceType
if TYPE_CHECKING:
from dbgpt.datasource.rdbms.base import RDBMSConnector
logger = logging.getLogger(__name__)
_DEFAULT_PROMPT_TEMPLATE = (
@@ -143,7 +145,7 @@ class RDBMSConnectorResource(DBResource[DBParameters]):
def __init__(
self,
name: str,
connector: Optional[RDBMSConnector] = None,
connector: Optional["RDBMSConnector"] = None,
db_name: Optional[str] = None,
db_type: Optional[str] = None,
dialect: Optional[str] = None,
@@ -168,7 +170,7 @@ class RDBMSConnectorResource(DBResource[DBParameters]):
)
@property
def connector(self) -> RDBMSConnector:
def connector(self) -> "RDBMSConnector":
"""Return the connector."""
if not self._connector:
raise ValueError("Connector is not set.")