from typing import Optional

from aios_kernel import Environment
from aios_kernel.sql_database_function import GetTableInfosFunction, ExecuteSqlFunction


class DBQuerierEnvironment(Environment):
    def __init__(self):
        super().__init__("fairy")
        self.add_ai_function(GetTableInfosFunction())
        self.add_ai_function(ExecuteSqlFunction())

    def _do_get_value(self, key: str) -> Optional[str]:
        pass


def init():
    return DBQuerierEnvironment()