Set LITELLM_LOG as error level

Giskard-AI · Nov 21, 2024 · 82712c7 · 82712c7
1 parent 78dd03e
commit 82712c7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/giskard/llm/client/litellm.py b/giskard/llm/client/litellm.py
@@ -1,5 +1,6 @@
 from typing import Any, Dict, Optional, Sequence
 
+import os
 import json
 import logging
 import re
@@ -12,6 +13,7 @@
 logger = logging.getLogger(__name__)
 
 try:
+    os.environ["LITELLM_LOG"] = "ERROR"
     import litellm
 except ImportError as err:
     raise LLMImportError(flavor="litellm") from err