@bacnh85/pi-serena 0.1.3 → 0.2.0

package/README.md

@@ -69,6 +69,32 @@ Use the Pi-facing `pattern` field with `serena_search_for_pattern`:
 
 The extension maps `pattern` to Serena's backend `substring_pattern` parameter internally, so users do not need to call the Serena implementation detail directly.
 
+### Content replacement
+
+Use Serena's current `replace_content` API through Pi-facing fields:
+
+```json
+{
+  "relative_path": "src/example.py",
+  "needle": "old text",
+  "repl": "new text",
+  "mode": "literal"
+}
+```
+
+For regex replacement, set `mode` to `regex` and provide a Python regular expression in `needle`:
+
+```json
+{
+  "relative_path": "src/example.py",
+  "needle": "beginning.*?end",
+  "repl": "replacement",
+  "mode": "regex"
+}
+```
+
+The Pi bridge also implements `serena_get_current_config` and `serena_restart_language_server` directly, so they work even when Serena's same-named native tools are inactive in single-project or default contexts.
+
 ## Commands
 
 - `/serena-status [project]`

package/index.ts

@@ -54,7 +54,8 @@ const referencingSchema = Type.Object({
 	...controlSchema,
 	name_path: Type.String({ description: "Exact symbol name path from find_symbol." }),
 	relative_path: Type.String({ description: "File containing the symbol, relative to project root." }),
-	include_body: Type.Optional(Type.Boolean()),
+	include_kinds: Type.Optional(Type.Array(Type.Number({ description: "LSP symbol kind integers to include." }))),
+	exclude_kinds: Type.Optional(Type.Array(Type.Number({ description: "LSP symbol kind integers to exclude." }))),
 	max_answer_chars: MAX_CHARS_PARAM,
 });
 
@@ -114,10 +115,13 @@ const searchPatternSchema = Type.Object({
 const replaceContentSchema = Type.Object({
 	...controlSchema,
 	relative_path: Type.String({ description: "File path relative to the Serena project root." }),
-	old_string: Type.Optional(Type.String({ description: "Exact text or regex to replace, depending on Serena version." })),
-	new_string: Type.Optional(Type.String({ description: "Replacement text, depending on Serena version." })),
-	content: Type.Optional(Type.String({ description: "Replacement content for Serena versions that use content-based replacement." })),
-	regex: Type.Optional(Type.String({ description: "Regex pattern for Serena versions that use regex replacement." })),
+	needle: Type.Optional(Type.String({ description: "Exact text or regex pattern to replace." })),
+	repl: Type.Optional(Type.String({ description: "Replacement text." })),
+	mode: Type.Optional(Type.Union([Type.Literal("literal"), Type.Literal("regex")], { description: "Interpret needle as exact text or as a Python regex." })),
+	old_string: Type.Optional(Type.String({ description: "Deprecated alias for literal needle." })),
+	new_string: Type.Optional(Type.String({ description: "Deprecated alias for repl with old_string." })),
+	content: Type.Optional(Type.String({ description: "Deprecated alias for repl with regex." })),
+	regex: Type.Optional(Type.String({ description: "Deprecated alias for regex needle." })),
 	allow_multiple_occurrences: Type.Optional(Type.Boolean({ description: "Allow replacing multiple matches when supported." })),
 });
 
@@ -148,6 +152,41 @@ export function normalizeSearchPatternParams(params: Record<string, unknown>): R
 	return normalized;
 }
 
+export function normalizeFindReferencesParams(params: Record<string, unknown>): Record<string, unknown> {
+	const normalized = { ...params };
+	delete normalized.include_body;
+	return normalized;
+}
+
+export function normalizeReplaceContentParams(params: Record<string, unknown>): Record<string, unknown> {
+	const normalized = { ...params };
+
+	if (normalized.needle === undefined) {
+		normalized.needle = normalized.old_string ?? normalized.regex;
+	}
+	if (normalized.repl === undefined) {
+		normalized.repl = normalized.new_string ?? normalized.content;
+	}
+	if (normalized.mode === undefined) {
+		if (normalized.regex !== undefined) normalized.mode = "regex";
+		else if (normalized.old_string !== undefined) normalized.mode = "literal";
+	}
+
+	delete normalized.old_string;
+	delete normalized.new_string;
+	delete normalized.content;
+	delete normalized.regex;
+
+	return normalized;
+}
+
+function validateReplaceContentParams(params: Record<string, unknown>): string | undefined {
+	if (typeof params.needle !== "string") return "serena_replace_content requires string parameter 'needle' (or deprecated alias 'old_string'/'regex').";
+	if (typeof params.repl !== "string") return "serena_replace_content requires string parameter 'repl' (or deprecated alias 'new_string'/'content').";
+	if (params.mode !== "literal" && params.mode !== "regex") return "serena_replace_content requires mode to be 'literal' or 'regex'.";
+	return undefined;
+}
+
 function truncateText(text: string): string {
 	const lines = text.split("\n");
 	let output = lines.slice(0, OUTPUT_MAX_LINES).join("\n");
@@ -229,10 +268,10 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		return worker;
 	};
 
-	const callSerena = async (ctx: any, tool: string, rawParams: Record<string, unknown>, lockPath?: string) => {
+	const callWorkerAction = async (ctx: any, action: string, rawParams: Record<string, unknown>, extraPayload: Record<string, unknown> = {}, lockPath?: string) => {
 		const { project, context, timeoutMs, params } = stripControlParams(rawParams);
 		const run = async () => {
-			const response = await getWorker(ctx).request({ action: "call", project, context, tool, params }, timeoutMs);
+			const response = await getWorker(ctx).request({ action, project, context, params, ...extraPayload }, timeoutMs);
 			return {
 				content: [{ type: "text" as const, text: resultText(response) }],
 				details: response,
@@ -241,6 +280,10 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		return lockPath ? withFileMutationQueue(lockPath, run) : run();
 	};
 
+	const callSerena = async (ctx: any, tool: string, rawParams: Record<string, unknown>, lockPath?: string) => {
+		return callWorkerAction(ctx, "call", rawParams, { tool }, lockPath);
+	};
+
 	const lockPathForRelativeFile = (rawParams: Record<string, unknown>): string | undefined => {
 		const project = normalizeProject(rawParams.project);
 		return typeof rawParams.relative_path === "string" && rawParams.relative_path.trim()
@@ -314,7 +357,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_find_referencing_symbols before changing public behavior or renaming a symbol."],
 		parameters: referencingSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "find_referencing_symbols", params);
+			return callSerena(ctx, "find_referencing_symbols", normalizeFindReferencesParams(params));
 		},
 	});
 
@@ -398,7 +441,10 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Prefer symbol-aware Serena edit tools for whole symbols; use serena_replace_content for non-symbol scoped replacements supported by Serena."],
 		parameters: replaceContentSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "replace_content", params, lockPathForRelativeFile(params));
+			const normalized = normalizeReplaceContentParams(params);
+			const error = validateReplaceContentParams(normalized);
+			if (error) return { content: [{ type: "text" as const, text: `Error: ${error}` }], details: { ok: false, error } };
+			return callSerena(ctx, "replace_content", normalized, lockPathForRelativeFile(params));
 		},
 	});
 
@@ -410,7 +456,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_restart_language_server when Serena symbol retrieval appears stale or language-server diagnostics are stuck."],
 		parameters: emptyToolSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "restart_language_server", params);
+			return callWorkerAction(ctx, "restart_language_server", params);
 		},
 	});
 
@@ -422,7 +468,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_get_current_config when project activation, contexts, modes, or tool availability are unclear."],
 		parameters: emptyToolSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "get_current_config", params);
+			return callWorkerAction(ctx, "config", params);
 		},
 	});
 
@@ -531,7 +577,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 
 	pi.on("before_agent_start", async (event) => {
 		const prompt = event.prompt.toLowerCase();
-		if (/\b(find references|rename|refactor|symbol|declaration|implementation|diagnostic|large repo|semantic|class|method|function)\b/.test(prompt)) {
+		if (/\b(find references|rename|refactor|symbol|declaration|implementation|diagnostic|large repo|semantic|class|method|function|investigate|understand|explore|search for|look at|find where|navigate|trace|examine|where is|what does)\b/.test(prompt)) {
 			return {
 				systemPrompt:
 					event.systemPrompt +
@@ -540,6 +586,15 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		}
 	});
 
+	// Unconditional Serena guidance on every session (like pi-subagent's agent-list hook)
+	pi.on("before_agent_start", async (event) => {
+		return {
+			systemPrompt:
+				event.systemPrompt +
+				"\n\nSerena semantic tools are available for code-symbol work: serena_find_symbol, serena_get_symbols_overview, serena_find_referencing_symbols, and more. Prefer these over repeated grep/read for exploring code structure, finding symbols, navigating references, and whole-symbol edits. Normal read/search tools remain appropriate for Markdown, JSON/YAML, docs, and exact text edits.",
+		};
+	});
+
 	pi.on("tool_call", async (event) => {
 		if (event.toolName.startsWith("serena_")) {
 			semanticMissCount = 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bacnh85/pi-serena",
-  "version": "0.1.3",
+  "version": "0.2.0",
   "description": "Pi extension that provides Serena semantic code tools through a persistent worker.",
   "license": "MIT",
   "publishConfig": {

package/worker.ts

@@ -154,6 +154,22 @@ def handle(req: dict[str, Any]) -> dict[str, Any]:
         opened = bool(req.get("open")) and agent.open_dashboard()
         return {"id": req_id, "ok": True, "opened": opened, "dashboardUrl": agent.get_dashboard_url()}
 
+    if action == "config":
+        project = str(req.get("project") or os.getcwd())
+        context = str(req.get("context") or "ide")
+        agent = get_agent(project, context)
+        with contextlib.redirect_stdout(sys.stderr):
+            result = agent.get_current_config_overview()
+        return {"id": req_id, "ok": True, "tool": "get_current_config", "result": result}
+
+    if action == "restart_language_server":
+        project = str(req.get("project") or os.getcwd())
+        context = str(req.get("context") or "ide")
+        agent = get_agent(project, context)
+        with contextlib.redirect_stdout(sys.stderr):
+            agent.reset_language_server_manager()
+        return {"id": req_id, "ok": True, "tool": "restart_language_server", "result": "OK"}
+
     if action == "call":
         project = str(req.get("project") or os.getcwd())
         context = str(req.get("context") or "ide")