From 81a60405273db865d7d1c5e3060bfc9878a50a5d Mon Sep 17 00:00:00 2001
From: harshalmore31 <harshalmore2468@gmail.com>
Date: Tue, 8 Jul 2025 21:37:31 +0530
Subject: [PATCH] fixes and clean up !

---
 swarms/utils/litellm_wrapper.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/swarms/utils/litellm_wrapper.py b/swarms/utils/litellm_wrapper.py
index d9832edb..3692cb65 100644
--- a/swarms/utils/litellm_wrapper.py
+++ b/swarms/utils/litellm_wrapper.py
@@ -429,9 +429,7 @@ class LiteLLM:
 
             if out is False:
                 raise ValueError(
-                    f"Model {self.model_name} does not support vision. "
-                    f"Use a vision-capable model like gpt-4-vision-preview, "
-                    f"claude-3-sonnet, or gemini-pro-vision."
+                    f"Model {self.model_name} does not support vision"
                 )
 
     def run(
@@ -570,7 +568,7 @@ class LiteLLM:
         try:
             messages = self._prepare_messages(task=task, img=img)
 
-            # Base completion parameters
+            # Prepare common completion parameters
             completion_params = {
                 "model": self.model_name,
                 "messages": messages,