import subprocess database = "/tmp/test1.ddb" class Tools: def __init__(self): pass def execute_sql_query(self, query: str) -> str: """ Executes the given SQL query in the database and returns the output in CSV format. :param query: SQL query to execute. :return: Query result in CSV format or error message. """ try: result = subprocess.run( ["duckdb", database, "-safe", "-readonly", "-csv", "-c", query], capture_output=True, text=True, check=True, ) return f"SQL query:\n\n{query}\n\nResult of the SQL query in CSV format:\n\n{result.stdout}\n" except subprocess.CalledProcessError as e: return f"SQL Execution failed: {e.stderr}" def get_database_schema(self) -> str: """ Returns the structure of all tables in the database. :return: Database structure or error message. """ try: result = subprocess.run( [ "duckdb", database, "-safe", "-readonly", "-list", "-noheader", "-c", "select sql from duckdb_tables;", ], capture_output=True, text=True, check=True, ) return f"The structure of the tables in the database:\n\n{result.stdout}" except subprocess.CalledProcessError as e: return f"SQL Execution failed: {e.stderr}"