From c60789eb54067eea70cde8d67ed33782df7cb090 Mon Sep 17 00:00:00 2001
From: Sambhav Dixit <94298612+sambhavnoobcoder@users.noreply.github.com>
Date: Fri, 25 Oct 2024 02:49:42 +0530
Subject: [PATCH] response_args should include image

- other functionality remains same , code is just slightly rearranged if anything .
---
 swarms/structs/agent.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py
index 5bb9e9b0..29d7037b 100644
--- a/swarms/structs/agent.py
+++ b/swarms/structs/agent.py
@@ -803,7 +803,14 @@ class Agent:
                             self.memory_query(task_prompt)
 
                         # Generate response using LLM
-                        response = self.llm(task_prompt, *args, **kwargs)
+                        response_args = (
+                                (task_prompt, *args)
+                                if img is None
+                                else (task_prompt, img, *args)
+                            )
+                        response = self.call_llm(
+                            *response_args, **kwargs
+                        )
 
                         # Check and execute tools
                         if self.tools is not None: