Update agent.py

agent.py

@@ -122,7 +122,7 @@ class TeLLAgent:
         else: 
             prompt = str(' ' + outputs["input"] + ' ' + outputs["intermediate_steps"][0][0].log.split('Action:')[0])
             outputs = self.agent_executor2.invoke( {"input": prompt})
-        return outputs['output']
+        return outputs 
     
 if __name__ == '__main__':
         chem_model = TeLLAgent( temp=0.1, streaming=  True,