@atomic-chat/core 0.5.1 → 0.13.2

package/dist/runtime/environment/linux-provisioner.d.ts

@@ -0,0 +1,102 @@
+import type { ContainerRuntimeStepParameters, EnvironmentDistribution, GpuFacts, ManagedAvailability, ManagedBlocker, PlatformImage, Sha256Digest } from '../../contracts/index.js';
+import { pullImage, type DockerExec, type ExecutionRecord } from '../container/index.js';
+import type { RuntimeDescriptorProvider } from './descriptor-provider.js';
+import type { EnvironmentManifestProvider } from './environment-manifest-provider.js';
+import type { InstallationStore } from './installations.js';
+import type { LinuxHost } from './linux-host.js';
+import { type LinuxAssessment, type LinuxBlocker } from './linux-plan.js';
+import type { EnvironmentProvisioner } from './service.js';
+/** The privileged recipe, as the host module builds it (`src/host/recipes`, task 2.5). Injected. */
+export interface HostRecipeBinding {
+    recipe_id: string;
+    recipe_digest: Sha256Digest;
+    /** The validated parameters for this plan; throws `MANAGED_HOST_STEP_INVALID` on a refusal. */
+    parameters(plan: NonNullable<LinuxAssessment['install_plan']>, host: Omit<ContainerRuntimeStepParameters, 'components'>): ContainerRuntimeStepParameters;
+    parametersDigest(parameters: ContainerRuntimeStepParameters): Sha256Digest;
+}
+/** The startup Docker executor (`ManagedContainersHandle`), as far as setup and removal need it. */
+export interface ProvisionerDocker {
+    exec: DockerExec;
+    socketPath: string;
+    journal: {
+        list(): ExecutionRecord[];
+        remove(containerId: string): Promise<void>;
+    };
+}
+/** What a probe of the host says about the environment, for the snapshot. */
+export interface HostView {
+    availability: ManagedAvailability;
+    gpus: GpuFacts[];
+    blockers: ManagedBlocker[];
+    selinux: boolean | null;
+    /** Windows only: Atomic Chat's own distribution, once imported (change `add-tensorrt-llm-windows`). */
+    distribution?: EnvironmentDistribution | null;
+    /**
+     * Windows only: the WSL VM's memory as the guest reads it (`/proc/meminfo`), which the model check
+     * compares against instead of the host's (spec "Память и драйвер оцениваются по гостю"). Null when unread.
+     */
+    memory_bytes?: number | null;
+}
+/**
+ * Unloads every loaded session of one engine, confirming the stop of its container (spec "Удаление
+ * при загруженной модели"); rejects when a stop cannot be confirmed, holding nothing off then. On
+ * success, new loads of that engine stay refused until the removal calls `release` — which it does
+ * however it ends — so no load starts a container on an installation being removed (final review
+ * M-1). The `tensorrt-llm` provider supplies the real one; the default reports nothing unloaded.
+ */
+export type UnloadEngineSessions = (engineId: string) => Promise<{
+    unloaded: number;
+    release?: () => void;
+}>;
+export declare const NOTHING_LOADED: UnloadEngineSessions;
+export interface LinuxProvisionerDeps {
+    host: LinuxHost;
+    descriptors: RuntimeDescriptorProvider;
+    /** The Linux environment manifest: the install recipe's qualified distributions. */
+    environmentManifests: EnvironmentManifestProvider<'linux'>;
+    recipe: HostRecipeBinding;
+    /** The one executor of this core; null while this host has no docker CLI. */
+    docker: () => Promise<ProvisionerDocker | null>;
+    installations: InstallationStore;
+    environmentId: string;
+    /** Removes this scope's engine caches of one descriptor (`removeEngineCaches`, task 2.12). */
+    removeEngineCaches: (descriptorId: string) => Promise<void>;
+    /**
+     * Removes this scope's managed model store; only for `retain_models: false`, and only when no other
+     * managed engine's installation is left (change `add-vllm-runtime`, D13).
+     */
+    removeStoreModels: () => Promise<void>;
+    unloadEngineSessions?: UnloadEngineSessions;
+    onAssessment?: (view: HostView) => void;
+    newId?: () => string;
+    now?: () => Date;
+    /** Container operations; the container module's by default, a fake in unit tests. */
+    pull?: typeof pullImage;
+}
+/** How an image this operation pulled is named among its `owned_resource_ids`. */
+export declare const ownedImageId: (image: PlatformImage) => string;
+/**
+ * A structured `LinuxBlocker` on the wire. A relogin is its own code: the operation waits on it. A
+ * missing environment manifest is missing metadata, not a host prerequisite (spec
+ * `runtime-environment-manifest`, "Без манифеста не предлагается только автоматическая установка").
+ */
+export declare function toManagedBlocker(blocker: LinuxBlocker): ManagedBlocker;
+/**
+ * The card the GPU check runs on, and the one a first load would pick: the largest memory among
+ * cards new enough for the descriptor; a unified-memory card (no VRAM of its own) counts as zero.
+ */
+export declare function pickGpu(gpus: GpuFacts[], minimumComputeCapability: string): GpuFacts | null;
+/**
+ * A repository as Docker names it in `RepoDigests`: a Docker Hub image without its registry
+ * (`docker.io/vllm/vllm-openai` → `vllm/vllm-openai`) and an official one by its bare name
+ * (`docker.io/library/ubuntu` → `ubuntu`); every other registry as written (`nvcr.io/…`).
+ */
+export declare function dockerRepositoryName(repository: string): string;
+/**
+ * Whether `inspectImage`'s answer names exactly `image` among its repo digests. The digest must be
+ * the pinned one; the repository is compared in the form Docker reports it in, since Docker drops
+ * the registry of a Docker Hub image (vLLM's `docker.io/vllm/vllm-openai`, change
+ * `add-vllm-runtime`; found in the live run on Windows).
+ */
+export declare function imageMatchesDigest(inspected: unknown, image: PlatformImage): boolean;
+export declare function createLinuxProvisioner(deps: LinuxProvisionerDeps): EnvironmentProvisioner;