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>
29 lines
756 B
Python
29 lines
756 B
Python
import json
|
|
from abc import ABC, abstractmethod
|
|
from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Type, Union
|
|
from dbgpt.util.json_utils import serialize
|
|
|
|
|
|
class Vis:
|
|
@abstractmethod
|
|
async def generate_param(self, **kwargs) -> Optional[str]:
|
|
"""
|
|
Display corresponding content using vis protocol
|
|
Args:
|
|
**kwargs:
|
|
|
|
Returns:
|
|
vis protocol text
|
|
"""
|
|
|
|
async def disply(self, **kwargs) -> Optional[str]:
|
|
return f"```{self.vis_tag()}\n{json.dumps(await self.generate_param(**kwargs), default=serialize, ensure_ascii=False)}\n```"
|
|
|
|
@classmethod
|
|
def vis_tag(cls) -> str:
|
|
"""
|
|
Current vis protocol module tag name
|
|
Returns:
|
|
|
|
"""
|