opendan/rootfs/agents/DBQueryer/environment.py

from typing import Optional

from aios.environment.environment import Environment
from aios.ai_functions.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()