Skip to content

chore(mcp-compat): add aggregate command and new save/search options (backend 793f241..99e33da) #27

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ditto-mcp-maintainer wants to merge 1 commit into
base: main
Choose a base branch
from
Open

chore(mcp-compat): add aggregate command and new save/search options (backend 793f241..99e33da) #27

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
90 changes: 81 additions & 9 deletions src/cli.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,28 @@ interface CommonOptions {
interface SaveOptions extends CommonOptions {
source?: string;
sourceContext?: string;
response?: string;
vendorId?: string;
}

interface SearchOptions extends CommonOptions {
includePublic?: boolean;
filterUsername?: string;
limit?: string;
since?: string;
until?: string;
timezone?: string;
filter?: string;
}

interface AggregateOptions extends CommonOptions {
metrics?: string[];
limit?: string;
since?: string;
until?: string;
timezone?: string;
filter?: string;
kg?: string;
}

interface FetchOptions extends CommonOptions {
Expand Down Expand Up @@ -471,25 +488,44 @@ async function cmdLogout(options: CommonOptions): Promise<void> {

async function cmdSave(content: string[], options: SaveOptions): Promise<void> {
const format = parseOutputFormat(options.output);
await callAndPrint(
"save_memory",
{
content: content.join(" "),
source: options.source ?? "cli",
sourceContext: options.sourceContext,
},
format,
);
const args: Record<string, unknown> = {
content: content.join(" "),
source: options.source ?? "cli",
sourceContext: options.sourceContext,
};
if (options.response) args.response = options.response;
if (options.vendorId) args.vendorId = options.vendorId;
await callAndPrint("save_memory", args, format);
}

async function cmdSearch(queries: string[], options: SearchOptions): Promise<void> {
const format = parseOutputFormat(options.output);
const args: Record<string, unknown> = { queries };
if (options.includePublic) args.includePublic = true;
if (options.filterUsername) args.filterUsername = options.filterUsername;
const limit = parseIntegerOption(options.limit, "--limit");
if (limit !== undefined) args.limit = limit;
if (options.since) args.since = options.since;
if (options.until) args.until = options.until;
if (options.timezone) args.timezone = options.timezone;
if (options.filter) args.filter = parseJSONOption(options.filter, "--filter");
await callAndPrint("search_memories", args, format);
}

async function cmdAggregate(groupBy: string, options: AggregateOptions): Promise<void> {
const format = parseOutputFormat(options.output);
const args: Record<string, unknown> = { groupBy };
if (options.metrics && options.metrics.length > 0) args.metrics = options.metrics;
const limit = parseIntegerOption(options.limit, "--limit");
if (limit !== undefined) args.limit = limit;
if (options.since) args.since = options.since;
if (options.until) args.until = options.until;
if (options.timezone) args.timezone = options.timezone;
if (options.filter) args.filter = parseJSONOption(options.filter, "--filter");
if (options.kg) args.kg = options.kg;
await callAndPrint("aggregate_memories", args, format);
}

async function cmdFetch(ids: string[], options: FetchOptions): Promise<void> {
const format = parseOutputFormat(options.output);
const memoryFormat = parseMemoryFormat(options.memoryFormat);
Expand Down Expand Up @@ -826,6 +862,14 @@ Environment:
.argument("<content...>", "memory content")
.option("--source <source>", "memory source", "cli")
.option("--source-context <context>", "source context, such as a filename")
.option(
"--response <text>",
"optional assistant side of the exchange; enables memory-network growth",
)
.option(
"--vendor-id <id>",
"caller-stable id for deduplication (e.g. session::turn-id)",
)
.addOption(outputOption())
.action(cmdSave),
` heyditto save "Project X uses Bun + SolidJS"
Expand All @@ -840,12 +884,40 @@ Environment:
.argument("<query...>", "one or more search queries")
.option("--include-public", "include public DittoHub memories")
.option("--filter-username <username>", "scope public results to a username")
.option("--limit <number>", "maximum memories to return per query (default 5, max 50)")
.option("--since <expression>", 'time lower bound, e.g. "last week", "30d", "2026-03"')
.option("--until <expression>", 'time upper bound, e.g. "2026-04"')
.option("--timezone <iana>", 'IANA timezone, e.g. "America/New_York"')
.option("--filter <json>", "structured filter object as JSON")
.addOption(outputOption())
.action(cmdSearch),
` heyditto search "typescript preferences"
heyditto search "launch notes" --include-public --filter-username peyton`,
);

addExamples(
program
.command("aggregate")
.alias("aggregate_memories")
.description("count and group memories by source, time, subject or theme")
.summary("count and group memories")
.argument("<group-by>", 'dimension to group by: "source", "cluster", "subject", "week", "month", "year"')
.option(
"--metrics <metric>",
"metric to compute (repeatable); e.g. count, sum",
)
.option("--limit <number>", "maximum groups to return (default 50, max 500)")
.option("--since <expression>", 'time lower bound, e.g. "last week", "30d"')
.option("--until <expression>", 'time upper bound')
.option("--timezone <iana>", 'IANA timezone')
.option("--filter <json>", "structured filter object as JSON")
.option("--kg <alias>", "knowledge graph alias")
.addOption(outputOption())
.action(cmdAggregate),
` heyditto aggregate source --limit 20 --output json
heyditto aggregate week --since "last month" --timezone America/New_York`,
);

program
.command("fetch")
.description("fetch memories by id")
Expand Down
19 changes: 19 additions & 0 deletions test/cli-help.test.mjs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,22 @@ test("search --help prints command help without auth", () => {
assert.match(result.stdout, /Usage: heyditto search/);
assert.match(result.stdout, /--include-public/);
assert.match(result.stdout, /--filter-username/);
assert.match(result.stdout, /--limit/);
assert.match(result.stdout, /--since/);
assert.match(result.stdout, /--until/);
assert.match(result.stdout, /--timezone/);
assert.match(result.stdout, /--filter/);
assert.equal(result.stderr, "");
});

test("aggregate --help prints command help without auth", () => {
const result = run(["aggregate", "--help"]);
assert.equal(result.status, 0);
assert.match(result.stdout, /Usage: heyditto aggregate/);
assert.match(result.stdout, /--metrics/);
assert.match(result.stdout, /--limit/);
assert.match(result.stdout, /--since/);
assert.match(result.stdout, /--filter/);
assert.equal(result.stderr, "");
});

Expand All @@ -54,6 +70,8 @@ test("save -h prints source options", () => {
assert.equal(result.status, 0);
assert.match(result.stdout, /--source/);
assert.match(result.stdout, /--source-context/);
assert.match(result.stdout, /--response/);
assert.match(result.stdout, /--vendor-id/);
});

test("update --help prints file and revision options", () => {
Expand Down Expand Up @@ -89,6 +107,7 @@ test("new MCP-backed command help is exposed", () => {
assert.match(result.stdout, /subject-edges\|get_subject_edges/);
assert.match(result.stdout, /graph-sharing\|set_knowledge_graph_sharing/);
assert.match(result.stdout, /delete\|delete_memory/);
assert.match(result.stdout, /aggregate\|aggregate_memories/);
});

test("graph sharing help exposes enable and disable", () => {
Expand Down