diff --git a/rag/llm/cv_model.py b/rag/llm/cv_model.py
index 0f1aa386d..5f487576e 100644
--- a/rag/llm/cv_model.py
+++ b/rag/llm/cv_model.py
@@ -59,6 +59,10 @@ class Base(ABC):
     def _image_prompt(self, text, images):
         if not images:
             return text
+
+        if isinstance(images, str):
+            images = [images]
+
         pmpt = [{"type": "text", "text": text}]
         for img in images:
             pmpt.append({
@@ -795,4 +799,4 @@ class GoogleCV(AnthropicCV, GeminiCV):
                 yield ans
         else:
             for ans in GeminiCV.chat_streamly(self, system, history, gen_conf, images):
-                yield ans
\ No newline at end of file
+                yield ans