@qvac/harness 0.3.3 → 0.5.0

package/dist-lib/lib/capabilities/download.d.ts

@@ -7,18 +7,18 @@ export type DownloadProgressEvent = {
     type: 'progress';
     transferred?: boolean;
 } & DownloadProgress;
-export type DownloadCompleteEvent = {
+export interface DownloadCompleteEvent {
     type: 'complete';
     assetId: string;
     downloaded?: boolean;
-};
-export type DownloadErrorEvent = {
+}
+export interface DownloadErrorEvent {
     type: 'error';
     message: string;
-};
-export type DownloadAbortedEvent = {
+}
+export interface DownloadAbortedEvent {
     type: 'aborted';
-};
+}
 export type DownloadEvent = DownloadProgressEvent | DownloadCompleteEvent | DownloadErrorEvent | DownloadAbortedEvent;
 export interface DownloadSdk extends ModelRegistrySdk {
     downloadAsset(options: {

package/dist-lib/lib/capabilities/download.js

@@ -1,5 +1,5 @@
 import { Readable } from 'streamx';
-import { errorMessage } from '../error-fields.js';
+import { errorMessage } from '../guards.js';
 import { resolveModelSrc } from '../models/model-source.js';
 const noop = () => { };
 export function download(sdkOrLoader, args, onSettled) {
@@ -30,12 +30,14 @@ export function download(sdkOrLoader, args, onSettled) {
         finish({ type: 'aborted' });
     };
     const onProgress = (p) => {
-        if (!settled)
+        if (!settled) {
             stream.push({ ...p, type: 'progress', transferred });
+        }
     };
     const start = async () => {
-        if (signal?.aborted)
+        if (signal?.aborted) {
             return finish({ type: 'aborted' });
+        }
         try {
             sdkRef = await loadSdk();
             const resolved = await resolveModelSrc(sdkRef, model, addon);
@@ -44,8 +46,9 @@ export function download(sdkOrLoader, args, onSettled) {
             const aggregate = aggregateLayout(srcs, assetSizes);
             let mainAssetId = '';
             for (let i = 0; i < srcs.length; i++) {
-                if (signal?.aborted)
+                if (signal?.aborted) {
                     return finish({ type: 'aborted' });
+                }
                 const progress = aggregate
                     ? aggregateProgressHandler(onProgress, {
                         setKey: mainSrc,
@@ -61,23 +64,26 @@ export function download(sdkOrLoader, args, onSettled) {
                     assetSrc: srcs[i],
                     seed,
                     onProgress: (update) => {
-                        if (update.percentage < 100)
+                        if (update.percentage < 100) {
                             transferred = true;
+                        }
                         progress(update);
                     }
                 });
                 requestId = op.requestId;
                 const assetId = await op;
-                if (i === 0)
+                if (i === 0) {
                     mainAssetId = assetId;
+                }
             }
             opDone = true;
             finish({ type: 'complete', assetId: mainAssetId, downloaded: transferred });
         }
         catch (err) {
             opDone = true;
-            if (signal?.aborted)
+            if (signal?.aborted) {
                 return finish({ type: 'aborted' });
+            }
             finish({ type: 'error', message: errorMessage(err) });
         }
     };
@@ -103,8 +109,9 @@ function aggregateLayout(sources, assetSizes) {
     let totalBytes = 0;
     for (const source of sources) {
         const size = assetSizes[source];
-        if (!Number.isFinite(size) || size <= 0)
+        if (!Number.isFinite(size) || size <= 0) {
             return null;
+        }
         offsets.push(totalBytes);
         sizes.push(size);
         totalBytes += size;

package/dist-lib/lib/capabilities/embed.js

@@ -7,7 +7,8 @@ export function createEmbedLane() {
             const { embedding } = await sdk.embed({ modelId, text: texts });
             return embedding;
         });
-        tail = job.catch(() => { });
+        tail = job.catch(noop);
         return job;
     };
 }
+function noop() { }

package/dist-lib/lib/capabilities/enhance.d.ts

@@ -7,4 +7,9 @@ export interface EnhanceResult {
     totalTokens?: number;
 }
 export type Enhancer = (prompt: string, signal?: BareAbortSignal) => Promise<EnhanceResult | null>;
-export declare function expandImagePrompt(sdk: CompletionSdk, modelId: string, model: string, prompt: string, signal?: BareAbortSignal): Promise<EnhanceResult | null>;
+export interface ExpandImagePromptOptions {
+    modelId: string;
+    model: string;
+    signal?: BareAbortSignal | undefined;
+}
+export declare function expandImagePrompt(sdk: CompletionSdk, prompt: string, { modelId, model, signal }: ExpandImagePromptOptions): Promise<EnhanceResult | null>;

package/dist-lib/lib/capabilities/enhance.js

@@ -3,7 +3,7 @@ import { raceAbort } from '../helpers/race-abort.js';
 const SYSTEM = "Rewrite the user's description as a single vivid prompt for an image generator. " +
     'Answer with the prompt alone: no preamble, no quotes, no explanation.';
 const MAX_PROMPT_CHARS = 400;
-export async function expandImagePrompt(sdk, modelId, model, prompt, signal) {
+export async function expandImagePrompt(sdk, prompt, { modelId, model, signal }) {
     const run = sdk.completion({
         modelId,
         history: [
@@ -18,12 +18,15 @@ export async function expandImagePrompt(sdk, modelId, model, prompt, signal) {
         final = await raceAbort(run.final, signal);
     }
     catch (err) {
-        if (run.requestId)
-            void sdk.cancel?.({ requestId: run.requestId }).catch(() => { });
+        if (run.requestId) {
+            void sdk.cancel?.({ requestId: run.requestId }).catch(noop);
+        }
         throw err;
     }
     const text = cleanAnswer(final.contentText).slice(0, MAX_PROMPT_CHARS);
-    if (!text)
+    if (!text) {
         return null;
+    }
     return { text, model, totalTokens: final.stats?.generatedTokens };
 }
+function noop() { }

package/dist-lib/lib/capabilities/invoke.js

@@ -1,9 +1,10 @@
+import { errorMessage } from '../guards.js';
 import { AbortedError } from '../helpers/race-abort.js';
 const CAPABILITY_TOOLS = {
     image: { tool: 'generate_image', enhances: 'prompt' },
     translate: { tool: 'translate' }
 };
-export const CAPABILITIES = Object.keys(CAPABILITY_TOOLS);
+export const CAPABILITIES = Object.keys(CAPABILITY_TOOLS).filter(isCapability);
 export function isCapability(name) {
     return name in CAPABILITY_TOOLS;
 }
@@ -30,13 +31,15 @@ export async function* invoke({ capability, args, tools, ctx, enhance = 'never',
             }
         }
     }
-    if (ctx.signal?.aborted)
+    if (ctx.signal?.aborted) {
         return yield { type: 'aborted' };
+    }
     yield { type: 'tool-call', name: entry.tool, args: called, effect: tool.effect(called, ctx) };
     try {
         const result = await tool.execute(called, ctx);
-        if (ctx.signal?.aborted)
+        if (ctx.signal?.aborted) {
             return yield { type: 'aborted' };
+        }
         yield { type: 'tool-result', name: entry.tool, result };
     }
     catch (err) {
@@ -44,7 +47,7 @@ export async function* invoke({ capability, args, tools, ctx, enhance = 'never',
             yield { type: 'aborted' };
             return;
         }
-        yield { type: 'error', message: err instanceof Error ? err.message : String(err) };
+        yield { type: 'error', message: errorMessage(err) };
         return;
     }
     yield {

package/dist-lib/lib/capabilities/kv-cache.js

@@ -1,3 +1,4 @@
+import { errorMessage } from '../guards.js';
 export async function deleteAllKvCaches(loadSdk) {
     return deleteCacheWith(loadSdk, { all: true });
 }
@@ -15,7 +16,7 @@ async function deleteCacheWith(loadSdk, params) {
     catch (error) {
         return {
             status: 'failure',
-            error: error instanceof Error ? error.message : String(error)
+            error: errorMessage(error)
         };
     }
 }

package/dist-lib/lib/capabilities/preview-model-source.d.ts

@@ -0,0 +1,69 @@
+import type { HarnessPreviewModelSourceResponse } from '../../spec/hyperschema/types.d.ts';
+export type ModelFitVerdict = 'likely-fits' | 'likely-too-large' | 'unknown';
+export type ModelSourceRefusal = 'invalid-source' | 'not-found' | 'private' | 'gated' | 'web-page' | 'unreachable' | 'not-gguf' | 'not-a-model' | 'no-model-files' | 'invalid-projection';
+export type ModelTaskType = 'llm' | 'embeddings' | 'speech-to-text' | 'text-to-speech' | 'image-generation';
+export interface ModelSourceFile {
+    src: string;
+    label: string;
+    sizeBytes: number;
+}
+export interface ModelSourcePreview {
+    ok: boolean;
+    refusal?: ModelSourceRefusal;
+    reason?: string;
+    src?: string;
+    files?: ModelSourceFile[];
+    sizeBytes?: number;
+    contextLength?: number;
+    architecture?: string;
+    sizeLabel?: string;
+    taskType?: string;
+    license?: string;
+    fit?: ModelFitVerdict;
+    fitReasons?: string[];
+    mmprojSrc?: string;
+}
+export interface PreviewModelSourceContext {
+    src: string;
+    mmprojSrc?: string;
+}
+export interface PreviewResponse {
+    ok: boolean;
+    status: number;
+    headers: {
+        get(name: string): string | null;
+    };
+    json(): Promise<unknown>;
+    arrayBuffer(): Promise<ArrayBuffer>;
+}
+export type PreviewFetch = (url: string, init?: {
+    method?: string;
+    headers?: Record<string, string>;
+    redirect?: 'follow';
+}) => Promise<PreviewResponse>;
+export interface FitQuery {
+    sha256Checksum: string;
+    name: string;
+    contextTokens: number;
+}
+export interface PreviewDeps {
+    fetch?: PreviewFetch;
+    stat?(path: string): Promise<FileStat | null>;
+    assessFit?(query: FitQuery): Promise<FitAssessment>;
+}
+export interface FileStat {
+    size: number;
+}
+export interface FitAssessment {
+    fit: ModelFitVerdict;
+    fitReasons: string[];
+}
+interface HuggingFaceRef {
+    repo: string;
+    revision: string;
+    file?: string;
+}
+export declare function previewModelSource(context: PreviewModelSourceContext, deps?: PreviewDeps): Promise<ModelSourcePreview>;
+export declare function readModelSourcePreview(response: HarnessPreviewModelSourceResponse): ModelSourcePreview;
+export declare function parseHuggingFaceRef(src: string): HuggingFaceRef | null;
+export {};