Update agent.py

agent.py

@@ -130,10 +130,10 @@ class TeLLAgent:
     def run(self, prompt):
         prompt = prompt + ' ' + str(self.file_path) + ' ' + str(self.image_path)
         outputs = self.agent_executor1.invoke( {"input": prompt})
-        if outputs["intermediate_steps"] ==[]:
+        try:
             prompt = str(' ' +outputs["input"]+ ' ' + outputs["output"].split('Action')[0].split('Final Answer')[0].replace("*", "") )
             outputs = self.agent_executor2.invoke( {"input":prompt })            
-        else: 
+        except: 
             prompt = str(' ' + outputs["input"] + ' ' + outputs["intermediate_steps"][0][0].log.split('Action')[0].replace("*", ""))
             outputs = self.agent_executor2.invoke( {"input": prompt})
         return outputs['output']