Hello @WhiteFang_Jr , @Logan M and

Hello , and everybody, I have one doubt,
I don't want SQLTableRetrieverQueryEngine to give schema of database objects also I don't want user to be able to drop or delete tables. I also want to stop users from access info about other objects which are not in include_tables param. I have tried overriding run_sql function in SQLDatabase class but it does not seem to work. Is there any other way? am I doing it wrong? please help

my code to overide function:
from llama_index import SQLDatabase
from typing import Dict, Tuple

class MySQLDatabase(SQLDatabase): # Assuming SQLDatabase is the original class
def run_sql(self, command: str) -> Tuple[str, Dict]:
"""Override the run_sql method to add custom functionality."""
# Check for DDL, DML, and schema-related commands
ddl_commands = {'CREATE', 'ALTER', 'DROP', 'TRUNCATE', 'COMMENT'}
dml_commands = {'INSERT', 'UPDATE', 'DELETE', 'MERGE'}
schema_commands_mysql = {'information_schema.','sqlite_master'} # Adjust for MySQL

if any(cmd in command.upper() for cmd in ddl_commands.union(dml_commands)):
raise ValueError("I appreciate your input! Let's keep our conversation safe and avoid any database modifications.")

if any(cmd in command.lower() for cmd in schema_commands_mysql):
raise ValueError("I appreciate your input! However, retrieving schema information is not allowed for security reasons.")


return super().run_sql(command)