agents-js-oasst / src /lib /agents /evalBuilder.ts
nsarrazin's picture
add file upload support
65d7bbe
raw
history blame
994 Bytes
import type { Tool } from "$lib/types";
// this function passes the tools & files to the context before calling eval
export async function evalBuilder(
code: string,
tools: Tool<any, any>[],
files: FileList | null,
updateCallback: (message: string, data: undefined | string | Blob) => void
) {
async function wrapperEval() {
if (files && files.length > 0) {
// @ts-ignore
globalThis["file"] = await files[0];
}
// add tools to context
for (const tool of tools) {
// @ts-ignore
globalThis[tool.name] = tool.call;
}
// @ts-ignore
globalThis["message"] = updateCallback;
const returnString = "\nreturn await generate(file);";
await Object.getPrototypeOf(async function () {}).constructor(
code + returnString
)();
// clean up tools
for (const tool of tools) {
// @ts-ignore
delete globalThis[tool.name];
// @ts-ignore
delete globalThis["file"];
}
}
return wrapperEval;
}