pi-ghost-autocomplete 0.10.0 → 0.11.0

package/dist/providers/pi-model-provider.d.ts

@@ -0,0 +1,78 @@
+/**
+ * SPDX-License-Identifier: SSPL-1.0
+ * SPDX-FileCopyrightText: 2025 Alessandro Pungitore <alessandro@pungitore.ch>
+ */
+import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
+import type { EmptyReason } from "../metrics.js";
+import type { PiModelProviderConfig } from "../config.js";
+import { type GhostProvider, type GhostRequest, type StreamDelta } from "./types.js";
+/**
+ * Pi-registry backend: complete through the session's own `ModelRegistry`.
+ *
+ * Unlike {@link CloudProvider} — which can only resolve models from pi-ai's
+ * *static builtin* catalog and must hand-resolve auth (skipping registries
+ * with a custom `baseUrl`) — this backend delegates both concerns to pi:
+ *
+ *  - **Model resolution** via `registry.find(provider, id)` / first match in
+ *    `registry.getAvailable()`. Works for dynamic providers (e.g. "radius")
+ *    and any model registered by other extensions, none of which exist in
+ *    the static catalog.
+ *  - **Auth + endpoint** via `registry.complete()`, which resolves
+ *    request-time auth (models.json keys, stored credentials, OAuth) and the
+ *    provider's real `baseUrl` exactly as the main agent sees them
+ *    (`ModelRuntime.prepareRequest`). No ghost API key needed.
+ *
+ * Available on pi 0.84+; streaming additionally requires 0.99's
+ * `registry.stream` (feature-detected, falls back to the pipeline's
+ * non-streaming `predict()` path on 0.84).
+ */
+export declare class PiModelProvider implements GhostProvider {
+    readonly name: string;
+    private readonly registry;
+    private readonly config;
+    private readonly streamFn;
+    private resolved;
+    private resolvedSpec;
+    private lastEmpty;
+    /**
+     * Streaming variant — attached in the constructor **only** when the
+     * registry exposes `stream()` (pi ≥ 0.99). The request pipeline selects
+     * the streaming path whenever `predictStream` is present, so a class
+     * method would make every host look streaming-capable and silently starve
+     * pi 0.84–0.98 of ghosts instead of falling back to `predict()`.
+     */
+    predictStream?: (req: GhostRequest) => AsyncIterable<StreamDelta>;
+    constructor(config: PiModelProviderConfig, registry: ModelRegistry);
+    get debounceMs(): number;
+    consumeLastEmptyReason(): EmptyReason | null;
+    /**
+     * Resolve `modelSpec` (`"provider/model-id"` or bare `"model-id"`) to a
+     * registry model. Successful lookups are cached per spec; failures retry
+     * on the next call (the registry may still be loading models.json early
+     * in a session, and a failed scan of the model list is cheap).
+     */
+    private resolveModel;
+    /**
+     * One completion through the registry. Auth, headers, and endpoint are
+     * resolved by pi at request time; we only supply sampling knobs, the
+     * abort signal, and the SEC-3 timeout.
+     */
+    predict(req: GhostRequest): Promise<string | null>;
+    /**
+     * Streaming implementation behind the conditionally-attached
+     * {@link predictStream} — only reachable when the registry exposes
+     * `stream()` (pi ≥ 0.99; see {@link StreamingRegistry}). Mirrors
+     * {@link CloudProvider.predictStream}: sanitize once on the final
+     * assembled text, never per delta.
+     */
+    private streamPredict;
+    /**
+     * Preflight for /ghost diagnostics and router chat-leg warnings: resolves
+     * the spec and (when the registry supports it) checks that the model's
+     * provider has configured auth.
+     */
+    preflight(): Promise<{
+        ok: boolean;
+        message: string;
+    }>;
+}