pi-spark 0.14.4 → 0.14.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-spark",
-  "version": "0.14.4",
+  "version": "0.14.5",
   "description": "Pi package that polishes your daily experience and keeps you at the frontier of agentic workflows.",
   "keywords": [
     "pi-coding-agent",

package/src/features/credits/providers/fireworks.ts

@@ -1,11 +1,8 @@
 import { fileURLToPath } from "node:url";
 
-import { credentials, loadPackageDefinition, Metadata } from "@grpc/grpc-js";
-import { loadSync } from "@grpc/proto-loader";
-
 import { toNumber } from "../../../utils/format";
 
-import type { ClientUnaryCall, ServiceClientConstructor, ServiceError } from "@grpc/grpc-js";
+import type { ClientUnaryCall, Metadata, ServiceClientConstructor, ServiceError } from "@grpc/grpc-js";
 import type { Credits, CreditsProvider } from "../types";
 
 type GatewayClient = InstanceType<ServiceClientConstructor>;
@@ -43,9 +40,16 @@ let client: GatewayClient | undefined;
 /** The API key maps to a fixed account, so cache the resolved resource name. */
 const accountByKey = new Map<string, string>();
 
-function getClient(): GatewayClient {
+async function getClient(): Promise<GatewayClient> {
   if (client) return client;
 
+  // @grpc/* is heavy to import (~45 ms cold); load it lazily so startup never pays for it unless
+  // Fireworks credits are actually fetched.
+  const [{ credentials, loadPackageDefinition }, { loadSync }] = await Promise.all([
+    import("@grpc/grpc-js"),
+    import("@grpc/proto-loader"),
+  ]);
+
   const protoPath = fileURLToPath(new URL("./fireworks.proto", import.meta.url));
   const definition = loadSync(protoPath, { keepCase: true, longs: String, defaults: true });
   const proto = loadPackageDefinition(definition) as unknown as { gateway: { Gateway: ServiceClientConstructor } };
@@ -55,12 +59,14 @@ function getClient(): GatewayClient {
 }
 
 async function unary<T>(method: string, request: object, apiKey: string, signal: AbortSignal): Promise<T> {
+  const grpc = await import("@grpc/grpc-js");
+  const gateway = await getClient();
+
   return new Promise<T>((resolve, reject) => {
-    const metadata = new Metadata();
+    const metadata = new grpc.Metadata();
     metadata.set("x-api-key", apiKey);
 
     const deadline = new Date(Date.now() + DEADLINE_MS);
-    const gateway = getClient();
     const invoke = gateway[method] as (
       request: object,
       metadata: Metadata,

package/src/features/web/client.ts

@@ -1,6 +1,4 @@
-import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
-
+import type { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import type { TextContent } from "@earendil-works/pi-ai";
 
 /** Exa's hosted MCP endpoint (Streamable HTTP). No API key required for the free tier. */
@@ -56,6 +54,13 @@ export class ExaClient {
 
     if (!this.connecting) {
       this.connecting = (async () => {
+        // The MCP SDK is heavy to import (~40 ms cold); load it lazily so startup never pays for
+        // it unless a web action actually runs.
+        const [{ Client }, { StreamableHTTPClientTransport }] = await Promise.all([
+          import("@modelcontextprotocol/sdk/client/index.js"),
+          import("@modelcontextprotocol/sdk/client/streamableHttp.js"),
+        ]);
+
         const next = new Client({ name: "pi-spark", version: "0" });
         const transport = new StreamableHTTPClientTransport(new URL(EXA_MCP_URL));
         await next.connect(transport as Parameters<Client["connect"]>[0]);