first ever agent uploaded to modaic hub
This commit is contained in:
18
__main__.py
Normal file
18
__main__.py
Normal file
@@ -0,0 +1,18 @@
|
||||
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",
|
||||
)
|
||||
Reference in New Issue
Block a user