mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-22 01:22:34 +00:00
e11b72c724
Co-authored-by: qidanrui <qidanrui@gmail.com> Co-authored-by: csunny <cfqsunny@163.com> Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
35 lines
1016 B
Python
35 lines
1016 B
Python
from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Type, Union
|
||
from .tags.vis_code import VisCode
|
||
from .tags.vis_chart import VisChart
|
||
from .tags.vis_dashboard import VisDashboard
|
||
from .tags.vis_agent_plans import VisAgentPlans
|
||
from .tags.vis_agent_message import VisAgentMessages
|
||
from .tags.vis_plugin import VisPlugin
|
||
from .base import Vis
|
||
|
||
|
||
class VisClient:
|
||
def __init__(self):
|
||
self._vis_tag: Dict[str, Vis] = {}
|
||
|
||
def register(self, vis_cls: Vis):
|
||
self._vis_tag[vis_cls.vis_tag()] = vis_cls()
|
||
|
||
def get(self, tag_name):
|
||
if tag_name not in self._vis_tag:
|
||
raise ValueError(f"Vis protocol tags not yet supported！[{tag_name}]")
|
||
return self._vis_tag[tag_name]
|
||
|
||
def tag_names(self):
|
||
self._vis_tag.keys()
|
||
|
||
|
||
vis_client = VisClient()
|
||
|
||
vis_client.register(VisCode)
|
||
vis_client.register(VisChart)
|
||
vis_client.register(VisDashboard)
|
||
vis_client.register(VisAgentPlans)
|
||
vis_client.register(VisAgentMessages)
|
||
vis_client.register(VisPlugin)
|