pi-codemcp 1.2.0 → 1.2.1

package/README.md

@@ -148,7 +148,7 @@ The Python sidecar enforces catalog cache TTL, execution timeout, per-tool timeo
 
 ## Search and execute flow
 
-The agent searches for a capability, inspects the selected exact stub when needed, and executes a compact plan:
+The agent searches for a capability, inspects the selected exact stub when needed, and executes a compact plan. Unscoped searches discover stale or missing server catalogs independently: available servers still return results, while `discovery_failures` explicitly reports unavailable servers. Server-scoped searches remain fail-fast.
 
 ```python
 issues = await linear.list_issues({"assignee": "me", "limit": 50})

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-codemcp",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "description": "Typed, sandboxed Code Mode access to configured MCP servers for Pi",
   "type": "module",
   "packageManager": "bun@1.3.10",

package/sidecar/gateway.py

@@ -36,6 +36,7 @@ from .models import (
     SearchDetail,
     SearchMode,
     SearchResponse,
+    ServerDiscoveryFailure,
     ServerToolSummary,
     StatusResponse,
     UpstreamStatus,
@@ -377,7 +378,11 @@ class GatewayRuntime:
             discovery_servers = ()
         else:
             discovery_servers = (server,)
-        await self._ensure_servers_discovered(discovery_servers)
+        if server is None:
+            discovery_failures = await self._discover_servers_for_unscoped_search(discovery_servers)
+        else:
+            await self._ensure_servers_discovered(discovery_servers)
+            discovery_failures = []
         self.stats_store.record_phase("discovery", _elapsed_ms(discovery_started))
         bounded_limit = min(max(limit, 1), 20)
         bounded_cursor = max(cursor, 0)
@@ -439,6 +444,7 @@ class GatewayRuntime:
             has_more=next_cursor is not None,
             project_scope_available=self.chain_store.project_store is not None,
             execution_limits=self._execution_limits_view(),
+            discovery_failures=discovery_failures,
             prelude=self.catalog.stub_prelude if include_prelude else None,
             results=results,
         )
@@ -981,6 +987,36 @@ class GatewayRuntime:
     async def _ensure_catalog_complete(self) -> None:
         await self._ensure_servers_discovered(self.handles.keys())
 
+    async def _discover_servers_for_unscoped_search(
+        self,
+        server_names: Iterable[str],
+    ) -> list[ServerDiscoveryFailure]:
+        requested = set(server_names)
+        missing = [
+            (name, handle)
+            for name, handle in self.handles.items()
+            if name in requested and handle.tools is None
+        ]
+        if not missing:
+            return []
+        results = await asyncio.gather(
+            *(handle.discover() for _, handle in missing),
+            return_exceptions=True,
+        )
+        failures: list[ServerDiscoveryFailure] = []
+        for (name, _handle), result in zip(missing, results, strict=True):
+            if isinstance(result, Exception):
+                failures.append(
+                    ServerDiscoveryFailure(
+                        server=name,
+                        error=_discovery_error_message(result),
+                    )
+                )
+            elif isinstance(result, BaseException):
+                raise result
+        await self._rebuild_catalog()
+        return failures
+
     async def _ensure_servers_discovered(self, server_names: Iterable[str]) -> None:
         requested = set(server_names)
         missing = [
@@ -1170,6 +1206,10 @@ def _elapsed_ms(started: float) -> float:
     return (time.perf_counter() - started) * 1_000
 
 
+def _discovery_error_message(error: Exception) -> str:
+    return str(error).strip() or type(error).__name__
+
+
 def _parse_code(code: str) -> ast.AST | None:
     normalized = textwrap.dedent(code).strip("\n")
     wrapped = f"async def __codemcp_main():\n{textwrap.indent(normalized, '    ')}\n"

package/sidecar/models.py

@@ -63,6 +63,13 @@ class ServerToolSummary(BaseModel):
     tool_count: int
 
 
+class ServerDiscoveryFailure(BaseModel):
+    model_config = ConfigDict(extra="forbid", strict=True)
+
+    server: str
+    error: str
+
+
 class ExecutionLimitsView(BaseModel):
     model_config = ConfigDict(extra="forbid", strict=True)
 
@@ -85,6 +92,7 @@ class SearchResponse(BaseModel):
     has_more: bool = False
     project_scope_available: bool
     execution_limits: ExecutionLimitsView
+    discovery_failures: list[ServerDiscoveryFailure] = Field(default_factory=list)
     prelude: str | None = None
     results: list[ToolSchemaView]
 

package/src/tools.ts

@@ -30,6 +30,7 @@ interface SearchRenderDetails extends CodeMcpOutputDetails {
   nextCursor?: number;
   detail: string;
   preview: string[];
+  discoveryFailures: string[];
 }
 
 const SearchParameters = Type.Object({
@@ -154,7 +155,7 @@ export function registerCodeMcpTools(
     name: "codemcp_search",
     label: "MCP Search",
     description:
-      "Search configured MCP capabilities or page through compact inventory. Default signature search includes exact stubs for up to three top matches plus compact alternatives; codemcp_inspect loads other selected stubs. Returns ranking evidence, pagination, scope, and execution limits; invalid server names fail with suggestions.",
+      "Search configured MCP capabilities or page through compact inventory. Default signature search includes exact stubs for up to three top matches plus compact alternatives; codemcp_inspect loads other selected stubs. Unscoped searches return available results plus explicit discovery failures; scoped searches remain fail-fast. Returns ranking evidence, pagination, scope, and execution limits; invalid server names fail with suggestions.",
     promptSnippet: "Discover compact MCP capabilities or inventory",
     promptGuidelines: [...SEARCH_PROMPT_GUIDELINES],
     parameters: SearchParameters,
@@ -186,6 +187,11 @@ export function registerCodeMcpTools(
             return [`${String(item.name)} ${String(item.tool_count ?? 0)}`];
           })
         : [];
+      const discoveryFailures = Array.isArray(result.discovery_failures)
+        ? result.discovery_failures.flatMap((item) =>
+            isRecord(item) && typeof item.server === "string" ? [item.server] : [],
+          )
+        : [];
       return {
         content: [{ type: "text", text: output.text }],
         details: {
@@ -197,6 +203,7 @@ export function registerCodeMcpTools(
           nextCursor: typeof result.next_cursor === "number" ? result.next_cursor : undefined,
           detail: typeof result.detail === "string" ? result.detail : "signatures",
           preview,
+          discoveryFailures,
         },
       };
     },
@@ -212,13 +219,18 @@ export function registerCodeMcpTools(
       if (isPartial) return new Text(theme.fg("warning", "Searching catalog..."), 0, 0);
       if (expanded) return renderExpandedJson(result.content);
       const details = result.details as SearchRenderDetails | undefined;
+      const discoveryFailures = details?.discoveryFailures ?? [];
+      const unavailable = discoveryFailures.length;
       let text = theme.fg(
-        "success",
-        `${details?.matchCount ?? 0} matches · ${details?.detail ?? "signatures"} · ${details?.totalToolCount ?? 0} tools · ${details?.serverCount ?? 0} servers`,
+        unavailable > 0 ? "warning" : "success",
+        `${details?.matchCount ?? 0} matches · ${details?.detail ?? "signatures"} · ${details?.totalToolCount ?? 0} tools · ${details?.serverCount ?? 0} servers${unavailable > 0 ? ` · ${unavailable} unavailable` : ""}`,
       );
       for (const name of details?.preview ?? []) {
         text += `\n${theme.fg("dim", `  ${name}`)}`;
       }
+      for (const server of discoveryFailures) {
+        text += `\n${theme.fg("warning", `  unavailable: ${server}`)}`;
+      }
       if (details?.hasMore) {
         text += `\n${theme.fg("muted", `  more at cursor ${details.nextCursor ?? "?"}`)}`;
       }