fix(server): allow more time for Pi model discovery

apps/server/src/provider/Layers/PiProvider.test.ts

@@ -1,7 +1,11 @@
 import * as NodeServices from "@effect/platform-node/NodeServices";
 import { assert, describe, it } from "@effect/vitest";
+import * as Deferred from "effect/Deferred";
 import * as Effect from "effect/Effect";
+import * as Fiber from "effect/Fiber";
+import * as Queue from "effect/Queue";
 import * as Sink from "effect/Sink";
+import { TestClock } from "effect/testing";
 import * as Stream from "effect/Stream";
 import { ChildProcess, ChildProcessSpawner } from "effect/unstable/process";
 
@@ -44,14 +48,98 @@ function piProbeSpawner(version: string) {
   });
 }
 
+function delayedPiProbeSpawner(startupDelayMs: number, started: Deferred.Deferred<void>) {
+  return ChildProcessSpawner.make((command) =>
+    Effect.gen(function* () {
+      const args = ChildProcess.isStandardCommand(command) ? command.args : [];
+      if (args.includes("--version")) return processHandle({ stdout: "pi 0.99.1\n" });
+      assert.include(args, "--no-session");
+      assert.notInclude(args, "--no-extensions");
+      const responses = yield* Queue.unbounded<Uint8Array>();
+      return ChildProcessSpawner.makeHandle({
+        ...processHandle({}),
+        exitCode: Effect.never,
+        stdin: Sink.forEach((chunk: Uint8Array) =>
+          Effect.gen(function* () {
+            const request = JSON.parse(new TextDecoder().decode(chunk)) as {
+              id: string;
+              type: string;
+            };
+            if (request.type === "get_state") {
+              yield* Deferred.succeed(started, undefined);
+              yield* Effect.sleep(startupDelayMs);
+            }
+            const data =
+              request.type === "get_available_models"
+                ? { models: [{ provider: "extension-provider", id: "custom-model" }] }
+                : request.type === "get_commands"
+                  ? { commands: [] }
+                  : {};
+            yield* Queue.offer(
+              responses,
+              encoder.encode(
+                `${JSON.stringify({ type: "response", id: request.id, success: true, data })}\n`,
+              ),
+            );
+          }),
+        ),
+        stdout: Stream.fromQueue(responses),
+      });
+    }),
+  );
+}
+
 const settings = {
   enabled: true,
-  binaryPath: "pi",
+  binaryPath: "pi-test",
   launchArgs: "",
   customModels: [],
 } as const;
 
 describe("PiProvider", () => {
+  it.effect("discovers extension models when RPC startup takes more than 15 seconds", () =>
+    Effect.gen(function* () {
+      const started = yield* Deferred.make<void>();
+      const probe = yield* checkPiProviderStatus(settings).pipe(
+        Effect.provideService(
+          ChildProcessSpawner.ChildProcessSpawner,
+          delayedPiProbeSpawner(20_000, started),
+        ),
+        Effect.forkChild,
+      );
+      yield* Deferred.await(started);
+      yield* TestClock.adjust("20 seconds");
+      const snapshot = yield* Fiber.join(probe);
+      assert.equal(snapshot.auth.status, "authenticated");
+      assert.deepEqual(
+        snapshot.models.map((model) => model.slug),
+        ["default", "extension-provider/custom-model"],
+      );
+    }).pipe(Effect.provide(NodeServices.layer)),
+  );
+
+  it.effect("still falls back when RPC startup exceeds the discovery deadline", () =>
+    Effect.gen(function* () {
+      const started = yield* Deferred.make<void>();
+      const probe = yield* checkPiProviderStatus(settings).pipe(
+        Effect.provideService(
+          ChildProcessSpawner.ChildProcessSpawner,
+          delayedPiProbeSpawner(60_000, started),
+        ),
+        Effect.forkChild,
+      );
+      yield* Deferred.await(started);
+      yield* TestClock.adjust("31 seconds");
+      const snapshot = yield* Fiber.join(probe);
+      assert.equal(snapshot.status, "ready");
+      assert.equal(snapshot.auth.status, "unknown");
+      assert.deepEqual(
+        snapshot.models.map((model) => model.slug),
+        ["default"],
+      );
+    }).pipe(Effect.provide(NodeServices.layer)),
+  );
+
   it.effect("requires the first published Pi version with entries and settlement hooks", () =>
     Effect.gen(function* () {
       const snapshot = yield* checkPiProviderStatus(settings).pipe(

apps/server/src/provider/Layers/PiProvider.ts

@@ -68,7 +68,7 @@ const PI_PRESENTATION = {
 } as const;
 
 const VERSION_PROBE_TIMEOUT_MS = 4_000;
-const PI_RPC_DISCOVERY_TIMEOUT_MS = 15_000;
+const PI_RPC_DISCOVERY_TIMEOUT_MS = 30_000;
 /**
  * get_entries arrived in 0.80.3 and agent_settled landed in source at 0.80.4.
  * Version 0.80.5 was the first published package containing both hooks. T3