Skip to content

base_sql_driver

BaseSqlDriver

Bases: ABC

Source code in griptape/drivers/sql/base_sql_driver.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
@define
class BaseSqlDriver(ABC):
    @dataclass
    class RowResult:
        cells: dict[str, Any]

    @abstractmethod
    def execute_query(self, query: str) -> Optional[list[RowResult]]: ...

    @abstractmethod
    def execute_query_raw(self, query: str) -> Optional[list[dict[str, Any]]]: ...

    @abstractmethod
    def get_table_schema(self, table_name: str, schema: Optional[str] = None) -> Optional[str]: ...

RowResult dataclass

Source code in griptape/drivers/sql/base_sql_driver.py 
12
13
14
@dataclass
class RowResult:
    cells: dict[str, Any]
cells: dict[str, Any] instance-attribute
__init__(cells)

execute_query(query) abstractmethod

Source code in griptape/drivers/sql/base_sql_driver.py 
16
17
@abstractmethod
def execute_query(self, query: str) -> Optional[list[RowResult]]: ...

execute_query_raw(query) abstractmethod

Source code in griptape/drivers/sql/base_sql_driver.py 
19
20
@abstractmethod
def execute_query_raw(self, query: str) -> Optional[list[dict[str, Any]]]: ...

get_table_schema(table_name, schema=None) abstractmethod

Source code in griptape/drivers/sql/base_sql_driver.py 
22
23
@abstractmethod
def get_table_schema(self, table_name: str, schema: Optional[str] = None) -> Optional[str]: ...