new(core.gbapp): New API endpoint.

This commit is contained in:
Rodrigo Rodriguez 2024-04-17 15:36:08 -03:00
parent df41f1c55b
commit 43feb52788

View file

@ -138,13 +138,13 @@ export class GBLLMOutputParser extends
} }
let { sources, text } = res; let { sources, text } = res;
await CollectionUtil.asyncForEach(sources, async (source) => { await CollectionUtil.asyncForEach(sources, async (source) => {
let found = false; let found = false;
if (source) { if (source) {
const gbaiName = DialogKeywords.getGBAIPath(this.min.botId, 'gbkb'); const gbaiName = DialogKeywords.getGBAIPath(this.min.botId, 'gbkb');
const localName = Path.join(process.env.PWD,'work', gbaiName, 'docs', source.file); const localName = Path.join(process.env.PWD, 'work', gbaiName, 'docs', source.file);
if (localName) { if (localName) {
const { url } = await ChatServices.pdfPageAsImage(this.min, localName, source.page); const { url } = await ChatServices.pdfPageAsImage(this.min, localName, source.page);
@ -214,7 +214,7 @@ export class ChatServices {
let output = ''; let output = '';
for(const filePaths of Object.keys(uniqueDocuments)) { for (const filePaths of Object.keys(uniqueDocuments)) {
const doc = uniqueDocuments[filePaths]; const doc = uniqueDocuments[filePaths];
const metadata = doc.metadata; const metadata = doc.metadata;
const filename = Path.basename(metadata.source); const filename = Path.basename(metadata.source);
@ -492,12 +492,14 @@ export class ChatServices {
}); });
const tool = new WikipediaQueryRun({ if (process.env.WIKIPEDIA_TOOL) {
topKResults: 3,
maxDocContentLength: 4000,
});
functions.push(tool);
const tool = new WikipediaQueryRun({
topKResults: 3,
maxDocContentLength: 4000,
});
functions.push(tool);
}
return functions; return functions;
} }
} }