Update agent.py

agent.py

@@ -117,10 +117,10 @@ class TeLLAgent:
         prompt = prompt + ' ' + str(self.file_path) + ' ' + str(self.image_path)
         outputs = self.agent_executor1.invoke( {"input": prompt})
         if outputs["intermediate_steps"] ==[]:
-            prompt = str(' ' +outputs["input"]+ ' ' + outputs["output"].split('Action:')[0] )
+            prompt = str(' ' +outputs["input"]+ ' ' + outputs["output"].split('Action')[0].rstrip('*') )
             outputs = self.agent_executor2.invoke( {"input":prompt })            
         else: 
-            prompt = str(' ' + outputs["input"] + ' ' + outputs["intermediate_steps"][0][0].log.split('Action:')[0])
+            prompt = str(' ' + outputs["input"] + ' ' + outputs["intermediate_steps"][0][0].log.split('Action')[0].rstrip('*'))
             outputs = self.agent_executor2.invoke( {"input": prompt})
         return outputs