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)

agent.push_to_hub("swagginty/TableRAG")