reduce LiteLLM logging in litellm_api script

Author: amanjaiswal73892

src/agentlab/llm/litellm_api.py

@@ -288,10 +288,14 @@ def get_message_builder(self) -> Type[MessageBuilder]:
     Some simple tests to run the LiteLLMModel with different models.
     """
 
+    import os
+
     from agentlab.agents.tool_use_agent import DEFAULT_PROMPT_CONFIG, ToolUseAgentArgs
     from agentlab.experiments.study import Study
     from agentlab.llm.litellm_api import LiteLLMModelArgs
 
+    os.environ["LITELLM_LOG"] = "WARNING"
+
     def get_agent(model_name: str) -> ToolUseAgentArgs:
         return ToolUseAgentArgs(
             model_args=LiteLLMModelArgs(