13 lines
416 B
Python
13 lines
416 B
Python
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")
|