@plastichub/osr-ai-tools
Version:
CLI and library for LLM tools
107 lines (106 loc) • 4.42 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.tools = exports.fileToBase64 = exports.mime = void 0;
const path_1 = require("path");
const write_1 = require("@plastichub/fs/write");
const fs = __importStar(require("fs"));
const mime_types_1 = require("mime-types");
const screenshot = require('screenshot-desktop');
const mime = (file = '') => (0, path_1.parse)(file).ext ? (0, mime_types_1.lookup)(file) : null;
exports.mime = mime;
const fileToBase64 = (filePath) => {
try {
const fileBuffer = fs.readFileSync(filePath);
const mimeType = (0, mime_types_1.lookup)(filePath);
if (!mimeType) {
throw new Error('Unable to determine MIME type.');
}
const base64Data = fileBuffer.toString('base64');
return `data:${mimeType};base64,${base64Data}`;
}
catch (error) {
console.error('fileToBase64 : Error reading file:', error);
return null;
}
};
exports.fileToBase64 = fileToBase64;
const __1 = require("../..");
const tools = (target, options) => {
const logger = (0, __1.toolLogger)((0, path_1.parse)(__filename).name, options);
return [
{
type: 'function',
function: {
name: 'capture_screen',
description: 'Capture a screenshot and store it as file (jpg). Returns the path to the file',
parameters: {
type: 'object',
properties: {
file: { type: 'string' }
},
required: ['file']
},
function: async (params) => {
try {
const outputPath = (0, path_1.join)(target, params.file);
const takeScreenshot = async () => {
logger.debug(`Capturing screenshot to ${outputPath}`);
return new Promise((resolve, reject) => {
screenshot({ format: 'jpg' }).then((img) => {
(0, write_1.sync)(outputPath, img);
resolve({ success: true, path: outputPath });
}).catch(reject);
});
};
const { path } = await takeScreenshot();
return {
"role": "user",
"content": [
/*
{
type: "image_url",
image_url: {
url: fileToBase64( path),
}
}
*/
]
};
}
catch (error) {
logger.error('Error capturing screenshot:', error);
return {
success: false,
error: error.message
};
}
},
parse: JSON.parse
}
}
];
};
exports.tools = tools;