Suppress litellm debug messages (#448)

monarch-initiative · Aug 28, 2024 · 465fcd1 · 465fcd1
2 parents b7183af + 939dc51
commit 465fcd1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ontogpt/clients/llm_client.py b/src/ontogpt/clients/llm_client.py
@@ -17,6 +17,8 @@
 # Just get the part before the slash in each model name
 SERVICES = {model.split("/")[0] for model in MODELS.keys() if len(model.split("/")) > 1}
 
+# Necessary to avoid repeated debug messages
+litellm.suppress_debug_info = True
 
 @dataclass
 class LLMClient: