@folterung/project-memory 0.1.19 → 0.1.21

package/packages/server/src/api/store.test.ts

@@ -0,0 +1,91 @@
+import { strict as assert } from "node:assert";
+import { describe, it } from "node:test";
+import Fastify from "fastify";
+import type { QdrantClient } from "@qdrant/js-client-rest";
+import { registerStoreRoutes } from "./store.js";
+
+function createMockStoreClient(): QdrantClient {
+  return {
+    getCollection: async () => {
+      throw new Error("not found");
+    },
+    createCollection: async () => {},
+    upsert: async () => {},
+  } as unknown as QdrantClient;
+}
+
+describe("store API", () => {
+  it("returns 400 when content is empty", async () => {
+    const app = Fastify();
+    registerStoreRoutes(app, { qdrantClient: createMockStoreClient() });
+    const res = await app.inject({
+      method: "POST",
+      url: "/store",
+      payload: { content: "   " },
+    });
+    assert.strictEqual(res.statusCode, 400);
+    const body = JSON.parse(res.body);
+    assert.match(body.message, /content is required/);
+  });
+
+  it("returns 400 when startLine is not a positive integer", async () => {
+    const app = Fastify();
+    registerStoreRoutes(app, { qdrantClient: createMockStoreClient() });
+    const res = await app.inject({
+      method: "POST",
+      url: "/store",
+      payload: { content: "note", startLine: 0 },
+    });
+    assert.strictEqual(res.statusCode, 400);
+    assert.match(JSON.parse(res.body).message, /startLine/);
+  });
+
+  it("returns 400 when endLine is not a positive integer", async () => {
+    const app = Fastify();
+    registerStoreRoutes(app, { qdrantClient: createMockStoreClient() });
+    const res = await app.inject({
+      method: "POST",
+      url: "/store",
+      payload: { content: "note", endLine: -1 },
+    });
+    assert.strictEqual(res.statusCode, 400);
+    assert.match(JSON.parse(res.body).message, /endLine/);
+  });
+
+  it("returns 200 and stores with anchor fields and scope when valid", async () => {
+    const app = Fastify();
+    registerStoreRoutes(app, { qdrantClient: createMockStoreClient() });
+    const res = await app.inject({
+      method: "POST",
+      url: "/store",
+      payload: {
+        content: "Summary of auth flow",
+        title: "Session summary: auth",
+        path: "src/auth.ts",
+        startLine: 10,
+        endLine: 20,
+        symbol: "login",
+        scope: "branch:main",
+        supersedesId: "old-id-123",
+      },
+    });
+    assert.strictEqual(res.statusCode, 200);
+    const body = JSON.parse(res.body);
+    assert.strictEqual(typeof body.id, "string");
+    assert.match(body.id, /^agent_note:/);
+    assert.strictEqual(body.message, "Stored.");
+  });
+
+  it("returns 200 for minimal valid body", async () => {
+    const app = Fastify();
+    registerStoreRoutes(app, { qdrantClient: createMockStoreClient() });
+    const res = await app.inject({
+      method: "POST",
+      url: "/store",
+      payload: { content: "Minimal note" },
+    });
+    assert.strictEqual(res.statusCode, 200);
+    const body = JSON.parse(res.body);
+    assert.match(body.id, /^agent_note:/);
+  });
+});

package/packages/server/src/api/store.ts

@@ -1,5 +1,6 @@
 import { randomUUID } from "node:crypto";
 import type { FastifyInstance } from "fastify";
+import type { QdrantClient } from "@qdrant/js-client-rest";
 import { createQdrantClient } from "../qdrant/client.js";
 import { ensureCollection, upsertPoints } from "../qdrant/client.js";
 import { embedText } from "../embedding/stub.js";
@@ -8,6 +9,11 @@ export interface StoreBody {
   content: string;
   title?: string;
   path?: string;
+  startLine?: number;
+  endLine?: number;
+  symbol?: string;
+  scope?: string;
+  supersedesId?: string;
 }
 
 const KIND_AGENT_NOTE = "agent_note";
@@ -16,18 +22,18 @@ function buildTextToEmbed(body: StoreBody): string {
   const parts: string[] = [];
   if (body.path) parts.push(`File: ${body.path}`);
   if (body.title) parts.push(`Title: ${body.title}`);
+  if (body.symbol) parts.push(`Symbol: ${body.symbol}`);
   parts.push(body.content);
   return parts.join("\n\n");
 }
 
 export function registerStoreRoutes(
   app: FastifyInstance,
-  options: { qdrantHost?: string; qdrantPort?: number; collection?: string }
+  options: { qdrantHost?: string; qdrantPort?: number; collection?: string; qdrantClient?: QdrantClient }
 ): void {
-  const client = createQdrantClient(
-    options.qdrantHost ?? "127.0.0.1",
-    options.qdrantPort ?? 6333
-  );
+  const client =
+    options.qdrantClient ??
+    createQdrantClient(options.qdrantHost ?? "127.0.0.1", options.qdrantPort ?? 6333);
   const collection = options.collection ?? "project_memory";
 
   app.post<{ Body: StoreBody }>(
@@ -41,16 +47,44 @@ export function registerStoreRoutes(
             content: { type: "string" },
             title: { type: ["string", "null"] },
             path: { type: ["string", "null"] },
+            startLine: { type: ["number", "null"] },
+            endLine: { type: ["number", "null"] },
+            symbol: { type: ["string", "null"] },
+            scope: { type: ["string", "null"] },
+            supersedesId: { type: ["string", "null"] },
           },
         },
       },
     },
     async (request, reply) => {
       const raw = request.body;
-      const body = {
+      const startLine = raw.startLine != null ? Number(raw.startLine) : undefined;
+      const endLine = raw.endLine != null ? Number(raw.endLine) : undefined;
+      if (startLine != null && (!Number.isInteger(startLine) || startLine < 1)) {
+        await reply.status(400).send({
+          statusCode: 400,
+          error: "Bad Request",
+          message: "startLine must be a positive integer when provided",
+        });
+        return;
+      }
+      if (endLine != null && (!Number.isInteger(endLine) || endLine < 1)) {
+        await reply.status(400).send({
+          statusCode: 400,
+          error: "Bad Request",
+          message: "endLine must be a positive integer when provided",
+        });
+        return;
+      }
+      const body: StoreBody = {
         content: raw.content ?? "",
         title: raw.title ?? undefined,
         path: raw.path ?? undefined,
+        startLine,
+        endLine,
+        symbol: raw.symbol ?? undefined,
+        scope: raw.scope ?? undefined,
+        supersedesId: raw.supersedesId ?? undefined,
       };
       const content = body.content.trim();
       if (!content) {
@@ -65,18 +99,32 @@ export function registerStoreRoutes(
       const textToEmbed = buildTextToEmbed({ ...body, content });
       const vector = embedText(textToEmbed);
       const id = randomUUID();
+      const storedAt = new Date().toISOString();
+      const source =
+        body.title?.trim().toLowerCase().startsWith("session summary")
+          ? "session-summary"
+          : "store";
+
+      const payload = {
+        kind: KIND_AGENT_NOTE,
+        path: body.path,
+        title: body.title,
+        text: content,
+        storedAt,
+        source,
+        ...(body.startLine != null && { startLine: body.startLine }),
+        ...(body.endLine != null && { endLine: body.endLine }),
+        ...(body.symbol != null && { symbol: body.symbol }),
+        ...(body.scope != null && { scope: body.scope }),
+        ...(body.supersedesId != null && { supersedesId: body.supersedesId }),
+      };
 
       await ensureCollection(client, collection);
       await upsertPoints(client, collection, [
         {
           id,
           vector,
-          payload: {
-            kind: KIND_AGENT_NOTE,
-            path: body.path,
-            title: body.title,
-            text: content,
-          },
+          payload,
         },
       ]);
 

package/packages/server/src/index.ts

@@ -5,6 +5,7 @@ import { registerExplainRoutes } from "./api/explain.js";
 import { registerSystemRoutes } from "./api/system.js";
 import { registerRefreshRoutes } from "./api/refresh.js";
 import { registerStoreRoutes } from "./api/store.js";
+import { registerPointsRoutes } from "./api/points.js";
 
 const PORT = Number(process.env.MEM_SERVER_PORT) || 31415;
 const CWD = process.env.MEM_CWD || process.cwd();
@@ -20,6 +21,7 @@ async function main() {
   registerSystemRoutes(app, { qdrantHost: QDRANT_HOST, qdrantPort: QDRANT_PORT, collection: COLLECTION });
   registerRefreshRoutes(app);
   registerStoreRoutes(app, { qdrantHost: QDRANT_HOST, qdrantPort: QDRANT_PORT, collection: COLLECTION });
+  registerPointsRoutes(app, { qdrantHost: QDRANT_HOST, qdrantPort: QDRANT_PORT, collection: COLLECTION });
 
   await app.listen({ port: PORT, host: "127.0.0.1" });
   console.log(`Memory server at http://127.0.0.1:${PORT}`);

package/packages/server/src/qdrant/client.ts

@@ -127,3 +127,18 @@ async function doSearch(
     payload: (r.payload ?? {}) as PointPayload,
   }));
 }
+
+export async function retrievePointsById(
+  client: QdrantClient,
+  collectionName: string,
+  ids: string[]
+): Promise<{ id: string; payload: PointPayload }[]> {
+  if (ids.length === 0) return [];
+  const retrieve = (client as { retrieve?: (c: string, o: { ids: string[] }) => Promise<{ id: unknown; payload?: Record<string, unknown> }[]> }).retrieve;
+  if (typeof retrieve !== "function") return [];
+  const result = await retrieve.call(client, collectionName, { ids, with_payload: true });
+  return (result ?? []).map((r) => ({
+    id: (r.id as string) ?? String(r.id),
+    payload: (r.payload ?? {}) as PointPayload,
+  }));
+}