From 0a89784116739a30163ff618aedd180bf75a702d Mon Sep 17 00:00:00 2001
From: Haoran Pan <167847254+TPLin22@users.noreply.github.com>
Date: Fri, 20 Dec 2024 17:54:13 +0800
Subject: [PATCH] fix: avoid warning for missing llama installation when not in
 use (#509)

* Update llm_utils to check llama use

* modify the logic of the condition
---
 rdagent/oai/llm_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/rdagent/oai/llm_utils.py b/rdagent/oai/llm_utils.py
index abe9e4ed..e8c914b1 100644
--- a/rdagent/oai/llm_utils.py
+++ b/rdagent/oai/llm_utils.py
@@ -46,7 +46,8 @@ except ImportError:
 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: