fix: avoid warning for missing llama installation when not in use (mi…

…crosoft#509) * Update llm_utils to check llama use * modify the logic of the condition
solari998 · Dec 20, 2024 · 5ec3422 · 5ec3422
1 parent d64a228
commit 5ec3422
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rdagent/oai/llm_utils.py b/rdagent/oai/llm_utils.py
@@ -46,7 +46,8 @@ def md5_hash(input_string: str) -> str:
 try:
     from llama import Llama
 except ImportError:
-    logger.warning("llama is not installed.")
+    if LLM_SETTINGS.use_llama2:
+        logger.warning("llama is not installed.")
 
 
 class ConvManager: