Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

return the image instead

Browse Source
This commit is contained in:
goulustis
2025-11-28 18:17:40 +08:00
parent 232c69d785
commit e831751e3f
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lang_agent/base.py
View File

@@ -25,7 +25,7 @@ class GraphBase(ABC):
def invoke(self, *nargs, **kwargs): def invoke(self, *nargs, **kwargs):
pass pass
def show_graph(self): def show_graph(self, ret_img:bool=False):
#NOTE: just a useful tool for debugging; has zero useful functionality #NOTE: just a useful tool for debugging; has zero useful functionality
err_str = f"{type(self)} does not have workflow, this is unsupported" err_str = f"{type(self)} does not have workflow, this is unsupported"
@@ -33,8 +33,12 @@ class GraphBase(ABC):
logger.info("creating image") logger.info("creating image")
img = Image.open(BytesIO(self.workflow.get_graph().draw_mermaid_png())) img = Image.open(BytesIO(self.workflow.get_graph().draw_mermaid_png()))
plt.imshow(img)
plt.show() if not ret_img:
plt.imshow(img)
plt.show()
else:
return img
class ToolNodeBase(GraphBase): class ToolNodeBase(GraphBase):