pi-mcp-extension 1.5.3 → 1.6.0

package/README.md

@@ -159,6 +159,8 @@ Config files are loaded from two locations. **Project config overrides global co
 
 ## Commands
 
+Server-name arguments provide autocomplete suggestions from the configured MCP servers. `/mcp:auth` suggests only servers configured for OAuth.
+
 | Command | Description |
 |---|---|
 | `/mcp` | Show status summary of all configured servers |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-mcp-extension",
-  "version": "1.5.3",
+  "version": "1.6.0",
   "description": "MCP (Model Context Protocol) client extension for the Pi coding agent — connect Pi to any MCP server",
   "type": "module",
   "keywords": [

package/src/index.ts

@@ -157,6 +157,42 @@ export function parseAuthArguments(args: string, serverNames: string[]): { name:
   return { name: reset ? name.replace(/\s+--reset$/, "").trimEnd() : name, reset };
 }
 
+interface ServerCompletionEntry {
+  name: string;
+  state: string;
+  config: {
+    transport: string;
+    lifecycle: string;
+    auth?: unknown;
+  };
+}
+
+export function serverNameCompletions(
+  argumentPrefix: string,
+  servers: ServerCompletionEntry[],
+  options: { authOnly?: boolean } = {},
+) {
+  const prefix = argumentPrefix.trimStart();
+  const completions = servers
+    .filter((server) => !options.authOnly || server.config.auth)
+    .filter((server) => server.name.startsWith(prefix))
+    .sort((a, b) => a.name.localeCompare(b.name))
+    .map((server) => ({
+      value: server.name,
+      label: server.name,
+      description: [
+        server.state,
+        server.config.transport,
+        server.config.lifecycle,
+        server.config.auth ? "OAuth" : undefined,
+      ]
+        .filter(Boolean)
+        .join(" · "),
+    }));
+
+  return completions.length > 0 ? completions : null;
+}
+
 export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string; globalConfigPath?: string; authStorageDir?: string; browserOpen?: (url: string) => void } = {}): Promise<void> {
   // ── 1. Load and validate config ──────────────────────────────────────────
   // cwd is available on the ExtensionContext passed to event handlers.
@@ -277,6 +313,7 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
         ctx.ui.notify(manager.getStatusSummary(), "info");
       }
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers()),
   });
 
   // ── 5. /mcp:stop — stop a server ─────────────────────────────────────────
@@ -297,6 +334,7 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
       await manager.stopServer(serverName);
       ctx.ui.notify(`pi-mcp: Stopped ${serverName}`, "info");
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers()),
   });
 
   // ── 6. /mcp:start — manually start a lazy server ─────────────────────────
@@ -320,6 +358,7 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
         ctx.ui.notify(`pi-mcp: Failed to start ${serverName} — ${msg}`, "error");
       }
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers()),
   });
 
   async function authorize(serverName: string, ctx: ExtensionContext, signal: AbortSignal, reset: boolean): Promise<void> {
@@ -565,5 +604,6 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
         ctx.ui.notify(`pi-mcp: ${err instanceof Error ? err.message : String(err)}`, err instanceof AuthCancelledError ? "info" : "error");
       }
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers(), { authOnly: true }),
   });
 }

package/src/tool-bridge.ts

@@ -226,7 +226,7 @@ export function buildToolName(prefix: string, serverName: string, toolName: stri
 type PiTextContent = { type: "text"; text: string };
 type PiImageContent = { type: "image"; data: string; mimeType: string };
 
-function convertMcpContent(items: unknown[]): (PiTextContent | PiImageContent)[] {
+export function convertMcpContent(items: unknown[]): (PiTextContent | PiImageContent)[] {
   return items.map((item: any) => {
     if (!item || typeof item !== "object") {
       return { type: "text", text: String(item) };
@@ -448,7 +448,8 @@ export class ToolBridge {
           // Tool execution errors (isError: true) — distinct from protocol errors
           if (result.isError) {
             const errorText = content
-              .map((c) => (c.type === "text" ? c.text : ""))
+              .filter((c): c is PiTextContent => c.type === "text")
+              .map((c) => c.text)
               .join("\n");
             throw new McpError(
               errorText || "Tool reported an error",