github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-26 05:23:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(vis): Fix think message error

Browse Source
This commit is contained in:
Fangyin Cheng 2025-06-27 17:55:36 +08:00
parent 000cf5a97b
commit 69fa83cb18
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
packages/dbgpt-core/src/dbgpt/vis/tags/vis_thinking.py
View File

@ -7,12 +7,12 @@ class VisThinking(Vis):
@classmethod @classmethod
async def build_message(cls, message: str) -> str: async def build_message(cls, message: str) -> str:
vis = VisThinking() vis = VisThinking()
return f"```{vis.vis_tag()}\n{message}\n```" return f"``````{vis.vis_tag()}\n{message}\n``````"
def sync_display(self, **kwargs) -> str: def sync_display(self, **kwargs) -> str:
"""Display the content using the vis protocol.""" """Display the content using the vis protocol."""
content = kwargs.get("content") content = kwargs.get("content")
return f"```{self.vis_tag()}\n{content}\n```" return f"``````{self.vis_tag()}\n{content}\n``````"
@classmethod @classmethod
def vis_tag(cls): def vis_tag(cls):