feat: add debug logging for LiteLLM

This commit is contained in:
GH05TCREW
2025-12-28 04:04:12 -07:00
parent 40902bea49
commit be869d1dda

View File

@@ -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(