from agent.agent import TableRAGAgent, TableRagIndexer, TableRAGConfig
from modaic.databases import MilvusVDBConfig, SQLiteConfig

config = TableRAGConfig()
indexer = TableRagIndexer(
    config=config,
    vdb_config=MilvusVDBConfig.from_local("index2.db"),
    sql_config=SQLiteConfig(db_path="tables.db"),
)
agent = TableRAGAgent(config=TableRAGConfig(), indexer=indexer)
# # x = indexer.sql_query("SELECT * FROM t_5th_new_zealand_parliament_0")
# # print(x)
# x = agent(user_query="Who is the New Zealand Parliament Member for Canterbury")
# print(x)
agent.push_to_hub(
    "swagginty/first-ever-daic",
    commit_message="first ever agent uploaded to modaic hub",
)