anything-llm/server/utils/agents/aibitat/plugins/sql-agent/index.js
Timothy Carambat 15cf921616
Support SQL Agent skill (#1411)
* Support SQL Agent skill

* add MSSQL agent connector

* Add frontend to agent skills
remove FAKE_DB mock
reset skills to pickup child-skill dynamically

* add prompt examples for tools on untooled

* add better logging on SQL agents

* Wipe toolruns on each chat relay so tools can be used within the same session

* update comments
2024-05-16 10:38:21 -07:00

22 lines
462 B
JavaScript

const { SqlAgentGetTableSchema } = require("./get-table-schema");
const { SqlAgentListDatabase } = require("./list-database");
const { SqlAgentListTables } = require("./list-table");
const { SqlAgentQuery } = require("./query");
const sqlAgent = {
name: "sql-agent",
startupConfig: {
params: {},
},
plugin: [
SqlAgentListDatabase,
SqlAgentListTables,
SqlAgentGetTableSchema,
SqlAgentQuery,
],
};
module.exports = {
sqlAgent,
};