new(basic.gblib): New batch features.

This commit is contained in:
Rodrigo Rodriguez 2024-08-13 21:03:36 -03:00
parent b87f19e9d3
commit b2520cb082

View file

@ -521,7 +521,8 @@ export class ChatServices {
{
schema: async () => db.getTableInfo(),
question: (input: { question: string }) => input.question ,
top_k: ()=>10
top_k: ()=>10,
table_info: ()=>'any'
},
prompt,
model,
@ -568,7 +569,7 @@ Pay attention to use only the column names you can see in the tables below. Be c
question: input => input.question,
query: input => input.query,
response: input => db.run(input.query),
top_k: ()=>10
top_k: ()=>10, table_info: ()=>'any'
},
{
result: finalResponsePrompt.pipe(model).pipe(new StringOutputParser()),