Ignore warnings from litellm

plasma-umass · Apr 8, 2024 · 41d159a · 41d159a
1 parent 1212647
commit 41d159a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/src/cwhy/conversation/__init__.py b/src/cwhy/conversation/__init__.py
@@ -1,7 +1,11 @@
 import json
 import textwrap
+import warnings
+
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    import litellm  # type: ignore
 
-import litellm  # type: ignore
 import llm_utils
 
 from . import utils

diff --git a/src/cwhy/cwhy.py b/src/cwhy/cwhy.py
@@ -2,9 +2,12 @@
 import os
 import subprocess
 import sys
-import logging
+import warnings
+
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    import litellm  # type: ignore
 
-import litellm  # type: ignore
 import llm_utils
 from openai import (
     NotFoundError,
@@ -18,11 +21,6 @@
 from .print_debug import dprint, enable_debug_printing
 
 
-# Turn off most logging
-litellm.set_verbose = False
-logging.getLogger().setLevel(logging.ERROR)
-
-
 def print_key_info():
     dprint("You need a key (or keys) from an AI service to use CWhy.")
     dprint()