INNER CODE UNIT · Python
that
abhijithneilabraham/tableQA · tableqa/agent.py:55
Function that returns the response for the user question.
"""
def __init__(self, data_dir, schema_dir=None, db_type='sqlite', username='', password='', database='db', host='localhost', port=None, drop_db=True, aws_db=False, aws_s3=False, access_key_id="", secret_access_key=""):
"""
Constructs all the necessary attributes for the agent object.
Arguments
----------
data_dir: `str` or `pathlib.Path` object
absolute path to folder containing all input files.
schema_dir: `str` or `pathlib.Path` object
path to folder containing `json` schemas of input files.
If not specified, auto-generated schema will be used.
db_type: `str`
defining the type of Database object to be used.
If not specified, default object is `sqlite`.