GeneralBots/botserver
2
1
Fork 0
Code Issues Pull requests 1 Projects Releases 323 Packages Wiki Activity Actions

new(basic.gblib): New batch features.

Browse source
This commit is contained in:
Rodrigo Rodriguez 2024-08-13 21:03:36 -03:00
parent b87f19e9d3
commit b2520cb082
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
packages/gpt.gblib/services/ChatServices.ts
View file

@ -521,7 +521,8 @@ export class ChatServices {
{ {
schema: async () => db.getTableInfo(), schema: async () => db.getTableInfo(),
question: (input: { question: string }) => input.question , question: (input: { question: string }) => input.question ,
top_k: ()=>10 top_k: ()=>10,
table_info: ()=>'any'
}, },
prompt, prompt,
model, 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, question: input => input.question,
query: input => input.query, query: input => input.query,
response: input => db.run(input.query), response: input => db.run(input.query),
top_k: ()=>10 top_k: ()=>10, table_info: ()=>'any'
}, },
{ {
result: finalResponsePrompt.pipe(model).pipe(new StringOutputParser()), result: finalResponsePrompt.pipe(model).pipe(new StringOutputParser()),