feat(Improve garak error handling):

agentic_security/probe_data/modules/garak_tool.py

@@ -1,22 +1,33 @@
-import subprocess
-import os
 import asyncio
+import importlib.util
+import os
+import subprocess
+
 from loguru import logger
-import asyncio
+
+# TODO: add probes modules
 
 
 class Module:
 
     def __init__(self, prompt_groups: [], tools_inbox: asyncio.Queue):
         self.tools_inbox = tools_inbox
+        if not self.is_garak_installed():
+            logger.error(
+                "Garak module is not installed. Please install it using 'pip install garak'"
+            )
+
+    def is_garak_installed(self) -> bool:
+        garak_spec = importlib.util.find_spec("garak")
+        return garak_spec is not None
 
     async def apply(self) -> []:
         env = os.environ.copy()
         env["OPENAI_API_BASE"] = "http://0.0.0.0:8718/proxy"
 
         # Command to be executed
         command = [
-            "python3",
+            "python",
             "-m",
             "garak",
             "--model_type",
@@ -32,7 +43,7 @@ async def apply(self) -> []:
             command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, env=env
         )
         out, err = await asyncio.to_thread(process.communicate)
-
+        yield "Started"
         is_empty = self.tools_inbox.empty()
         logger.info(f"Is inbox empty? {is_empty}")
         while not self.tools_inbox.empty():
@@ -43,3 +54,7 @@ async def apply(self) -> []:
         logger.info("Garak tool finished.")
         logger.info(f"stdout: {out}")
         logger.error(f"exit code: {process.returncode}")
+        if process.returncode != 0:
+            logger.error(f"Error executing command: {command}")
+            logger.error(f"err: {err}")
+            return