utils/react.py

from langchain import hub
from langchain.agents import AgentExecutor, create_react_agent

def run_agent_executor(agent_executor: AgentExecutor, input_data: dict):
        for chunk in agent_executor.stream(input_data):
            if "actions" in chunk:
                for action in chunk["actions"]:
                    print(f"Calling Tool: `{action.tool}` with input `{action.tool_input}`")
            # Observation
            elif "steps" in chunk:
                for step in chunk["steps"]:
                    print(f"Tool Result: `{step.observation}`")
            # Final result
            elif "output" in chunk:
                print(f'Final Output: {chunk["output"]}')
            else:
                raise ValueError()
            print("---")