From 7b0bfe95040c0f7ddcf8885d688ceb676f44b9e5 Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Sat, 11 Jan 2025 16:57:59 +0100 Subject: [PATCH] Update cognee-mcp/cognee_mcp/server.py Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> --- cognee-mcp/cognee_mcp/server.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/cognee-mcp/cognee_mcp/server.py b/cognee-mcp/cognee_mcp/server.py index 07d5babe0..2f7efd787 100644 --- a/cognee-mcp/cognee_mcp/server.py +++ b/cognee-mcp/cognee_mcp/server.py @@ -196,10 +196,12 @@ async def handle_call_tool( elif name == "visualize": with open(os.devnull, "w") as fnull: with redirect_stdout(fnull), redirect_stderr(fnull): - """Create a thumbnail from an image""" - await cognee.visualize - img = get_freshest_png(".") - return types.Image(data=img.tobytes(), format="png") + try: + await cognee.visualize + img = get_freshest_png(".") + return types.Image(data=img.tobytes(), format="png") + except (FileNotFoundError, IOError, ValueError) as e: + raise ValueError(f"Failed to create visualization: {str(e)}") else: raise ValueError(f"Unknown tool: {name}")