@modular-prompt/driver 0.14.0 → 0.16.0

package/dist/models-config/resolve.js

@@ -0,0 +1,208 @@
+/**
+ * models.yaml のマージ・解決
+ */
+import { loadModelsConfigFile } from './loader.js';
+import { getModelsConfigPath, getUserModelsConfigPath } from './paths.js';
+function shallowMergeRecords(base, overlay) {
+    if (!base && !overlay) {
+        return undefined;
+    }
+    return { ...(base ?? {}), ...(overlay ?? {}) };
+}
+function mergeModelsSection(base, overlay, mode) {
+    if (!base && !overlay) {
+        return undefined;
+    }
+    if (mode === 'override' && overlay) {
+        return { ...overlay };
+    }
+    return { ...(base ?? {}), ...(overlay ?? {}) };
+}
+/**
+ * 2 つの ModelsConfig を shallow merge する（overlay 優先）
+ */
+export function mergeModelsConfig(base, overlay, mode = 'merge') {
+    const merged = {};
+    merged.drivers = shallowMergeRecords(base?.drivers, overlay?.drivers);
+    merged.defaultOptions = shallowMergeRecords(base?.defaultOptions, overlay?.defaultOptions);
+    const baseModels = base?.models;
+    const overlayModels = overlay?.models;
+    merged.models = mergeModelsSection(baseModels, overlayModels, mode);
+    return merged;
+}
+/**
+ * ユーザーレベルの models.yaml を読み込む（存在しない場合は空）
+ */
+export function loadUserModelsConfig(profile) {
+    const config = loadModelsConfigFile(getUserModelsConfigPath()) ?? {};
+    const resolvedProfile = resolveModelsProfile(profile);
+    if (!resolvedProfile || resolvedProfile === 'default') {
+        return config;
+    }
+    return mergeModelsConfig(config, loadModelsConfigFile(getModelsConfigPath(resolvedProfile)), 'merge');
+}
+/**
+ * 明示 profile、環境変数、実行コンテキストの順に user profile を解決する。
+ * Vitest / NODE_ENV=test では testing profile を自動的に追加する。
+ */
+function resolveModelsProfile(profile) {
+    if (profile !== undefined) {
+        return profile;
+    }
+    if (process.env.MODULAR_PROMPT_MODELS_PROFILE) {
+        return process.env.MODULAR_PROMPT_MODELS_PROFILE;
+    }
+    if (process.env.NODE_ENV === 'test' || Boolean(process.env.VITEST)) {
+        return 'testing';
+    }
+    return undefined;
+}
+/**
+ * base / user / overlay を解決する
+ *
+ * マージ優先（下ほど高）: base → user（source=merge のとき）→ overlay
+ */
+export function resolveModelsConfig(options) {
+    if (!options) {
+        return loadUserModelsConfig();
+    }
+    const source = options.source ?? 'merge';
+    const mode = options.mode ?? 'merge';
+    let config = options.base ? { ...options.base } : {};
+    if (source === 'merge') {
+        config = mergeModelsConfig(config, loadUserModelsConfig(options.profile), mode);
+    }
+    if (options.overlay) {
+        config = mergeModelsConfig(config, options.overlay, mode);
+    }
+    return config;
+}
+/**
+ * ModelsConfig を ApplicationConfig に変換する
+ */
+export function toApplicationConfig(config) {
+    const models = config.models
+        ? Object.values(config.models).map(entryToModelSpec)
+        : undefined;
+    return {
+        models,
+        drivers: config.drivers,
+        defaultOptions: config.defaultOptions,
+    };
+}
+/**
+ * ModelSpecEntry を ModelSpec に変換（runtime は metadata に保持）
+ */
+export function entryToModelSpec(entry) {
+    const metadata = { ...(entry.metadata ?? {}) };
+    if (entry.runtime) {
+        metadata.runtime = entry.runtime;
+    }
+    return {
+        model: entry.model,
+        provider: entry.provider,
+        capabilities: entry.capabilities ?? [],
+        maxInputTokens: entry.maxInputTokens,
+        maxOutputTokens: entry.maxOutputTokens,
+        maxTotalTokens: entry.maxTotalTokens,
+        tokensPerMinute: entry.tokensPerMinute,
+        requestsPerMinute: entry.requestsPerMinute,
+        cost: entry.cost,
+        priority: entry.priority,
+        disabled: entry.disabled,
+        defaultOptions: entry.defaultOptions,
+        driverOptions: entry.driverOptions,
+        metadata: Object.keys(metadata).length > 0 ? metadata : undefined,
+    };
+}
+/**
+ * alias から ModelSpec を解決する
+ */
+export function resolveModelAlias(alias, config) {
+    const entry = config.models?.[alias];
+    if (!entry) {
+        return null;
+    }
+    return entryToModelSpec(entry);
+}
+/**
+ * モデル名を alias または生の model 名として解決する
+ */
+export function resolveModelName(name, config, inferProvider) {
+    const byAlias = resolveModelAlias(name, config);
+    if (byAlias) {
+        return byAlias;
+    }
+    return {
+        model: name,
+        provider: inferProvider(name),
+        capabilities: [],
+    };
+}
+/**
+ * merged models からデフォルト ModelSpec を導出する
+ *
+ * 1. alias `default` があればそれを使用
+ * 2. なければ models の先頭エントリ（best-effort）
+ */
+export function resolveDefaultModelFromConfig(config) {
+    if (config.models?.default) {
+        return entryToModelSpec(config.models.default);
+    }
+    const entries = Object.entries(config.models ?? {});
+    if (entries.length === 0) {
+        return null;
+    }
+    const [, first] = entries[0];
+    return entryToModelSpec(first);
+}
+/**
+ * ModelReferenceInput を ModelSpec に解決する
+ */
+export function resolveModelReference(ref, config) {
+    if (ref.ref) {
+        const spec = resolveModelAlias(ref.ref, config);
+        if (!spec) {
+            return null;
+        }
+        if (ref.provider) {
+            spec.provider = ref.provider;
+        }
+        if (ref.model) {
+            spec.model = ref.model;
+        }
+        return spec;
+    }
+    if (ref.provider && ref.model) {
+        return {
+            model: ref.model,
+            provider: ref.provider,
+            capabilities: [],
+        };
+    }
+    return null;
+}
+/**
+ * inline 設定（experiment YAML 等）を resolved config にマージして ApplicationConfig を構築
+ */
+export function buildApplicationConfig(resolved, inline, inlineMode) {
+    const merged = inline
+        ? mergeModelsConfig(resolved, inline, inlineMode ?? 'merge')
+        : resolved;
+    return toApplicationConfig(merged);
+}
+/**
+ * ModelsConfig の全モデルを DriverRegistry に登録する
+ */
+export function registerModelsFromConfig(registry, config) {
+    if (!config.models) {
+        return;
+    }
+    for (const entry of Object.values(config.models)) {
+        if (entry.disabled) {
+            continue;
+        }
+        registry.registerModel(entryToModelSpec(entry));
+    }
+}
+//# sourceMappingURL=resolve.js.map

package/dist/models-config/resolve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolve.js","sourceRoot":"","sources":["../../src/models-config/resolve.ts"],"names":[],"mappings":"AAAA;;GAEG;AAKH,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAS1E,SAAS,mBAAmB,CAC1B,IAAmB,EACnB,OAAsB;IAEtB,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACtB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAO,CAAC;AACtD,CAAC;AAED,SAAS,kBAAkB,CACzB,IAAgD,EAChD,OAAmD,EACnD,IAAqB;IAErB,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACtB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,IAAI,KAAK,UAAU,IAAI,OAAO,EAAE,CAAC;QACnC,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;IACxB,CAAC;IAED,OAAO,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;AACjD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAC/B,IAAqC,EACrC,OAAwC,EACxC,OAAwB,OAAO;IAE/B,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,MAAM,CAAC,OAAO,GAAG,mBAAmB,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IACtE,MAAM,CAAC,cAAc,GAAG,mBAAmB,CACzC,IAAI,EAAE,cAAc,EACpB,OAAO,EAAE,cAAc,CACxB,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,EAAE,MAAM,CAAC;IAChC,MAAM,aAAa,GAAG,OAAO,EAAE,MAAM,CAAC;IAEtC,MAAM,CAAC,MAAM,GAAG,kBAAkB,CAAC,UAAU,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IAEpE,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAAgB;IACnD,MAAM,MAAM,GAAG,oBAAoB,CAAC,uBAAuB,EAAE,CAAC,IAAI,EAAE,CAAC;IACrE,MAAM,eAAe,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAEtD,IAAI,CAAC,eAAe,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO,iBAAiB,CACtB,MAAM,EACN,oBAAoB,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC,EAC1D,OAAO,CACR,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAS,oBAAoB,CAAC,OAAgB;IAC5C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,CAAC;QAC9C,OAAO,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC;IACnD,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAA6B;IAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,oBAAoB,EAAE,CAAC;IAChC,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC;IACzC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC;IAErC,IAAI,MAAM,GAAiB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAEnE,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;QACvB,MAAM,GAAG,iBAAiB,CACxB,MAAM,EACN,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,EACrC,IAAI,CACL,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5D,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAAoB;IACtD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM;QAC1B,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC;QACpD,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO;QACL,MAAM;QACN,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,cAAc,EAAE,MAAM,CAAC,cAAc;KACtC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAqB;IACpD,MAAM,QAAQ,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,CAAC;IAC/C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAClB,QAAQ,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IACnC,CAAC;IAED,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,QAAQ,EAAE,KAAK,CAAC,QAA0B;QAC1C,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,EAAE;QACtC,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,eAAe,EAAE,KAAK,CAAC,eAAe;QACtC,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,eAAe,EAAE,KAAK,CAAC,eAAe;QACtC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;QAC1C,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KAClE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAC/B,KAAa,EACb,MAAoB;IAEpB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC;IACrC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,IAAY,EACZ,MAAoB,EACpB,aAAgD;IAEhD,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAChD,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO;QACL,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE,aAAa,CAAC,IAAI,CAAC;QAC7B,YAAY,EAAE,EAAE;KACjB,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,6BAA6B,CAC3C,MAAoB;IAEpB,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QAC3B,OAAO,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;IACpD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAC7B,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,GAAwB,EACxB,MAAoB;IAEpB,IAAI,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,iBAAiB,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjB,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC,QAA0B,CAAC;QACjD,CAAC;QACD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;QACzB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;QAC9B,OAAO;YACL,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,QAAQ,EAAE,GAAG,CAAC,QAA0B;YACxC,YAAY,EAAE,EAAE;SACjB,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CACpC,QAAsB,EACtB,MAA8B,EAC9B,UAA4B;IAE5B,MAAM,MAAM,GAAG,MAAM;QACnB,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAsB,EAAE,UAAU,IAAI,OAAO,CAAC;QAC5E,CAAC,CAAC,QAAQ,CAAC;IACb,OAAO,mBAAmB,CAAC,MAAM,CAAC,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CACtC,QAAwB,EACxB,MAAoB;IAEpB,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QACD,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,CAAC;AACH,CAAC"}

package/dist/models-config/types.d.ts

@@ -0,0 +1,62 @@
+/**
+ * ~/.modular-prompt/models.yaml スキーマ型
+ */
+import type { ApplicationConfig } from '../driver-registry/config-based-factory.js';
+import type { DriverCapability, DriverProvider, ModelSpec } from '../driver-registry/types.js';
+/** models セクションのマージモード */
+export type ModelsMergeMode = 'merge' | 'override';
+/**
+ * user models.yaml の読み込み方
+ * - merge: user yaml を読み込み base / overlay とマージ（デフォルト）
+ * - overlay: user yaml を無視し base / overlay のみ使用
+ */
+export type ModelsConfigSource = 'merge' | 'overlay';
+/** 利用側が明示投入する overlay とマージモード */
+export interface ModelsConfigOptions {
+    /**
+     * user yaml の読み込み方（デフォルト: merge）
+     * overlay を指定したとき、user 全域 config を無視する場合は overlay を使う
+     */
+    source?: ModelsConfigSource;
+    /** 追加で読み込む user profile（例: `testing` → models.testing.yaml） */
+    profile?: string;
+    /** アプリ同梱など、user より下のレイヤー */
+    base?: ModelsConfig;
+    /** 利用側 overlay（profile.modelsConfig 等） */
+    overlay?: ModelsConfig;
+    /** models セクションのマージモード（デフォルト: merge） */
+    mode?: ModelsMergeMode;
+}
+/** YAML 上のモデルエントリ（alias は親 Record のキー） */
+export interface ModelSpecEntry {
+    model: string;
+    provider: DriverProvider | string;
+    capabilities?: DriverCapability[];
+    runtime?: string;
+    maxInputTokens?: number;
+    maxOutputTokens?: number;
+    maxTotalTokens?: number;
+    tokensPerMinute?: number;
+    requestsPerMinute?: number;
+    cost?: ModelSpec['cost'];
+    priority?: number;
+    disabled?: boolean;
+    defaultOptions?: ModelSpec['defaultOptions'];
+    driverOptions?: ModelSpec['driverOptions'];
+    metadata?: Record<string, unknown>;
+}
+/** models.yaml のルート構造 */
+export interface ModelsConfig {
+    /** alias → ModelSpec エントリ */
+    models?: Record<string, ModelSpecEntry>;
+    drivers?: ApplicationConfig['drivers'];
+    defaultOptions?: ApplicationConfig['defaultOptions'];
+}
+/** workflow / inline からのモデル参照 */
+export interface ModelReferenceInput {
+    /** models.yaml の alias */
+    ref?: string;
+    provider?: string;
+    model?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/models-config/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/models-config/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4CAA4C,CAAC;AACpF,OAAO,KAAK,EAAE,gBAAgB,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAE/F,0BAA0B;AAC1B,MAAM,MAAM,eAAe,GAAG,OAAO,GAAG,UAAU,CAAC;AAEnD;;;;GAIG;AACH,MAAM,MAAM,kBAAkB,GAAG,OAAO,GAAG,SAAS,CAAC;AAErD,iCAAiC;AACjC,MAAM,WAAW,mBAAmB;IAClC;;;OAGG;IACH,MAAM,CAAC,EAAE,kBAAkB,CAAC;IAC5B,+DAA+D;IAC/D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,4BAA4B;IAC5B,IAAI,CAAC,EAAE,YAAY,CAAC;IACpB,0CAA0C;IAC1C,OAAO,CAAC,EAAE,YAAY,CAAC;IACvB,wCAAwC;IACxC,IAAI,CAAC,EAAE,eAAe,CAAC;CACxB;AAED,0CAA0C;AAC1C,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,cAAc,GAAG,MAAM,CAAC;IAClC,YAAY,CAAC,EAAE,gBAAgB,EAAE,CAAC;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,IAAI,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,cAAc,CAAC,EAAE,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC7C,aAAa,CAAC,EAAE,SAAS,CAAC,eAAe,CAAC,CAAC;IAC3C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,yBAAyB;AACzB,MAAM,WAAW,YAAY;IAC3B,6BAA6B;IAC7B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IACxC,OAAO,CAAC,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACvC,cAAc,CAAC,EAAE,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;CACtD;AAED,iCAAiC;AACjC,MAAM,WAAW,mBAAmB;IAClC,0BAA0B;IAC1B,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB"}

package/dist/models-config/types.js

@@ -0,0 +1,5 @@
+/**
+ * ~/.modular-prompt/models.yaml スキーマ型
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/models-config/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/models-config/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/pytorch/process/index.d.ts

@@ -0,0 +1,35 @@
+/**
+ * PyTorch Driver 外部インターフェース
+ */
+import { Readable } from 'stream';
+import type { InferenceCapabilities, InferenceFormatTestResult, InferenceMessage, InferenceRenderResult, InferenceTokenizeResult, InferenceToolDefinition } from '../../local-inference/protocol.js';
+export interface PyTorchProcessOptions {
+    /** デフォルトの ~/.modular-prompt/runtimes/pytorch/.venv を上書き */
+    venvPath?: string;
+    /** Python 子プロセスへ渡す PYTORCH_DEVICE（例: cpu, cuda） */
+    device?: string;
+}
+export declare class PyTorchProcess {
+    readonly modelName: string;
+    private readonly client;
+    constructor(modelName: string, options?: PyTorchProcessOptions);
+    ensureInitialized(): Promise<void>;
+    getCapabilities(): Promise<InferenceCapabilities>;
+    formatTest(messages: InferenceMessage[], options?: {
+        primer?: string;
+    }): Promise<InferenceFormatTestResult>;
+    render(messages: InferenceMessage[], options?: Record<string, unknown> & {
+        primer?: string;
+    }, tools?: InferenceToolDefinition[], reasoningEffort?: 'low' | 'medium' | 'high'): Promise<InferenceRenderResult>;
+    tokenize(messages: InferenceMessage[], tools?: InferenceToolDefinition[], reasoningEffort?: 'low' | 'medium' | 'high'): Promise<InferenceTokenizeResult>;
+    generate(prompt: string | number[], options?: Record<string, unknown>, images?: string[], maxImageSize?: number): Promise<Readable>;
+    exit(): Promise<void>;
+    cancelActiveRequest(): void;
+    getStatus(): {
+        modelName: string;
+        queueLength: number;
+        isStreamingActive: boolean;
+        isJsonBuffering: boolean;
+    };
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/pytorch/process/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/pytorch/process/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AASlC,OAAO,KAAK,EACV,qBAAqB,EACrB,yBAAyB,EACzB,gBAAgB,EAChB,qBAAqB,EACrB,uBAAuB,EACvB,uBAAuB,EACxB,MAAM,mCAAmC,CAAC;AAO3C,MAAM,WAAW,qBAAqB;IACpC,2DAA2D;IAC3D,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,mDAAmD;IACnD,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAcD,qBAAa,cAAc;IACzB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAyB;gBAEpC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,qBAAqB;IA2BxD,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIlC,eAAe,IAAI,OAAO,CAAC,qBAAqB,CAAC;IAIjD,UAAU,CACd,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAC5B,OAAO,CAAC,yBAAyB,CAAC;IAI/B,MAAM,CACV,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG;QAAE,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,EACvD,KAAK,CAAC,EAAE,uBAAuB,EAAE,EACjC,eAAe,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,GAC1C,OAAO,CAAC,qBAAqB,CAAC;IAI3B,QAAQ,CACZ,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,KAAK,CAAC,EAAE,uBAAuB,EAAE,EACjC,eAAe,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,GAC1C,OAAO,CAAC,uBAAuB,CAAC;IAI7B,QAAQ,CACZ,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,EACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,MAAM,CAAC,EAAE,MAAM,EAAE,EACjB,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,QAAQ,CAAC;IAId,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,mBAAmB,IAAI,IAAI;IAI3B,SAAS;;;;;;CAGV"}

package/dist/pytorch/process/index.js

@@ -0,0 +1,69 @@
+/**
+ * PyTorch Driver 外部インターフェース
+ */
+import { existsSync } from 'fs';
+import { getPytorchPythonDir, getVenvPath, resolvePackageRootFromProcessModule, SETUP_PYTORCH_MONOREPO, } from '../../runtime/index.js';
+import { InferenceProcessClient } from '../../local-inference/process-client.js';
+import { mapOptionsToPython } from '../../mlx-ml/process/parameter-mapper.js';
+const packageRoot = resolvePackageRootFromProcessModule(import.meta.url);
+const pytorchPythonDir = getPytorchPythonDir(packageRoot);
+function resolveVenvPath(options) {
+    return (options?.venvPath ??
+        process.env.MODULAR_PROMPT_PYTORCH_VENV ??
+        getVenvPath('pytorch'));
+}
+function usesDefaultRuntimeVenv(venvPath) {
+    return venvPath === getVenvPath('pytorch');
+}
+export class PyTorchProcess {
+    modelName;
+    client;
+    constructor(modelName, options) {
+        this.modelName = modelName;
+        const venvPath = resolveVenvPath(options);
+        if (!existsSync(venvPath)) {
+            throw new Error(`PyTorch venv not found at ${venvPath}. ` +
+                `Run: ${SETUP_PYTORCH_MONOREPO}`);
+        }
+        const device = options?.device ?? process.env.PYTORCH_DEVICE;
+        const extraEnv = device ? { PYTORCH_DEVICE: device } : undefined;
+        this.client = new InferenceProcessClient({
+            modelName,
+            pythonProjectDir: pytorchPythonDir,
+            venvPath,
+            runtimeProfile: usesDefaultRuntimeVenv(venvPath) ? 'pytorch' : undefined,
+            extraEnv,
+            loggerPrefix: 'PyTorch',
+            mapSamplingOptions: (opts) => mapOptionsToPython(opts, false),
+            processExitErrorMessage: (code, signal) => `PyTorch process exited unexpectedly (code=${code}, signal=${signal})`,
+        });
+    }
+    async ensureInitialized() {
+        return this.client.ensureInitialized();
+    }
+    async getCapabilities() {
+        return this.client.getCapabilities();
+    }
+    async formatTest(messages, options) {
+        return this.client.formatTest(messages, options);
+    }
+    async render(messages, options, tools, reasoningEffort) {
+        return this.client.render(messages, options, tools, reasoningEffort);
+    }
+    async tokenize(messages, tools, reasoningEffort) {
+        return this.client.tokenize(messages, tools, reasoningEffort);
+    }
+    async generate(prompt, options, images, maxImageSize) {
+        return this.client.generate(prompt, options, images, maxImageSize);
+    }
+    async exit() {
+        return this.client.exit();
+    }
+    cancelActiveRequest() {
+        this.client.cancelActiveRequest();
+    }
+    getStatus() {
+        return this.client.getStatus();
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/pytorch/process/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/pytorch/process/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAChC,OAAO,EACL,mBAAmB,EACnB,WAAW,EACX,mCAAmC,EACnC,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AASjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,0CAA0C,CAAC;AAG9E,MAAM,WAAW,GAAG,mCAAmC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzE,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;AAS1D,SAAS,eAAe,CAAC,OAA+B;IACtD,OAAO,CACL,OAAO,EAAE,QAAQ;QACjB,OAAO,CAAC,GAAG,CAAC,2BAA2B;QACvC,WAAW,CAAC,SAAS,CAAC,CACvB,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAgB;IAC9C,OAAO,QAAQ,KAAK,WAAW,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,OAAO,cAAc;IAChB,SAAS,CAAS;IACV,MAAM,CAAyB;IAEhD,YAAY,SAAiB,EAAE,OAA+B;QAC5D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,MAAM,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QAC1C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CACb,6BAA6B,QAAQ,IAAI;gBACvC,QAAQ,sBAAsB,EAAE,CACnC,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAC7D,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAEjE,IAAI,CAAC,MAAM,GAAG,IAAI,sBAAsB,CAAC;YACvC,SAAS;YACT,gBAAgB,EAAE,gBAAgB;YAClC,QAAQ;YACR,cAAc,EAAE,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACxE,QAAQ;YACR,YAAY,EAAE,SAAS;YACvB,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,kBAAkB,CAAC,IAAuC,EAAE,KAAK,CAAC;YAChG,uBAAuB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CACxC,6CAA6C,IAAI,YAAY,MAAM,GAAG;SACzE,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,OAAO,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,UAAU,CACd,QAA4B,EAC5B,OAA6B;QAE7B,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,MAAM,CACV,QAA4B,EAC5B,OAAuD,EACvD,KAAiC,EACjC,eAA2C;QAE3C,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAA4B,EAC5B,KAAiC,EACjC,eAA2C;QAE3C,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,MAAyB,EACzB,OAAiC,EACjC,MAAiB,EACjB,YAAqB;QAErB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAC5B,CAAC;IAED,mBAAmB;QACjB,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;IACpC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC;CACF"}

package/dist/pytorch/pytorch-driver.d.ts

@@ -0,0 +1,35 @@
+import { LocalInferenceDriver } from '../local-inference/driver.js';
+import type { InferenceCapabilities } from '../local-inference/protocol.js';
+import type { FormatterOptions } from '../formatter/types.js';
+import type { PyTorchQueryOptions } from './pytorch-options.js';
+export interface PyTorchModelCapabilities {
+    methods: InferenceCapabilities['methods'];
+    specialTokens: InferenceCapabilities['special_tokens'];
+    features: {
+        hasChatTemplate: boolean;
+        vocabSize?: number;
+        modelMaxLength?: number;
+        chatTemplate?: InferenceCapabilities['features']['chat_template'];
+    };
+    chatRestrictions?: InferenceCapabilities['chat_restrictions'];
+}
+export interface PyTorchDriverConfig {
+    model: string;
+    defaultOptions?: Partial<PyTorchQueryOptions>;
+    formatterOptions?: FormatterOptions;
+    /** 外部 venv パス（未指定時は ~/.modular-prompt/runtimes/pytorch/.venv） */
+    venvPath?: string;
+    /** PYTORCH_DEVICE（例: cpu, cuda）。未指定時は Python 側で cpu */
+    device?: string;
+}
+/**
+ * Transformers + PyTorch バックエンド（LIP）。
+ * 共通ロジックは LocalInferenceDriver に委譲する。
+ */
+export declare class PyTorchDriver extends LocalInferenceDriver {
+    constructor(config: PyTorchDriverConfig);
+    get defaultOptions(): Partial<PyTorchQueryOptions>;
+    set defaultOptions(value: Partial<PyTorchQueryOptions>);
+    getCapabilities(): Promise<PyTorchModelCapabilities>;
+}
+//# sourceMappingURL=pytorch-driver.d.ts.map

package/dist/pytorch/pytorch-driver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pytorch-driver.d.ts","sourceRoot":"","sources":["../../src/pytorch/pytorch-driver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAC5E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAIhE,MAAM,WAAW,wBAAwB;IACvC,OAAO,EAAE,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAC1C,aAAa,EAAE,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;IACvD,QAAQ,EAAE;QACR,eAAe,EAAE,OAAO,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,YAAY,CAAC,EAAE,qBAAqB,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,CAAC;KACnE,CAAC;IACF,gBAAgB,CAAC,EAAE,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;CAC/D;AAED,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC9C,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC,iEAAiE;IACjE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,uDAAuD;IACvD,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,qBAAa,aAAc,SAAQ,oBAAoB;gBACzC,MAAM,EAAE,mBAAmB;IAgBvC,IAAI,cAAc,IAAI,OAAO,CAAC,mBAAmB,CAAC,CAEjD;IAED,IAAI,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,mBAAmB,CAAC,EAErD;IAEK,eAAe,IAAI,OAAO,CAAC,wBAAwB,CAAC;CAoB3D"}

package/dist/pytorch/pytorch-driver.js

@@ -0,0 +1,48 @@
+import { LocalInferenceDriver } from '../local-inference/driver.js';
+import { PyTorchProcess } from './process/index.js';
+import { pytorchLocalInferenceAdapters } from './pytorch-local-inference-adapters.js';
+/**
+ * Transformers + PyTorch バックエンド（LIP）。
+ * 共通ロジックは LocalInferenceDriver に委譲する。
+ */
+export class PyTorchDriver extends LocalInferenceDriver {
+    constructor(config) {
+        const process = new PyTorchProcess(config.model, {
+            venvPath: config.venvPath,
+            device: config.device,
+        });
+        super({
+            model: config.model,
+            process,
+            adapters: pytorchLocalInferenceAdapters,
+            formatterOptions: config.formatterOptions,
+            defaultOptions: config.defaultOptions,
+            loggerPrefix: 'PyTorch',
+        });
+    }
+    get defaultOptions() {
+        return super.defaultOptions;
+    }
+    set defaultOptions(value) {
+        super.defaultOptions = value ?? {};
+    }
+    async getCapabilities() {
+        await this.ensureInitialized();
+        const runtimeInfo = this.getRuntimeInfo();
+        if (!runtimeInfo) {
+            throw new Error('Failed to retrieve model capabilities');
+        }
+        return {
+            methods: runtimeInfo.methods,
+            specialTokens: runtimeInfo.special_tokens,
+            features: {
+                hasChatTemplate: runtimeInfo.features.apply_chat_template,
+                vocabSize: runtimeInfo.features.vocab_size,
+                modelMaxLength: runtimeInfo.features.model_max_length,
+                chatTemplate: runtimeInfo.features.chat_template,
+            },
+            chatRestrictions: runtimeInfo.chat_restrictions,
+        };
+    }
+}
+//# sourceMappingURL=pytorch-driver.js.map

package/dist/pytorch/pytorch-driver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pytorch-driver.js","sourceRoot":"","sources":["../../src/pytorch/pytorch-driver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AAIpE,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AAwBtF;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,oBAAoB;IACrD,YAAY,MAA2B;QACrC,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,KAAK,EAAE;YAC/C,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,KAAK,CAAC;YACJ,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO;YACP,QAAQ,EAAE,6BAA6B;YACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,YAAY,EAAE,SAAS;SACxB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,KAAK,CAAC,cAA8C,CAAC;IAC9D,CAAC;IAED,IAAI,cAAc,CAAC,KAAmC;QACpD,KAAK,CAAC,cAAc,GAAG,KAAK,IAAI,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1C,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,OAAO;YAC5B,aAAa,EAAE,WAAW,CAAC,cAAc;YACzC,QAAQ,EAAE;gBACR,eAAe,EAAE,WAAW,CAAC,QAAQ,CAAC,mBAAmB;gBACzD,SAAS,EAAE,WAAW,CAAC,QAAQ,CAAC,UAAU;gBAC1C,cAAc,EAAE,WAAW,CAAC,QAAQ,CAAC,gBAAgB;gBACrD,YAAY,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa;aACjD;YACD,gBAAgB,EAAE,WAAW,CAAC,iBAAiB;SAChD,CAAC;IACJ,CAAC;CACF"}

package/dist/pytorch/pytorch-local-inference-adapters.d.ts

@@ -0,0 +1,3 @@
+import type { LocalInferenceAdapters } from '../local-inference/adapters.js';
+export declare const pytorchLocalInferenceAdapters: LocalInferenceAdapters;
+//# sourceMappingURL=pytorch-local-inference-adapters.d.ts.map

package/dist/pytorch/pytorch-local-inference-adapters.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pytorch-local-inference-adapters.d.ts","sourceRoot":"","sources":["../../src/pytorch/pytorch-local-inference-adapters.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AAE7E,eAAO,MAAM,6BAA6B,EAAE,sBAa3C,CAAC"}

package/dist/pytorch/pytorch-local-inference-adapters.js

@@ -0,0 +1,19 @@
+import { mergePyTorchQueryOptions, toPyTorchSamplingOptions } from './pytorch-options.js';
+import { createModelSpecificProcessor, selectApi } from '../mlx-ml/process/model-specific.js';
+import { selectResponseProcessor } from '../mlx-ml/process/model-handlers.js';
+import { generateMergedPrompt } from '../mlx-ml/process/prompt-builder.js';
+import { formatToolDefinitionsAsText } from '../mlx-ml/tool-call-parser/index.js';
+import { convertMessages, convertToolDefinitions, extractImagePaths, } from '../mlx-ml/mlx-message-utils.js';
+export const pytorchLocalInferenceAdapters = {
+    mergeQueryOptions: (defaults, options) => mergePyTorchQueryOptions(defaults, options),
+    toSamplingOptions: (merged) => toPyTorchSamplingOptions(merged),
+    createModelProcessor: (model) => createModelSpecificProcessor(model),
+    selectResponseProcessor: (model, runtimeInfo, opts) => selectResponseProcessor(model, runtimeInfo, opts),
+    convertToolDefinitions,
+    convertMessages,
+    extractImagePaths,
+    formatToolDefinitionsAsText,
+    generateMergedPrompt,
+    selectApi,
+};
+//# sourceMappingURL=pytorch-local-inference-adapters.js.map

package/dist/pytorch/pytorch-local-inference-adapters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pytorch-local-inference-adapters.js","sourceRoot":"","sources":["../../src/pytorch/pytorch-local-inference-adapters.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AAC1F,OAAO,EAAE,4BAA4B,EAAE,SAAS,EAAE,MAAM,qCAAqC,CAAC;AAC9F,OAAO,EAAE,uBAAuB,EAAE,MAAM,qCAAqC,CAAC;AAC9E,OAAO,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAC3E,OAAO,EAAE,2BAA2B,EAAE,MAAM,qCAAqC,CAAC;AAClF,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,gCAAgC,CAAC;AAGxC,MAAM,CAAC,MAAM,6BAA6B,GAA2B;IACnE,iBAAiB,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,CACvC,wBAAwB,CAAC,QAAQ,EAAE,OAAO,CAA4B;IACxE,iBAAiB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC;IAC/D,oBAAoB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,4BAA4B,CAAC,KAAK,CAAC;IACpE,uBAAuB,EAAE,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,CACpD,uBAAuB,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC;IACnD,sBAAsB;IACtB,eAAe;IACf,iBAAiB;IACjB,2BAA2B;IAC3B,oBAAoB;IACpB,SAAS;CACV,CAAC"}

package/dist/pytorch/pytorch-options.d.ts

@@ -0,0 +1,8 @@
+import type { QueryOptions } from '../types.js';
+/** PyTorch ドライバーのクエリオプション（MLX と同型のサンプリング） */
+export type PyTorchQueryOptions = QueryOptions & {
+    trustRemoteCode?: boolean;
+};
+export declare function mergePyTorchQueryOptions(defaults?: Partial<PyTorchQueryOptions>, overrides?: QueryOptions): PyTorchQueryOptions;
+export declare function toPyTorchSamplingOptions(merged: Partial<PyTorchQueryOptions>): Record<string, unknown>;
+//# sourceMappingURL=pytorch-options.d.ts.map

package/dist/pytorch/pytorch-options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pytorch-options.d.ts","sourceRoot":"","sources":["../../src/pytorch/pytorch-options.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEhD,6CAA6C;AAC7C,MAAM,MAAM,mBAAmB,GAAG,YAAY,GAAG;IAC/C,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B,CAAC;AAUF,wBAAgB,wBAAwB,CACtC,QAAQ,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,EACvC,SAAS,CAAC,EAAE,YAAY,GACvB,mBAAmB,CAErB;AAED,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,OAAO,CAAC,mBAAmB,CAAC,GACnC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CASzB"}

package/dist/pytorch/pytorch-options.js

@@ -0,0 +1,21 @@
+const PYTORCH_SAMPLING_KEYS = [
+    'maxTokens',
+    'temperature',
+    'topP',
+    'topK',
+    'trustRemoteCode',
+];
+export function mergePyTorchQueryOptions(defaults, overrides) {
+    return { ...defaults, ...overrides };
+}
+export function toPyTorchSamplingOptions(merged) {
+    const result = {};
+    for (const key of PYTORCH_SAMPLING_KEYS) {
+        const value = merged[key];
+        if (value !== undefined) {
+            result[key] = value;
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=pytorch-options.js.map

package/dist/pytorch/pytorch-options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pytorch-options.js","sourceRoot":"","sources":["../../src/pytorch/pytorch-options.ts"],"names":[],"mappings":"AAOA,MAAM,qBAAqB,GAAG;IAC5B,WAAW;IACX,aAAa;IACb,MAAM;IACN,MAAM;IACN,iBAAiB;CACT,CAAC;AAEX,MAAM,UAAU,wBAAwB,CACtC,QAAuC,EACvC,SAAwB;IAExB,OAAO,EAAE,GAAG,QAAQ,EAAE,GAAG,SAAS,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,MAAoC;IAEpC,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,KAAK,MAAM,GAAG,IAAI,qBAAqB,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/query-logger.js

@@ -55,7 +55,7 @@ export class QueryLogger {
             since: this.startTime,
             filterByContext: true,
         });
-        const errors = logEntries.filter(e => e.level === 'error');
+        const errors = logEntries.filter((e) => e.level === 'error');
         return {
             logEntries: logEntries.length > 0 ? logEntries : undefined,
             errors: errors.length > 0 ? errors : undefined,

package/dist/query-logger.js.map

@@ -1 +1 @@
-{"version":3,"file":"query-logger.js","sourceRoot":"","sources":["../src/query-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAiB,MAAM,uBAAuB,CAAC;AAE9D;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,WAAW;IACL,MAAM,CAAS;IACxB,SAAS,CAAO;IAExB,YAAY,MAAc,EAAE,UAAkB,QAAQ;QACpD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM;YACN,OAAO;YACP,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;IAC9B,CAAC;IAED,yCAAyC;IACzC,IAAI,CAAC,OAAiC;QACpC,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC5B,IAAI,OAAO,EAAE,CAAC;YACZ,sBAAsB;YACtB,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;YACnC,MAAM,OAAO,GAA4B,EAAE,GAAG,IAAI,EAAE,CAAC;YACrD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7C,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,GAAG,CAAC,CAAC;YAC3E,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,0BAA0B;IAC1B,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,mDAAmD;IACnD,OAAO;QACL,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YAC3C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,CAAC;QAE3D,OAAO;YACL,UAAU,EAAE,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;YAC1D,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;SAC/C,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"query-logger.js","sourceRoot":"","sources":["../src/query-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAiB,MAAM,uBAAuB,CAAC;AAE9D;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,WAAW;IACL,MAAM,CAAS;IACxB,SAAS,CAAO;IAExB,YAAY,MAAc,EAAE,UAAkB,QAAQ;QACpD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM;YACN,OAAO;YACP,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;IAC9B,CAAC;IAED,yCAAyC;IACzC,IAAI,CAAC,OAAiC;QACpC,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC5B,IAAI,OAAO,EAAE,CAAC;YACZ,sBAAsB;YACtB,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;YACnC,MAAM,OAAO,GAA4B,EAAE,GAAG,IAAI,EAAE,CAAC;YACrD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7C,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,GAAG,CAAC,CAAC;YAC3E,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,0BAA0B;IAC1B,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,mDAAmD;IACnD,OAAO;QACL,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YAC3C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,CAAC;QAEvE,OAAO;YACL,UAAU,EAAE,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;YAC1D,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;SAC/C,CAAC;IACJ,CAAC;CACF"}

package/dist/runtime/check.d.ts

@@ -0,0 +1,10 @@
+import type { RuntimeProfile } from './paths.js';
+import { isRuntimeReady } from './paths.js';
+export declare class RuntimeNotReadyError extends Error {
+    readonly profile: RuntimeProfile;
+    readonly setupCommand: string;
+    constructor(profile: RuntimeProfile);
+}
+export { isRuntimeReady };
+export declare function assertRuntimeReady(profile: RuntimeProfile): void;
+//# sourceMappingURL=check.d.ts.map

package/dist/runtime/check.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"check.d.ts","sourceRoot":"","sources":["../../src/runtime/check.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAe,cAAc,EAAE,MAAM,YAAY,CAAC;AAOzD,qBAAa,oBAAqB,SAAQ,KAAK;IAC7C,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC;IACjC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;gBAElB,OAAO,EAAE,cAAc;CAepC;AAED,OAAO,EAAE,cAAc,EAAE,CAAC;AAE1B,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI,CAIhE"}

package/dist/runtime/check.js

@@ -0,0 +1,25 @@
+import { getVenvPath, isRuntimeReady } from './paths.js';
+import { SETUP_MLX_MONOREPO, SETUP_PYTORCH_MONOREPO, SETUP_MLX_CLI, } from './setup-commands.js';
+export class RuntimeNotReadyError extends Error {
+    profile;
+    setupCommand;
+    constructor(profile) {
+        const setupCommand = profile === 'mlx'
+            ? SETUP_MLX_MONOREPO
+            : profile === 'pytorch'
+                ? SETUP_PYTORCH_MONOREPO
+                : SETUP_MLX_CLI.replace(' setup mlx', ` setup ${profile}`);
+        super(`${profile} runtime is not set up at ${getVenvPath(profile)}. ` +
+            `Run: ${setupCommand}`);
+        this.name = 'RuntimeNotReadyError';
+        this.profile = profile;
+        this.setupCommand = setupCommand;
+    }
+}
+export { isRuntimeReady };
+export function assertRuntimeReady(profile) {
+    if (!isRuntimeReady(profile)) {
+        throw new RuntimeNotReadyError(profile);
+    }
+}
+//# sourceMappingURL=check.js.map