From be869d1dda9da3a75f7007d19043201080e555ee Mon Sep 17 00:00:00 2001
From: GH05TCREW <harmasic@gmail.com>
Date: Sun, 28 Dec 2025 04:04:12 -0700
Subject: [PATCH] feat: add debug logging for LiteLLM

---
 pentestagent/llm/llm.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pentestagent/llm/llm.py b/pentestagent/llm/llm.py
index 52c0bed..2520f4e 100644
--- a/pentestagent/llm/llm.py
+++ b/pentestagent/llm/llm.py
@@ -1,6 +1,7 @@
 """LiteLLM wrapper for PentestAgent."""
 
 import asyncio
+import os
 import random
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, AsyncIterator, List, Optional
@@ -53,6 +54,14 @@ class LLM:
 
             # Drop unsupported params for models that don't support them
             litellm.drop_params = True
+
+            # Enable debug mode if PENTESTAGENT_DEBUG is set
+            if os.getenv("PENTESTAGENT_DEBUG", "false").lower() in ("true", "1", "yes"):
+                litellm.set_verbose = True
+                import logging
+
+                logging.basicConfig(level=logging.DEBUG)
+
             self._litellm = litellm
         except ImportError as e:
             raise ImportError(