Add database query

This commit is contained in:
wugren
2023-11-28 17:07:39 +08:00
parent 22453ca85c
commit e00fc6c8e4
5 changed files with 640 additions and 0 deletions
+18
View File
@@ -0,0 +1,18 @@
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()