Unable to fetch the agent user input into Custom Tool

[ramsagar2215]

I am using two tools and created the agent using the same. In some cases, I have to pass the agent user input into the tool, is there a way to read the agent user prompt into the Tool as mentioned below

def run(self, query, agent_query)

Code starts from here

from langchain.agents.tools import Tool
from langchain.llms import OpenAI
from langchain.chains import LLMChain
from langchain.agents import initialize_agent

class DataRetrieval(object):
     name = "Fetch"
     description = "des"

    def __init__(self):
       pass

    def run(self, query: str) -> str:
        """use the tool."""
        <something>
        return None

sql_query = Tool(
    "read_data",
    DataRetrieval().run,
    """A tool to read/fetch the data"""
)

class DataDump(object):
    name = 'Dump'
    description = ''


    def __init__(self):
       pass

    def run(self, query: str) -> str:
        """use the tool."""
        <something>
        return None

data_dump = Tool(
    "data_dump",
    DataDump().run,
    """A tool to push the data to db"""
)

tools = [sql_query, data_dump]

llm = OpenAI(temperature=0)
agent = initialize_agent(tools,
                         llm,
                         agent="zero-shot-react-description",
                         verbose=True,
                         max_iterations=3)

Thanks in advance.