19 lines
477 B
Python
19 lines
477 B
Python
|
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()
|