@fingerskier/augment 0.1.0

package/dist/mcp/server.js

@@ -0,0 +1,91 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import * as z from "zod/v4";
+import { LINK_TYPES, MEMORY_KINDS } from "../constants.js";
+import { connectOrStartDaemon } from "../daemon/client.js";
+import { inferProjectName } from "../memory/project.js";
+export async function startMcpServer() {
+    const server = new McpServer({
+        name: "augment",
+        version: "0.1.0",
+    });
+    server.registerTool("init_project", {
+        description: "Infer or upsert the current project and return its local project record.",
+        inputSchema: {
+            project_name: z.string().optional(),
+        },
+    }, async ({ project_name }) => {
+        const client = await connectOrStartDaemon();
+        const resolved = project_name ?? (await inferProjectName());
+        return textResult(await client.initProject({ project_name: resolved }));
+    });
+    server.registerTool("list_projects", {
+        description: "List known Augment projects.",
+        inputSchema: {
+            name: z.string().optional(),
+        },
+    }, async ({ name }) => textResult(await (await connectOrStartDaemon()).listProjects(name)));
+    server.registerTool("search", {
+        description: "Search memories and return text suitable for direct context injection.",
+        inputSchema: {
+            project_name: z.string().optional(),
+            query: z.string(),
+            limit: z.number().int().positive().max(20).optional(),
+            max_chars: z.number().int().positive().max(50_000).optional(),
+            kinds: z.array(z.enum(MEMORY_KINDS)).optional(),
+            include_linked: z.boolean().optional(),
+        },
+    }, async (input) => {
+        const client = await connectOrStartDaemon();
+        const project_name = input.project_name ?? (await inferProjectName());
+        const output = (await client.search({ ...input, project_name }));
+        return {
+            content: [{ type: "text", text: output.text ?? JSON.stringify(output, null, 2) }],
+        };
+    });
+    server.registerTool("upsert", {
+        description: "Create or update a memory file and immediately reindex it.",
+        inputSchema: {
+            project_id: z.number().int().positive(),
+            memory_id: z.number().int().positive().optional(),
+            kind: z.enum(MEMORY_KINDS).optional(),
+            name: z.string(),
+            content: z.string(),
+            tags: z.array(z.string()).optional(),
+        },
+    }, async (input) => textResult(await (await connectOrStartDaemon()).upsert(input)));
+    server.registerTool("read", {
+        description: "Read a full memory record and body by local memory ID.",
+        inputSchema: {
+            id: z.number().int().positive(),
+        },
+    }, async ({ id }) => textResult(await (await connectOrStartDaemon()).read(id)));
+    server.registerTool("link", {
+        description: "Create a directional link between two local memories.",
+        inputSchema: {
+            from_memory_id: z.number().int().positive(),
+            to_memory_id: z.number().int().positive(),
+            type: z.enum(LINK_TYPES).optional(),
+        },
+    }, async (input) => textResult(await (await connectOrStartDaemon()).link(input)));
+    server.registerTool("list_links", {
+        description: "List inbound and outbound links involving a memory.",
+        inputSchema: {
+            memory_id: z.number().int().positive(),
+        },
+    }, async ({ memory_id }) => textResult(await (await connectOrStartDaemon()).listLinks(memory_id)));
+    server.registerTool("unlink", {
+        description: "Remove a link by local link ID.",
+        inputSchema: {
+            id: z.number().int().positive(),
+        },
+    }, async ({ id }) => textResult(await (await connectOrStartDaemon()).unlink(id)));
+    await server.connect(new StdioServerTransport());
+}
+function textResult(value) {
+    const text = typeof value === "object" && value !== null && "text" in value ? String(value.text) : JSON.stringify(value, null, 2);
+    return {
+        content: [{ type: "text", text }],
+    };
+}
+//# sourceMappingURL=server.js.map

package/dist/mcp/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,KAAK,CAAC,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;QAC3B,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,OAAO;KACjB,CAAC,CAAC;IAEH,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;QACE,WAAW,EAAE,0EAA0E;QACvF,WAAW,EAAE;YACX,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SACpC;KACF,EACD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;QACzB,MAAM,MAAM,GAAG,MAAM,oBAAoB,EAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,YAAY,IAAI,CAAC,MAAM,gBAAgB,EAAE,CAAC,CAAC;QAC5D,OAAO,UAAU,CAAC,MAAM,MAAM,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IAC1E,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,eAAe,EACf;QACE,WAAW,EAAE,8BAA8B;QAC3C,WAAW,EAAE;YACX,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAC5B;KACF,EACD,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,oBAAoB,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CACxF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,QAAQ,EACR;QACE,WAAW,EAAE,wEAAwE;QACrF,WAAW,EAAE;YACX,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;YACnC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;YACjB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE;YACrD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;YAC7D,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,EAAE;YAC/C,cAAc,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;SACvC;KACF,EACD,KAAK,EAAE,KAAK,EAAE,EAAE;QACd,MAAM,MAAM,GAAG,MAAM,oBAAoB,EAAE,CAAC;QAC5C,MAAM,YAAY,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,MAAM,gBAAgB,EAAE,CAAC,CAAC;QACtE,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,EAAE,YAAY,EAAE,CAAC,CAAsB,CAAC;QACtF,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;SAClF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,QAAQ,EACR;QACE,WAAW,EAAE,4DAA4D;QACzE,WAAW,EAAE;YACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACvC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;YACjD,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE;YACrC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;YACnB,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;SACrC;KACF,EACD,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,oBAAoB,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAChF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,MAAM,EACN;QACE,WAAW,EAAE,wDAAwD;QACrE,WAAW,EAAE;YACX,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;SAChC;KACF,EACD,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAC5E,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,MAAM,EACN;QACE,WAAW,EAAE,uDAAuD;QACpE,WAAW,EAAE;YACX,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YAC3C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACzC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE;SACpC;KACF,EACD,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAC9E,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,YAAY,EACZ;QACE,WAAW,EAAE,qDAAqD;QAClE,WAAW,EAAE;YACX,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;SACvC;KACF,EACD,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,oBAAoB,EAAE,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAC/F,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,QAAQ,EACR;QACE,WAAW,EAAE,iCAAiC;QAC9C,WAAW,EAAE;YACX,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;SAChC;KACF,EACD,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,oBAAoB,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAC9E,CAAC;IAEF,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,oBAAoB,EAAE,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,UAAU,CAAC,KAAc;IAChC,MAAM,IAAI,GAAG,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,MAAM,CAAE,KAA2B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACzJ,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC;KAC3C,CAAC;AACJ,CAAC"}

package/dist/memory/files.d.ts

@@ -0,0 +1,11 @@
+import { type LinkType, type MemoryKind } from "../constants.js";
+import type { MemoryFrontmatter, ParsedMemory } from "../types.js";
+export declare function normalizeKind(kind: string | undefined): MemoryKind;
+export declare function normalizeLinkType(type: string | undefined): LinkType;
+export declare function memoryRelativePath(projectName: string, kind: MemoryKind, name: string): string;
+export declare function relativePathToMetadata(relativePath: string): Pick<MemoryFrontmatter, "project" | "kind" | "name">;
+export declare function readMemory(memoryRoot: string, relativePath: string): Promise<ParsedMemory>;
+export declare function writeMemory(memoryRoot: string, relativePath: string, frontmatter: MemoryFrontmatter, body: string): Promise<void>;
+export declare function normalizeRelativePath(value: string): string;
+export declare function normalizeFrontmatter(value: Record<string, unknown>, fallback: Pick<MemoryFrontmatter, "project" | "kind" | "name">): MemoryFrontmatter;
+export declare function memoryToEmbeddingText(memory: Pick<ParsedMemory, "frontmatter" | "body" | "relativePath">): string;

package/dist/memory/files.js

@@ -0,0 +1,107 @@
+import { mkdir, readFile, stat, writeFile } from "node:fs/promises";
+import path from "node:path";
+import matter from "gray-matter";
+import { LINK_TYPES, MEMORY_KINDS } from "../constants.js";
+import { gzipDensity, sha256, stableJson } from "../util/hash.js";
+import { projectParts, safeSegment } from "./project.js";
+export function normalizeKind(kind) {
+    const normalized = (kind ?? "WORK").toUpperCase();
+    if (!MEMORY_KINDS.includes(normalized)) {
+        throw new Error(`Invalid memory kind: ${kind}`);
+    }
+    return normalized;
+}
+export function normalizeLinkType(type) {
+    const normalized = (type ?? "RELATED").toUpperCase();
+    if (!LINK_TYPES.includes(normalized)) {
+        throw new Error(`Invalid link type: ${type}`);
+    }
+    return normalized;
+}
+export function memoryRelativePath(projectName, kind, name) {
+    const parts = projectParts(projectName);
+    return path.posix.join(parts.org, parts.repo, kind, `${safeSegment(name)}.md`);
+}
+export function relativePathToMetadata(relativePath) {
+    const normalized = relativePath.replace(/\\/g, "/");
+    const parts = normalized.split("/");
+    if (parts.length < 4) {
+        throw new Error(`Memory path must be org/repo/kind/name.md: ${relativePath}`);
+    }
+    const [org, repo, kind, ...nameParts] = parts;
+    const filename = nameParts.join("/");
+    return {
+        project: `${safeSegment(org)}/${safeSegment(repo)}`,
+        kind: normalizeKind(kind),
+        name: safeSegment(path.posix.basename(filename, ".md")),
+    };
+}
+export async function readMemory(memoryRoot, relativePath) {
+    const normalizedRelativePath = normalizeRelativePath(relativePath);
+    const absolutePath = path.join(memoryRoot, normalizedRelativePath);
+    const raw = await readFile(absolutePath, "utf8");
+    const parsed = matter(raw);
+    const fallback = relativePathToMetadata(normalizedRelativePath);
+    const stats = await stat(absolutePath);
+    const frontmatter = normalizeFrontmatter(parsed.data, fallback);
+    const body = parsed.content.trim();
+    const frontmatterHash = sha256(stableJson(frontmatter));
+    return {
+        frontmatter,
+        body,
+        relativePath: normalizedRelativePath,
+        absolutePath,
+        contentHash: sha256(body),
+        frontmatterHash,
+        size: body.length,
+        density: gzipDensity(body),
+        mtime: stats.mtimeMs,
+    };
+}
+export async function writeMemory(memoryRoot, relativePath, frontmatter, body) {
+    const absolutePath = path.join(memoryRoot, normalizeRelativePath(relativePath));
+    await mkdir(path.dirname(absolutePath), { recursive: true });
+    const file = matter.stringify(`${body.trim()}\n`, frontmatter);
+    await writeFile(absolutePath, file, "utf8");
+}
+export function normalizeRelativePath(value) {
+    const normalized = value.replace(/\\/g, "/").replace(/^\/+/, "");
+    const parts = normalized.split("/").filter(Boolean);
+    if (parts.some((part) => part === "." || part === "..")) {
+        throw new Error(`Invalid relative memory path: ${value}`);
+    }
+    return parts.join("/");
+}
+export function normalizeFrontmatter(value, fallback) {
+    const now = new Date().toISOString();
+    const tags = Array.isArray(value.tags) ? value.tags.map(String).filter(Boolean) : [];
+    const rawLinks = Array.isArray(value.links) ? value.links : [];
+    return {
+        kind: normalizeKind(String(value.kind ?? fallback.kind)),
+        project: String(value.project ?? fallback.project),
+        name: safeSegment(String(value.name ?? fallback.name)),
+        tags,
+        links: rawLinks.map(normalizeLink),
+        created_at: String(value.created_at ?? now),
+        updated_at: String(value.updated_at ?? now),
+    };
+}
+export function memoryToEmbeddingText(memory) {
+    return [
+        memory.frontmatter.project,
+        memory.frontmatter.kind,
+        memory.frontmatter.name,
+        memory.relativePath,
+        memory.frontmatter.tags.join(" "),
+        "",
+        memory.body,
+    ].join("\n");
+}
+function normalizeLink(value) {
+    const input = value;
+    return {
+        to: normalizeRelativePath(String(input.to ?? "")),
+        type: normalizeLinkType(String(input.type ?? "RELATED")),
+    };
+}
+//# sourceMappingURL=files.js.map

package/dist/memory/files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"files.js","sourceRoot":"","sources":["../../src/memory/files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACpE,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAkC,MAAM,iBAAiB,CAAC;AAE3F,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEzD,MAAM,UAAU,aAAa,CAAC,IAAwB;IACpD,MAAM,UAAU,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;IAClD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAwB,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,KAAK,CAAC,wBAAwB,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,UAAwB,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAwB;IACxD,MAAM,UAAU,GAAG,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;IACrD,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAsB,CAAC,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,EAAE,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,UAAsB,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,WAAmB,EAAE,IAAgB,EAAE,IAAY;IACpF,MAAM,KAAK,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;IACxC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACjF,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,YAAoB;IACzD,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IACpD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACpC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,8CAA8C,YAAY,EAAE,CAAC,CAAC;IAChF,CAAC;IAED,MAAM,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC;IAC9C,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACrC,OAAO;QACL,OAAO,EAAE,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE;QACnD,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC;QACzB,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;KACxD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,UAAkB,EAAE,YAAoB;IACvE,MAAM,sBAAsB,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;IACnE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC;IACnE,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IACjD,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3B,MAAM,QAAQ,GAAG,sBAAsB,CAAC,sBAAsB,CAAC,CAAC;IAChE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,CAAC;IACvC,MAAM,WAAW,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAChE,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IACnC,MAAM,eAAe,GAAG,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,OAAO;QACL,WAAW;QACX,IAAI;QACJ,YAAY,EAAE,sBAAsB;QACpC,YAAY;QACZ,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC;QACzB,eAAe;QACf,IAAI,EAAE,IAAI,CAAC,MAAM;QACjB,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,KAAK,EAAE,KAAK,CAAC,OAAO;KACrB,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,UAAkB,EAClB,YAAoB,EACpB,WAA8B,EAC9B,IAAY;IAEZ,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,qBAAqB,CAAC,YAAY,CAAC,CAAC,CAAC;IAChF,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7D,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;IAC/D,MAAM,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAa;IACjD,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACjE,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACpD,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,KAA8B,EAC9B,QAA8D;IAE9D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACrF,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IAE/D,OAAO;QACL,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC;QACxD,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC;QAClD,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI;QACJ,KAAK,EAAE,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC;QAClC,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU,IAAI,GAAG,CAAC;QAC3C,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU,IAAI,GAAG,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAAmE;IACvG,OAAO;QACL,MAAM,CAAC,WAAW,CAAC,OAAO;QAC1B,MAAM,CAAC,WAAW,CAAC,IAAI;QACvB,MAAM,CAAC,WAAW,CAAC,IAAI;QACvB,MAAM,CAAC,YAAY;QACnB,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QACjC,EAAE;QACF,MAAM,CAAC,IAAI;KACZ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,MAAM,KAAK,GAAG,KAAgC,CAAC;IAC/C,OAAO;QACL,EAAE,EAAE,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QACjD,IAAI,EAAE,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC;KACzD,CAAC;AACJ,CAAC"}

package/dist/memory/project.d.ts

@@ -0,0 +1,8 @@
+export interface ProjectParts {
+    org: string;
+    repo: string;
+}
+export declare function parseGitRemote(remote: string): string | undefined;
+export declare function inferProjectName(cwd?: string): Promise<string>;
+export declare function projectParts(projectName: string): ProjectParts;
+export declare function safeSegment(value: string): string;

package/dist/memory/project.js

@@ -0,0 +1,59 @@
+import { execFile } from "node:child_process";
+import path from "node:path";
+import { promisify } from "node:util";
+import { sha256 } from "../util/hash.js";
+const execFileAsync = promisify(execFile);
+export function parseGitRemote(remote) {
+    const trimmed = remote.trim().replace(/\.git$/, "");
+    const sshMatch = /^git@[^:]+:([^/]+)\/(.+)$/.exec(trimmed);
+    if (sshMatch) {
+        return `${safeSegment(sshMatch[1])}/${safeSegment(sshMatch[2])}`;
+    }
+    try {
+        const url = new URL(trimmed);
+        const parts = url.pathname.split("/").filter(Boolean);
+        if (parts.length >= 2) {
+            return `${safeSegment(parts[parts.length - 2])}/${safeSegment(parts[parts.length - 1])}`;
+        }
+    }
+    catch {
+        return undefined;
+    }
+    return undefined;
+}
+export async function inferProjectName(cwd = process.cwd()) {
+    try {
+        const remote = await execGit(cwd, ["config", "--get", "remote.origin.url"]);
+        const remoteProject = parseGitRemote(remote);
+        if (remoteProject) {
+            return remoteProject;
+        }
+    }
+    catch {
+        // Fall through to local path identity.
+    }
+    const resolved = path.resolve(cwd);
+    const basename = path.basename(resolved) || "project";
+    const hash = sha256(resolved).slice(0, 8);
+    return `local/${safeSegment(basename)}-${hash}`;
+}
+export function projectParts(projectName) {
+    const [org, ...rest] = projectName.split("/");
+    const repo = rest.join("/");
+    if (!org || !repo) {
+        throw new Error(`Project name must be org/repo style: ${projectName}`);
+    }
+    return {
+        org: safeSegment(org),
+        repo: safeSegment(repo),
+    };
+}
+export function safeSegment(value) {
+    const safe = value.trim().replace(/\.md$/i, "").replace(/[^a-zA-Z0-9._-]+/g, "-").replace(/^-+|-+$/g, "");
+    return safe || "untitled";
+}
+async function execGit(cwd, args) {
+    const result = await execFileAsync("git", args, { cwd, timeout: 3_000 });
+    return result.stdout.trim();
+}
+//# sourceMappingURL=project.js.map

package/dist/memory/project.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.js","sourceRoot":"","sources":["../../src/memory/project.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEzC,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;AAO1C,MAAM,UAAU,cAAc,CAAC,MAAc;IAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,2BAA2B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3D,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACnE,CAAC;IAED,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7B,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACtD,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACtB,OAAO,GAAG,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3F,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE;IACxD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,mBAAmB,CAAC,CAAC,CAAC;QAC5E,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,uCAAuC;IACzC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,SAAS,CAAC;IACtD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,OAAO,SAAS,WAAW,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,WAAmB;IAC9C,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,wCAAwC,WAAW,EAAE,CAAC,CAAC;IACzE,CAAC;IACD,OAAO;QACL,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC;QACrB,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC;KACxB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAa;IACvC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;IAC1G,OAAO,IAAI,IAAI,UAAU,CAAC;AAC5B,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,GAAW,EAAE,IAAc;IAChD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IACzE,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;AAC9B,CAAC"}

package/dist/memory/scan.d.ts

@@ -0,0 +1 @@
+export declare function findMemoryFiles(memoryRoot: string): Promise<string[]>;

package/dist/memory/scan.js

@@ -0,0 +1,29 @@
+import { readdir } from "node:fs/promises";
+import path from "node:path";
+export async function findMemoryFiles(memoryRoot) {
+    const results = [];
+    await walk(memoryRoot, "", results);
+    return results.sort();
+}
+async function walk(root, relativeDir, results) {
+    let entries;
+    try {
+        entries = await readdir(path.join(root, relativeDir), { withFileTypes: true });
+    }
+    catch {
+        return;
+    }
+    for (const entry of entries) {
+        const relativePath = path.posix.join(relativeDir.replace(/\\/g, "/"), entry.name);
+        if (entry.isDirectory()) {
+            if (entry.name === ".git" || entry.name === "node_modules") {
+                continue;
+            }
+            await walk(root, relativePath, results);
+        }
+        else if (entry.isFile() && entry.name.toLowerCase().endsWith(".md")) {
+            results.push(relativePath);
+        }
+    }
+}
+//# sourceMappingURL=scan.js.map

package/dist/memory/scan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scan.js","sourceRoot":"","sources":["../../src/memory/scan.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,UAAkB;IACtD,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,IAAI,CAAC,UAAU,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IACpC,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;AACxB,CAAC;AAED,KAAK,UAAU,IAAI,CAAC,IAAY,EAAE,WAAmB,EAAE,OAAiB;IACtE,IAAI,OAAO,CAAC;IACZ,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IACjF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;IACT,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAClF,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACxB,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBAC3D,SAAS;YACX,CAAC;YACD,MAAM,IAAI,CAAC,IAAI,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC;aAAM,IAAI,KAAK,CAAC,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/service.d.ts

@@ -0,0 +1,43 @@
+import type { AugmentConfig } from "./config.js";
+import { AugmentDatabase } from "./db.js";
+import type { EmbeddingProvider } from "./embeddings.js";
+import type { LinkRecord, MemoryRecord, ProjectRecord, SearchInput, SearchResult, UpsertInput } from "./types.js";
+export interface InitProjectInput {
+    project_name?: string;
+}
+export interface UpsertResult {
+    memory: MemoryRecord;
+    action: "created" | "updated" | "similar-updated";
+    text: string;
+}
+export interface SearchOutput {
+    text: string;
+    results: SearchResult[];
+}
+export declare class AugmentService {
+    private readonly config;
+    private readonly embeddingProvider;
+    readonly db: AugmentDatabase;
+    private initialized;
+    constructor(config: AugmentConfig, embeddingProvider?: EmbeddingProvider);
+    close(): Promise<void>;
+    init(): Promise<void>;
+    initProject(input?: InitProjectInput): Promise<ProjectRecord>;
+    listProjects(name?: string): Promise<ProjectRecord[]>;
+    rebuildIndex(): Promise<{
+        indexed: number;
+    }>;
+    upsert(input: UpsertInput): Promise<UpsertResult>;
+    read(id: number): Promise<MemoryRecord>;
+    link(fromMemoryId: number, toMemoryId: number, typeInput?: string): Promise<{
+        link: LinkRecord;
+        text: string;
+    }>;
+    listLinks(memoryId: number): Promise<LinkRecord[]>;
+    unlink(linkId: number): Promise<{
+        text: string;
+    }>;
+    search(input: SearchInput): Promise<SearchOutput>;
+    private findSimilarMemory;
+    private ensureEmbedding;
+}