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

fix(llm.gblib): Talk to data local db use fix.

Browse source
This commit is contained in:
Rodrigo Rodriguez 2024-11-29 21:49:35 -03:00
parent 420543768b
commit afc72c5d55
1 changed files with 22 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
packages/kb.gbapp/services/KBService.ts
View file

@ -47,8 +47,7 @@ import { CSVLoader } from '@langchain/community/document_loaders/fs/csv';
import { DocxLoader } from '@langchain/community/document_loaders/fs/docx'; import { DocxLoader } from '@langchain/community/document_loaders/fs/docx';
import { EPubLoader } from '@langchain/community/document_loaders/fs/epub'; import { EPubLoader } from '@langchain/community/document_loaders/fs/epub';
import { PDFLoader } from '@langchain/community/document_loaders/fs/pdf'; import { PDFLoader } from '@langchain/community/document_loaders/fs/pdf';
import { rimraf } from 'rimraf'; import svg2img from 'svg2img';
import getColors from 'get-image-colors'; import getColors from 'get-image-colors';
import { Document } from 'langchain/document'; import { Document } from 'langchain/document';
import { RecursiveCharacterTextSplitter } from 'langchain/text_splitter'; import { RecursiveCharacterTextSplitter } from 'langchain/text_splitter';
@ -1067,14 +1066,24 @@ export class KBService implements IGBKBService {
const logoBinary = await page.goto(logo); const logoBinary = await page.goto(logo);
let buffer = await logoBinary.buffer(); let buffer = await logoBinary.buffer();
const logoFilename = path.basename(logo); let logoFilename = path.basename(logo);
// Replace sharp with jimp // Replace sharp with jimp
if (buffer.toString().includes('<svg')) { if (buffer.toString().includes('<svg')) {
logoFilename = logoFilename.replace('.svg', '.png');
// For SVG files, convert using svg2img // For SVG files, convert using svg2img
const svg2img = require('svg2img');
buffer = await new Promise((resolve, reject) => { buffer = await new Promise((resolve, reject) => {
svg2img(buffer, {width: 48, height: 48}, (error: any, buffer: Buffer) => { svg2img(buffer, {
resvg: {
fitTo: {
mode: 'width', // or height
value: 48,
},
}
}, (error: any, buffer: Buffer) => {
if (error) { if (error) {
reject(error); reject(error);
} else { } else {