Update helper/keywords.py

helper/keywords.py

@@ -159,4 +159,18 @@ def is_structured_task(prompt: str) -> bool:
     return False
 
 def multiple_questions(prompt: str) -> bool:
-    return prompt.count("?") >= 3
+    return prompt.count("?") >= 3
+
+def contains_images(messages):
+    for msg in messages:
+        content = msg.get("content")
+
+        if isinstance(content, list):
+            for part in content:
+                if isinstance(part, dict) and part.get("type") in ["image_urls", "image_url"]:
+                    return True
+
+        if "image" in msg or "image_url" in msg:
+            return True
+
+    return False