mirror of
https://github.com/Mintplex-Labs/anything-llm
synced 2026-04-26 01:25:15 +02:00
File creation agent skills (#5280)
* Powerpoint File Creation (#5278) * wip * download card * UI for downloading * move to fs system with endpoint to pull files * refactor UI * final-pass * remove save-file-browser skill and refactor * remove fileDownload event * reset * reset file * reset timeout * persist toggle * Txt creation (#5279) * wip * download card * UI for downloading * move to fs system with endpoint to pull files * refactor UI * final-pass * remove save-file-browser skill and refactor * remove fileDownload event * reset * reset file * reset timeout * wip * persist toggle * add arbitrary text creation file * Add PDF document generation with markdown formatting (#5283) add support for branding in bottom right corner refactor core utils and frontend rendering * Xlsx document creation (#5284) add Excel doc & sheet creation * Basic docx creation (#5285) * Basic docx creation * add test theme support + styling and title pages * simplify skill selection * handle TG attachments * send documents over tg * lazy import * pin deps * fix lock * i18n for file creation (#5286) i18n for file-creation connect #5280 * theme overhaul * Add PPTX subagent for better results * forgot files * Add PPTX subagent for better results (#5287) * Add PPTX subagent for better results * forgot files * make sub-agent use proper tool calling if it can and better UI hints
This commit is contained in:
Timothy Carambat
@@ -0,0 +1,138 @@
|
||||
const createFilesLib = require("../lib.js");
|
||||
const { applyBranding } = require("./utils.js");
|
||||
|
||||
module.exports.CreatePdfFile = {
|
||||
name: "create-pdf-file",
|
||||
plugin: function () {
|
||||
return {
|
||||
name: "create-pdf-file",
|
||||
setup(aibitat) {
|
||||
aibitat.function({
|
||||
super: aibitat,
|
||||
name: this.name,
|
||||
description:
|
||||
"Create a PDF document from markdown or plain text content. " +
|
||||
"The content will be styled and converted to a professional PDF document. " +
|
||||
"Supports markdown formatting including headers, lists, code blocks, tables, and more.",
|
||||
examples: [
|
||||
{
|
||||
prompt: "Create a PDF report about quarterly sales",
|
||||
call: JSON.stringify({
|
||||
filename: "quarterly-sales-report.pdf",
|
||||
content:
|
||||
"# Quarterly Sales Report\n\n## Q1 2024 Summary\n\n### Key Metrics\n- Total Revenue: $1.2M\n- Growth: 15% YoY\n- New Customers: 234\n\n### Top Products\n1. Product A - $400K\n2. Product B - $350K\n3. Product C - $250K\n\n## Recommendations\n\nBased on the analysis, we recommend focusing on...",
|
||||
}),
|
||||
},
|
||||
{
|
||||
prompt: "Create a PDF document with meeting minutes",
|
||||
call: JSON.stringify({
|
||||
filename: "meeting-minutes.pdf",
|
||||
content:
|
||||
"# Team Meeting Minutes\n\n**Date:** January 15, 2024\n**Attendees:** John, Sarah, Mike, Lisa\n\n## Agenda Items\n\n### 1. Project Status Update\nThe project is on track for Q2 delivery. Key milestones:\n- [ ] Phase 1 complete\n- [x] Phase 2 in progress\n- [ ] Phase 3 pending\n\n### 2. Budget Review\n| Category | Allocated | Spent |\n|----------|-----------|-------|\n| Development | $50,000 | $35,000 |\n| Marketing | $20,000 | $12,000 |\n\n### Action Items\n- John: Complete technical review by Friday\n- Sarah: Schedule stakeholder meeting",
|
||||
}),
|
||||
},
|
||||
{
|
||||
prompt: "Create a PDF with code documentation",
|
||||
call: JSON.stringify({
|
||||
filename: "api-documentation.pdf",
|
||||
content:
|
||||
"# API Documentation\n\n## Authentication\n\nAll API requests require a Bearer token:\n\n```javascript\nfetch('/api/data', {\n headers: {\n 'Authorization': 'Bearer YOUR_TOKEN'\n }\n});\n```\n\n## Endpoints\n\n### GET /api/users\n\nReturns a list of all users.\n\n**Response:**\n```json\n{\n \"users\": [...],\n \"total\": 100\n}\n```",
|
||||
}),
|
||||
},
|
||||
],
|
||||
parameters: {
|
||||
$schema: "http://json-schema.org/draft-07/schema#",
|
||||
type: "object",
|
||||
properties: {
|
||||
filename: {
|
||||
type: "string",
|
||||
description:
|
||||
"The filename for the PDF document. The .pdf extension will be added automatically if not provided.",
|
||||
},
|
||||
content: {
|
||||
type: "string",
|
||||
description:
|
||||
"The markdown or plain text content to convert to PDF. Supports full markdown syntax including headers (#, ##, ###), bold (**text**), italic (*text*), lists, code blocks, tables, and more.",
|
||||
},
|
||||
},
|
||||
required: ["filename", "content"],
|
||||
additionalProperties: false,
|
||||
},
|
||||
handler: async function ({
|
||||
filename = "document.pdf",
|
||||
content = "",
|
||||
}) {
|
||||
try {
|
||||
this.super.handlerProps.log(`Using the create-pdf-file tool.`);
|
||||
|
||||
const hasExtension = /\.pdf$/i.test(filename);
|
||||
if (!hasExtension) filename = `${filename}.pdf`;
|
||||
|
||||
if (this.super.requestToolApproval) {
|
||||
const approval = await this.super.requestToolApproval({
|
||||
skillName: this.name,
|
||||
payload: { filename },
|
||||
description: `Create PDF document "${filename}"`,
|
||||
});
|
||||
if (!approval.approved) {
|
||||
this.super.introspect(
|
||||
`${this.caller}: User rejected the ${this.name} request.`
|
||||
);
|
||||
return approval.message;
|
||||
}
|
||||
}
|
||||
|
||||
this.super.introspect(
|
||||
`${this.caller}: Creating PDF document "${filename}"`
|
||||
);
|
||||
|
||||
const { markdownToPdf } = await import("@mdpdf/mdpdf");
|
||||
const { PDFDocument, rgb, StandardFonts } = await import(
|
||||
"pdf-lib"
|
||||
);
|
||||
|
||||
const rawBuffer = await markdownToPdf(content);
|
||||
const pdfDoc = await PDFDocument.load(rawBuffer);
|
||||
await applyBranding(pdfDoc, { rgb, StandardFonts });
|
||||
|
||||
const buffer = await pdfDoc.save();
|
||||
const bufferSizeKB = (buffer.length / 1024).toFixed(2);
|
||||
const displayFilename = filename.split("/").pop();
|
||||
|
||||
const savedFile = await createFilesLib.saveGeneratedFile({
|
||||
fileType: "pdf",
|
||||
extension: "pdf",
|
||||
buffer,
|
||||
displayFilename,
|
||||
});
|
||||
|
||||
this.super.socket.send("fileDownloadCard", {
|
||||
filename: savedFile.displayFilename,
|
||||
storageFilename: savedFile.filename,
|
||||
fileSize: savedFile.fileSize,
|
||||
});
|
||||
|
||||
createFilesLib.registerOutput(this.super, "PdfFileDownload", {
|
||||
filename: savedFile.displayFilename,
|
||||
storageFilename: savedFile.filename,
|
||||
fileSize: savedFile.fileSize,
|
||||
});
|
||||
|
||||
this.super.introspect(
|
||||
`${this.caller}: Successfully created PDF document "${displayFilename}"`
|
||||
);
|
||||
|
||||
return `Successfully created PDF document "${displayFilename}" (${bufferSizeKB}KB).`;
|
||||
} catch (e) {
|
||||
this.super.handlerProps.log(
|
||||
`create-pdf-file error: ${e.message}`
|
||||
);
|
||||
this.super.introspect(`Error: ${e.message}`);
|
||||
return `Error creating PDF document: ${e.message}`;
|
||||
}
|
||||
},
|
||||
});
|
||||
},
|
||||
};
|
||||
},
|
||||
};
|
||||
@@ -0,0 +1,70 @@
|
||||
const createFilesLib = require("../lib.js");
|
||||
|
||||
/**
|
||||
* Applies AnythingLLM branding to a PDF document.
|
||||
* Adds a logo watermark or fallback text to the bottom-right of each page.
|
||||
* @param {PDFDocument} pdfDoc - The pdf-lib PDFDocument instance
|
||||
* @param {Object} pdfLib - The pdf-lib module exports (rgb, StandardFonts)
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async function applyBranding(pdfDoc, { rgb, StandardFonts }) {
|
||||
const font = await pdfDoc.embedFont(StandardFonts.HelveticaOblique);
|
||||
const pages = pdfDoc.getPages();
|
||||
|
||||
const logoPng = createFilesLib.getLogo({
|
||||
forDarkBackground: false,
|
||||
format: "buffer",
|
||||
});
|
||||
const logoImage = logoPng ? await pdfDoc.embedPng(logoPng) : null;
|
||||
|
||||
const logoWidth = 80;
|
||||
const logoHeight = logoImage
|
||||
? (logoImage.height / logoImage.width) * logoWidth
|
||||
: 0;
|
||||
|
||||
const marginRight = 20;
|
||||
const marginBottom = 20;
|
||||
|
||||
for (const page of pages) {
|
||||
const { width } = page.getSize();
|
||||
|
||||
if (logoImage) {
|
||||
const createdWithText = "created with";
|
||||
const fontSize = 7;
|
||||
const textWidth = font.widthOfTextAtSize(createdWithText, fontSize);
|
||||
const logoX = width - marginRight - logoWidth;
|
||||
|
||||
page.drawText(createdWithText, {
|
||||
x: logoX + (logoWidth - textWidth) / 2,
|
||||
y: marginBottom + logoHeight + 2,
|
||||
size: fontSize,
|
||||
font,
|
||||
color: rgb(0.6, 0.6, 0.6),
|
||||
opacity: 0.6,
|
||||
});
|
||||
|
||||
page.drawImage(logoImage, {
|
||||
x: logoX,
|
||||
y: marginBottom,
|
||||
width: logoWidth,
|
||||
height: logoHeight,
|
||||
opacity: 0.6,
|
||||
});
|
||||
} else {
|
||||
const fallbackText = "Created with AnythingLLM";
|
||||
const fontSize = 9;
|
||||
const textWidth = font.widthOfTextAtSize(fallbackText, fontSize);
|
||||
page.drawText(fallbackText, {
|
||||
x: width - marginRight - textWidth,
|
||||
y: marginBottom,
|
||||
size: fontSize,
|
||||
font,
|
||||
color: rgb(0.6, 0.6, 0.6),
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
applyBranding,
|
||||
};
|
||||
Reference in New Issue
Block a user