mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-15 14:11:14 +00:00
feat(agent):Fix agent bug (#1953)
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
@@ -31,9 +31,9 @@ class SqlInput(BaseModel):
|
||||
class ChartAction(Action[SqlInput]):
|
||||
"""Chart action class."""
|
||||
|
||||
def __init__(self):
|
||||
"""Create a chart action."""
|
||||
super().__init__()
|
||||
def __init__(self, **kwargs):
|
||||
"""Chart action init."""
|
||||
super().__init__(**kwargs)
|
||||
self._render_protocol = VisChart()
|
||||
|
||||
@property
|
||||
|
@@ -16,9 +16,9 @@ logger = logging.getLogger(__name__)
|
||||
class CodeAction(Action[None]):
|
||||
"""Code Action Module."""
|
||||
|
||||
def __init__(self):
|
||||
"""Create a code action."""
|
||||
super().__init__()
|
||||
def __init__(self, **kwargs):
|
||||
"""Code action init."""
|
||||
super().__init__(**kwargs)
|
||||
self._render_protocol = VisCode()
|
||||
self._code_execution_config = {}
|
||||
|
||||
|
@@ -39,9 +39,9 @@ class ChartItem(BaseModel):
|
||||
class DashboardAction(Action[List[ChartItem]]):
|
||||
"""Dashboard action class."""
|
||||
|
||||
def __init__(self):
|
||||
"""Create a dashboard action."""
|
||||
super().__init__()
|
||||
def __init__(self, **kwargs):
|
||||
"""Dashboard action init."""
|
||||
super().__init__(**kwargs)
|
||||
self._render_protocol = VisDashboard()
|
||||
|
||||
@property
|
||||
|
@@ -41,9 +41,9 @@ class IndicatorInput(BaseModel):
|
||||
class IndicatorAction(Action[IndicatorInput]):
|
||||
"""Indicator Action."""
|
||||
|
||||
def __init__(self):
|
||||
"""Init Indicator Action."""
|
||||
super().__init__()
|
||||
def __init__(self, **kwargs):
|
||||
"""Init indicator action."""
|
||||
super().__init__(**kwargs)
|
||||
self._render_protocol = VisApiResponse()
|
||||
|
||||
@property
|
||||
|
@@ -34,9 +34,9 @@ class ToolInput(BaseModel):
|
||||
class ToolAction(Action[ToolInput]):
|
||||
"""Tool action class."""
|
||||
|
||||
def __init__(self):
|
||||
"""Create a plugin action."""
|
||||
super().__init__()
|
||||
def __init__(self, **kwargs):
|
||||
"""Tool action init."""
|
||||
super().__init__(**kwargs)
|
||||
self._render_protocol = VisPlugin()
|
||||
|
||||
@property
|
||||
|
Reference in New Issue
Block a user