fix: fixed screenshot() not running if generator not consumed (such a…

…s when not last statement in codeblock)

gptme/tools/screenshot.py

@@ -4,11 +4,9 @@
 
 import os
 import subprocess
-from collections.abc import Generator
 from datetime import datetime
 from pathlib import Path
 
-from ..message import Message
 from .base import ToolSpec
 
 
@@ -28,21 +26,15 @@ def _screenshot(path: Path) -> Path:
     return path
 
 
-def screenshot(path: Path | None = None) -> Generator[Message, None, None]:
+def screenshot(path: Path | None = None) -> Path:
     """
     Take a screenshot and save it to a file.
     """
     if path is None:
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         path = Path(f"screenshot_{timestamp}.png")
 
-    try:
-        path = _screenshot(path)
-        yield Message("system", f"Screenshot saved to: {path}")
-    except NotImplementedError as e:
-        yield Message("system", str(e))
-    except subprocess.CalledProcessError:
-        yield Message("system", "Failed to capture screenshot.")
+    return _screenshot(path)
 
 
 tool = ToolSpec(