From b2520cb082ff7e890d8f74e154629e25f03d3acc Mon Sep 17 00:00:00 2001 From: Rodrigo Rodriguez Date: Tue, 13 Aug 2024 21:03:36 -0300 Subject: [PATCH] new(basic.gblib): New batch features. --- packages/gpt.gblib/services/ChatServices.ts | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/packages/gpt.gblib/services/ChatServices.ts b/packages/gpt.gblib/services/ChatServices.ts index 981ff188..72299bd6 100644 --- a/packages/gpt.gblib/services/ChatServices.ts +++ b/packages/gpt.gblib/services/ChatServices.ts @@ -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()),