new(whatsapp.gblib): LLM SQL.

This commit is contained in:
Rodrigo Rodriguez 2024-08-13 18:45:45 -03:00
parent f8f419f810
commit 9c53ae24bc

View file

@ -468,24 +468,32 @@ export class ChatServices {
question
});
} else if (LLMMode === 'sql') {
const con = min[`llmconnection`];
const dialect = con['storageDriver'];
const host = con['storageServer'];
const port = con['storagePort'];
const storageName = con['storageName'];
const username = con['storageUsername'];
const password = con['storagePassword'];
// const con = min[`llmconnection`];
const dataSource = new DataSource({
type: dialect as any,
host: host,
port: port,
database: storageName,
username: username,
password: password,
synchronize: false,
logging: true,
});
// const dialect = con['storageDriver'];
// const host = con['storageServer'];
// const port = con['storagePort'];
// const storageName = con['storageName'];
// const username = con['storageUsername'];
// const password = con['storagePassword'];
const dataSource = new DataSource({
type: 'sqlite',
database: "/home/gbadmin3910/DATA/BotServer/work/frukigbot1.gbai/data.db",
});
// const dataSource = new DataSource({
// type: dialect as any,
// host: host,
// port: port,
// database: storageName,
// username: username,
// password: password,
// synchronize: false,
// logging: true,
// });
const db = await SqlDatabase.fromDataSourceParams({
appDataSource: dataSource
@ -493,7 +501,7 @@ export class ChatServices {
const chain = new SqlDatabaseChain({
llm: model,
database: db
database: db,
});
result = await chain.run(question);