Alex Newman
40 lines
1.1 KiB
TypeScript
40 lines
1.1 KiB
TypeScript
import { query } from "@anthropic-ai/claude-agent-sdk";
|
|
import * as fs from "fs";
|
|
import * as path from "path";
|
|
|
|
const pathToFolder = "/Users/alexnewman/Scripts/claude-mem/datasets/emails-markdown/";
|
|
const pathToPlugin = "/Users/alexnewman/Scripts/claude-mem/plugin/";
|
|
|
|
// Or read from a directory
|
|
const filesToProcess = fs
|
|
.readdirSync(pathToFolder)
|
|
.filter((f) => f.endsWith(".md"))
|
|
.map((f) => path.join(pathToFolder, f));
|
|
|
|
var i = 0;
|
|
|
|
for (const file of filesToProcess) {
|
|
i++;
|
|
// Limit for testing
|
|
if (i > 3) break;
|
|
|
|
console.log(`\n=== Processing ${file} ===\n`);
|
|
|
|
for await (const message of query({
|
|
prompt: `Read ${file} and think about how it relates to the injected context above (if any).`,
|
|
options: {
|
|
cwd: pathToFolder,
|
|
plugins: [{ type: "local", path: pathToPlugin }],
|
|
},
|
|
})) {
|
|
if (message.type === "system" && message.subtype === "init") {
|
|
console.log("Plugins:", message.plugins);
|
|
console.log("Commands:", message.slash_commands);
|
|
}
|
|
|
|
if (message.type === "assistant") {
|
|
console.log("Assistant:", message.message.content);
|
|
}
|
|
}
|
|
}
|