@yandy0725/pi-memory 1.2.0 → 1.2.1

package/index.ts

@@ -7,7 +7,6 @@ import { runDream } from "./src/dream";
 import { runExtract } from "./src/extract";
 import {
 	buildInjection,
-	buildSurfacingPrompt,
 	injectSurfacedContent,
 	loadIndexSnapshot,
 	runSideQuery,
@@ -98,17 +97,16 @@ export default function (pi: ExtensionAPI) {
 				if (ctx.hasUI) ctx.ui.setStatus("surfacing", "Searching relevant memories…");
 				const manifest = await scanTopics(memoryDir);
 				if (manifest.length > 0) {
-					const queryPrompt = buildSurfacingPrompt(manifest, event.prompt.slice(0, 4000), injectedTopics);
 					const selected = await runSideQuery(
-						queryPrompt,
 						manifest,
+						event.prompt.slice(0, 4000),
+						injectedTopics,
 						autoSurfacing.maxFiles,
 						autoSurfacing.thinkLevel,
 						autoSurfacing.model,
 						ctx.modelRegistry,
 						ctx.model,
 						memoryDir,
-						injectedTopics,
 					);
 					if (selected.length > 0) {
 						const content = await injectSurfacedContent(

package/package.json

@@ -3,7 +3,7 @@
 	"publishConfig": {
 		"access": "public"
 	},
-	"version": "1.2.0",
+	"version": "1.2.1",
 	"description": "File-system driven persistent memory layer for pi coding agent",
 	"license": "MIT",
 	"repository": {

package/src/inject.ts

@@ -60,25 +60,15 @@ export async function scanTopics(memoryDir: string): Promise<TopicManifest[]> {
 	return manifests.sort((a, b) => b.mtimeMs - a.mtimeMs);
 }
 
-export function buildSurfacingPrompt(
-	manifest: TopicManifest[],
-	userPrompt: string,
-	injectedTopics: Set<string>,
-): string {
+function buildSurfacingPrompt(manifest: TopicManifest[], userPrompt: string): string {
 	const lines = manifest.map((t) => {
-		const marker = injectedTopics.has(t.filename) ? " [already injected]" : "";
-		return `[${t.type}] ${t.filename} — ${t.description.slice(0, 80)}${marker}`;
+		return `[${t.type}] ${t.filename} — ${t.description.slice(0, 80)}`;
 	});
-	const alreadyInjected =
-		injectedTopics.size > 0
-			? `\n\nNote: ${injectedTopics.size} topic file(s) have already been injected in this session and are marked [already injected]. Prefer selecting uninjected topics.`
-			: "";
 
 	return [
 		"You are a memory relevance selector. Below is a list of memory topic files and a user message.",
 		"Select up to N topic files that are relevant to the user's current query.",
 		"Response format: JSON with a 'selected_files' array of filenames.",
-		alreadyInjected,
 		"",
 		"=== Topic Files ===",
 		...lines,
@@ -148,19 +138,20 @@ function parseSelectedFiles(result: string, candidates: TopicManifest[], maxFile
 /** Run a lightweight headless side-query to select relevant topic files.
  *  Returns [] on timeout/failure — no fallback. */
 export async function runSideQuery(
-	prompt: string,
 	manifest: TopicManifest[],
+	userPrompt: string,
+	injectedTopics: Set<string>,
 	maxFiles: number,
 	thinkLevel: ThinkLevel,
 	model: string | undefined,
 	modelRegistry: ModelRegistry,
 	parentModel: Model<any> | undefined,
 	memoryDir: string,
-	injectedTopics: Set<string>,
 ): Promise<string[]> {
 	const candidates = manifest.filter((t) => !injectedTopics.has(t.filename));
 	if (candidates.length === 0) return [];
-	const task = buildSideQueryTask(prompt, maxFiles);
+	const surfacingPrompt = buildSurfacingPrompt(candidates, userPrompt);
+	const task = buildSideQueryTask(surfacingPrompt, maxFiles);
 	try {
 		const result = await runHeadlessAgent({
 			task,