how to stop specific sqldatabse tool call for langchain's agent (e.g create_sql_agent)?

[PR0C0S]

Checked other resources

• I added a very descriptive title to this question.
• I searched the LangChain documentation with the integrated search.
• I used the GitHub search to find a similar question and didn't find it.

Commit to Help

• I commit to help with one of those options 👆

Example Code

class QueryCaptureCallback(BaseCallbackHandler):
    def __init__(self):
        self.queries = ""
        self.stop_execution  = False
        

    def on_agent_action(self, action, *, run_id, parent_run_id = None, **kwargs):
        if action.tool == "sql_db_query_checker":
            # print(f"Tool Invoked: {action.tool}, Input: {action.tool_input}")  # Debugging
            self.queries = action.tool_input['query']
            self.stop_execution  = True

callback_handler = QueryCaptureCallback()
        sql_chain= create_sql_agent(
            llm, 
            db=db, 
            agent_type="openai-tools", 
            verbose=False, 
            top_k=5,
            agent_executor_kwargs={"callbacks": [callback_handler]},
        )

Description

im trying to fetch sql queries from create_sql_agent. but the agent runs the whole chain until the query executes. how can i stop execution of queries in this chain?

System Info

"pip freeze"
platform (windows)
python version (3.10)
langchain:
langchain==0.3.9
langchain-community==0.3.9
langchain-openai==0.2.11