anything-llm/collector/processSingleFile/convert/asOfficeMime.js
2023-12-20 19:41:16 -08:00

47 lines
1.4 KiB
JavaScript

const { v4 } = require("uuid");
const officeParser = require("officeparser");
const {
createdDate,
trashFile,
writeToServerDocuments,
} = require("../../utils/files");
const { tokenizeString } = require("../../utils/tokenizer");
const { default: slugify } = require("slugify");
async function asOfficeMime({ fullFilePath = "", filename = "" }) {
console.log(`-- Working ${filename} --`);
let content = "";
try {
content = await officeParser.parseOfficeAsync(fullFilePath);
} catch (error) {
console.error(`Could not parse office or office-like file`, error);
}
if (!content.length) {
console.error(`Resulting text content was empty for ${filename}.`);
trashFile(fullFilePath);
return { success: false, reason: `No text content found in ${filename}.` };
}
const data = {
id: v4(),
url: "file://" + fullFilePath,
title: filename,
docAuthor: "no author found",
description: "No description found.",
docSource: "Office file uploaded by the user.",
chunkSource: filename,
published: createdDate(fullFilePath),
wordCount: content.split(" ").length,
pageContent: content,
token_count_estimate: tokenizeString(content).length,
};
writeToServerDocuments(data, `${slugify(filename)}-${data.id}`);
trashFile(fullFilePath);
console.log(`[SUCCESS]: ${filename} converted & ready for embedding.\n`);
return { success: true, reason: null };
}
module.exports = asOfficeMime;