@modular-prompt/driver 0.14.0 → 0.16.0

package/dist/mlx-ml/mlx-driver.js

@@ -1,434 +1,57 @@
-import { isToolResult } from '../types.js';
-import { formatPromptAsMessages } from '../formatter/converter.js';
-import { formatCompletionPrompt } from '../formatter/completion-formatter.js';
+import { LocalInferenceDriver } from '../local-inference/driver.js';
+import { hasMessageElement } from '../local-inference/prompt-utils.js';
 import { MlxProcess } from './process/index.js';
-import { createModelSpecificProcessor, selectApi } from './process/model-specific.js';
-import { selectResponseProcessor } from './process/model-handlers.js';
-import { extractJSON } from '@modular-prompt/utils';
-import { formatToolDefinitionsAsText } from './tool-call-parser/index.js';
-import { convertMessages, convertToolDefinitions, extractImagePaths } from './mlx-message-utils.js';
-import { QueryLogger } from '../query-logger.js';
-import { extractCacheablePrefix } from '../cache-utils.js';
+import { mlxLocalInferenceAdapters } from './mlx-local-inference-adapters.js';
+import { bindMlxCacheOnCapabilitiesLoaded, createMlxCacheSupport, } from './mlx-cache-support.js';
 import { MlxCacheController } from './mlx-cache-controller.js';
-import { buildQueryUsage, createAbortedStreamResult, isAborted, watchAbortSignal, } from '../query-utils.js';
-// ========================================================================
-// Utility Functions (exported for testing)
-// ========================================================================
+export { hasMessageElement };
 /**
- * Check if the prompt contains MessageElement
+ * MLX ML driver using Python subprocess.
+ * 共通ロジックは LocalInferenceDriver に委譲する。
  */
-export function hasMessageElement(prompt) {
-    const checkElements = (elements) => {
-        if (!elements)
-            return false;
-        return elements.some(element => {
-            const el = element;
-            return el?.type === 'message';
-        });
-    };
-    return checkElements(prompt.instructions) ||
-        checkElements(prompt.data) ||
-        checkElements(prompt.output);
-}
-const META_MARKER = '\x1e__META__:';
-function extractStreamMeta(content) {
-    const idx = content.lastIndexOf(META_MARKER);
-    if (idx === -1)
-        return { content, meta: {} };
-    const jsonStr = content.slice(idx + META_MARKER.length);
-    try {
-        return { content: content.slice(0, idx), meta: JSON.parse(jsonStr) };
-    }
-    catch {
-        return { content: content.slice(0, idx), meta: {} };
-    }
-}
-function createStreamIterable(stream, isAborted) {
-    const chunks = [];
-    let resolveCompletion;
-    let settled = false;
-    const settle = (error) => {
-        if (settled)
-            return;
-        settled = true;
-        const raw = chunks.join('');
-        const { content, meta } = extractStreamMeta(raw);
-        const aborted = isAborted?.() ?? false;
-        resolveCompletion({ content, meta, error: aborted ? null : error });
-    };
-    const completion = new Promise((resolve) => {
-        resolveCompletion = resolve;
-    });
-    const iterable = {
-        async *[Symbol.asyncIterator]() {
-            try {
-                let buffer = '';
-                let markerFound = false;
-                for await (const chunk of stream) {
-                    if (isAborted?.()) {
-                        break;
-                    }
-                    const str = chunk.toString();
-                    chunks.push(str);
-                    if (markerFound)
-                        continue;
-                    buffer += str;
-                    const markerIdx = buffer.indexOf(META_MARKER);
-                    if (markerIdx !== -1) {
-                        const text = buffer.slice(0, markerIdx);
-                        if (text)
-                            yield text;
-                        markerFound = true;
-                    }
-                    else {
-                        const safeLen = buffer.length - (META_MARKER.length - 1);
-                        if (safeLen > 0) {
-                            yield buffer.slice(0, safeLen);
-                            buffer = buffer.slice(safeLen);
-                        }
-                    }
-                }
-                if (!markerFound && buffer)
-                    yield buffer;
-            }
-            catch (error) {
-                settle(error);
-                const aborted = isAborted?.() ?? false;
-                if (!aborted) {
-                    throw error;
-                }
-            }
-            finally {
-                settle(null);
-            }
-        }
-    };
-    return { iterable, completion };
-}
-/**
- * MLX ML driver using Python subprocess
- */
-export class MlxDriver {
-    process;
-    model;
-    _defaultOptions;
-    runtimeInfo = null;
-    modelProcessor;
-    formatterOptions;
-    maxImageSize;
-    queryLogger = new QueryLogger('MLX');
-    cacheController;
-    cacheControllerBound = false;
-    get defaultOptions() {
-        return this._defaultOptions;
-    }
-    set defaultOptions(value) {
-        this._defaultOptions = value;
-    }
+export class MlxDriver extends LocalInferenceDriver {
+    cacheControllerRaw;
+    cacheDisabledForVlm = false;
+    cacheBindingState = { bound: false };
     constructor(config) {
-        this.model = config.model;
-        this._defaultOptions = config.defaultOptions || {};
-        this.formatterOptions = config.formatterOptions || {};
-        this.maxImageSize = config.maxImageSize ?? 768;
-        this.process = new MlxProcess(config.model, {
+        const mlxProcess = new MlxProcess(config.model, {
+            backend: config.backend,
             textOnly: config.textOnly,
             drafterModel: config.drafterModel,
-            draftBlockSize: config.draftBlockSize
+            draftBlockSize: config.draftBlockSize,
         });
-        this.modelProcessor = createModelSpecificProcessor(config.model);
-        this.cacheController = config.cacheController;
+        const cacheSupport = config.cacheController
+            ? createMlxCacheSupport(config.cacheController)
+            : undefined;
+        super({
+            model: config.model,
+            process: mlxProcess,
+            adapters: mlxLocalInferenceAdapters,
+            formatterOptions: config.formatterOptions,
+            maxImageSize: config.maxImageSize,
+            defaultOptions: config.defaultOptions,
+            loggerPrefix: 'MLX',
+            cache: cacheSupport,
+            onCapabilitiesLoaded: async (runtimeInfo, ctx) => {
+                if (!cacheSupport)
+                    return;
+                await bindMlxCacheOnCapabilitiesLoaded(cacheSupport, this.cacheBindingState, runtimeInfo, ctx, () => {
+                    this.queryLogger.log.info('VLM models do not support prompt caching — cacheController disabled');
+                    this.cacheDisabledForVlm = true;
+                    this.disableCacheSupport();
+                });
+            },
+        });
+        this.cacheControllerRaw = config.cacheController;
         if (config.drafterModel) {
             this.queryLogger.log.info(`Drafter model: ${config.drafterModel}`);
         }
     }
-    /**
-     * Initialize process and cache runtime info
-     */
-    async ensureInitialized() {
-        // Ensure process is initialized
-        await this.process.ensureInitialized();
-        // Cache runtime info if not already cached
-        if (!this.runtimeInfo) {
-            try {
-                this.runtimeInfo = await this.process.getCapabilities();
-                // Update formatterOptions with special tokens from runtime info
-                if (this.runtimeInfo.special_tokens) {
-                    this.formatterOptions.specialTokens = this.runtimeInfo.special_tokens;
-                }
-                // Update model processor with runtime context
-                this.modelProcessor.setRuntimeContext({
-                    chatRestrictions: this.runtimeInfo.chat_restrictions,
-                    modelKind: this.runtimeInfo.model_kind,
-                });
-                // Bind cache controller if provided and not yet bound
-                // NOTE: instanceof guard means VLM check only covers MlxCacheController.
-                // A custom PromptCacheController on a VLM model would bypass this — add a
-                // model-kind guard here if another implementation is introduced.
-                if (this.cacheController instanceof MlxCacheController && !this.cacheControllerBound) {
-                    if (this.runtimeInfo.model_kind === 'vlm') {
-                        this.queryLogger.log.info('VLM models do not support prompt caching — cacheController disabled');
-                        this.cacheController = undefined;
-                    }
-                    else {
-                        await this.cacheController.bind(this.process, this.formatterOptions, (msgs) => this.modelProcessor.applyChatSpecificProcessing(msgs));
-                        this.cacheControllerBound = true;
-                    }
-                }
-            }
-            catch (error) {
-                this.queryLogger.log.error('Failed to get MLX runtime info:', error instanceof Error ? error.message : String(error));
-            }
-        }
-    }
-    /**
-     * VLMモデルかどうかを判定
-     */
-    isVLM() {
-        return this.runtimeInfo?.model_kind === 'vlm';
-    }
-    /**
-     * Determine which API to use (chat or completion)
-     * Simple logic based on runtime info only
-     */
-    determineApi(options) {
-        return selectApi(options?.apiStrategy || 'auto', options?.mode, !!this.runtimeInfo?.features.apply_chat_template, this.modelProcessor.hasCompletionProcessor());
-    }
-    /**
-     * モデルがnativeツール対応かを判定
-     * tool_call_format（Python側検出結果）を唯一の判断基準とする
-     */
-    hasNativeToolSupport() {
-        return !!this.runtimeInfo?.features?.chat_template?.tool_call_format?.call_start;
-    }
-    /**
-     * Execute query and return stream
-     * Common logic for query and streamQuery
-     */
-    async executeQuery(prompt, mlxOptions, options) {
-        // APIを選択
-        const api = this.determineApi(options);
-        // tools変換
-        const tools = options?.tools ? convertToolDefinitions(options.tools) : undefined;
-        // completion API または nativeツール非対応の場合、tool定義をテキストとしてプロンプトに注入
-        let augmentedPrompt = prompt;
-        if (options?.tools && options.tools.length > 0 && (api === 'completion' || !this.hasNativeToolSupport())) {
-            const toolsText = formatToolDefinitionsAsText(options.tools, this.runtimeInfo?.special_tokens, this.runtimeInfo?.features?.chat_template?.tool_call_format);
-            augmentedPrompt = {
-                ...prompt,
-                instructions: [
-                    ...prompt.instructions,
-                    { type: 'text', content: toolsText }
-                ]
-            };
-        }
-        // Record all queries, regardless of API mode or cache usage
-        this.cacheController?.recordQuery?.();
-        let stream;
-        if (api === 'completion') {
-            let formattedPrompt = formatCompletionPrompt(augmentedPrompt, this.formatterOptions);
-            formattedPrompt = this.modelProcessor.applyCompletionSpecificProcessing(formattedPrompt);
-            stream = await this.process.completion(formattedPrompt, mlxOptions);
-        }
-        else {
-            const messages = formatPromptAsMessages(augmentedPrompt, this.formatterOptions);
-            const vlm = this.isVLM();
-            let mlxMessages = convertMessages(messages, vlm);
-            mlxMessages = this.modelProcessor.applyChatSpecificProcessing(mlxMessages);
-            const nativeTools = this.hasNativeToolSupport() && tools?.length ? tools : undefined;
-            const images = vlm
-                ? messages.flatMap(m => 'content' in m && !isToolResult(m) ? extractImagePaths(m.content) : [])
-                : [];
-            // Cache: chat APIのみ、以下の条件を全て満たす場合にキャッシュを使用
-            // - options.cache !== false（呼び出し側が明示的に無効化していない）
-            // - trustRemoteCode未指定（明示的なtrue/falseどちらもapply_chat_template kwargsに影響）
-            let cachePath;
-            let cacheTrimTokens;
-            let cacheWriteTokens = 0;
-            const cacheGrowthBefore = this.cacheController instanceof MlxCacheController
-                ? this.cacheController.getStats().cacheGrowthTokens
-                : 0;
-            const trustRemoteCode = mlxOptions.trustRemoteCode;
-            if (this.cacheController && options?.cache !== false && trustRemoteCode === undefined) {
-                const prefix = extractCacheablePrefix(augmentedPrompt);
-                const hasCacheableContent = prefix.instructions.length > 0 ||
-                    prefix.data.length > 0;
-                if (hasCacheableContent) {
-                    const cacheStart = performance.now();
-                    const handle = await this.cacheController.prepare({
-                        model: this.model,
-                        instructions: prefix.instructions,
-                        data: prefix.data,
-                        tools: nativeTools ? options.tools : undefined,
-                        reasoningEffort: options?.reasoningEffort,
-                        readOnly: options?.cache === 'read-only',
-                    });
-                    cachePath = handle.ref || undefined;
-                    cacheTrimTokens = handle.trimTokens;
-                    if (cachePath) {
-                        this.queryLogger.log.debug(`cache prepare ${(performance.now() - cacheStart).toFixed(0)}ms`, `(${prefix.instructions.length}i+${prefix.data.length}d)`, cacheTrimTokens != null ? `trim=${cacheTrimTokens}` : '');
-                    }
-                }
-            }
-            if (this.cacheController instanceof MlxCacheController) {
-                cacheWriteTokens = Math.max(0, this.cacheController.getStats().cacheGrowthTokens - cacheGrowthBefore);
-            }
-            stream = await this.process.chat(mlxMessages, undefined, mlxOptions, nativeTools, images.length > 0 ? images : undefined, images.length > 0 ? this.maxImageSize : undefined, options?.reasoningEffort, cachePath, cacheTrimTokens);
-            const cacheTokensUsed = cachePath
-                ? (cacheTrimTokens ?? (this.cacheController instanceof MlxCacheController
-                    ? this.cacheController.readCacheTokenCount(cachePath) : 0))
-                : 0;
-            return { stream, cacheTokensUsed, cacheWriteTokens };
-        }
-        return { stream, cacheTokensUsed: 0, cacheWriteTokens: 0 };
-    }
-    /**
-     * Query the AI model with a compiled prompt
-     */
-    async query(prompt, options) {
-        // Use streamQuery for consistency with other drivers
-        const { stream, result } = await this.streamQuery(prompt, options);
-        // Consume the stream to trigger completion
-        // This is necessary because the result promise only resolves when the stream is fully consumed
-        // eslint-disable-next-line @typescript-eslint/no-unused-vars
-        for await (const _chunk of stream) {
-            // Just consume the stream, don't need to do anything with the chunks
-        }
-        return result;
+    get defaultOptions() {
+        return super.defaultOptions;
     }
-    /**
-     * Stream query implementation
-     */
-    async streamQuery(prompt, options) {
-        await this.ensureInitialized();
-        const signal = options?.signal;
-        if (isAborted(signal)) {
-            return createAbortedStreamResult(this.queryLogger.collect());
-        }
-        let abortRequested = false;
-        const cleanupAbort = watchAbortSignal(signal, () => {
-            abortRequested = true;
-            this.process.cancelActiveRequest();
-        });
-        const checkAborted = () => abortRequested || isAborted(signal);
-        // Merge options (only override if explicitly provided)
-        const mlxOptions = {
-            ...this.defaultOptions,
-            ...(options?.maxTokens !== undefined && { maxTokens: options.maxTokens }),
-            ...(options?.temperature !== undefined && { temperature: options.temperature }),
-            ...(options?.topP !== undefined && { topP: options.topP }),
-            ...(options?.topK !== undefined && { topK: options.topK }),
-        };
-        this.queryLogger.mark(mlxOptions);
-        const queryStart = performance.now();
-        const { stream, cacheTokensUsed, cacheWriteTokens } = await this.executeQuery(prompt, mlxOptions, options);
-        if (checkAborted()) {
-            this.process.cancelActiveRequest();
-        }
-        const streamStart = performance.now();
-        this.queryLogger.log.debug(`setup ${(streamStart - queryStart).toFixed(0)}ms`);
-        const { iterable, completion } = createStreamIterable(stream, checkAborted);
-        const queryLogger = this.queryLogger;
-        let firstChunkTime = 0;
-        const wrappedIterable = {
-            [Symbol.asyncIterator]() {
-                const inner = iterable[Symbol.asyncIterator]();
-                let firstChunk = true;
-                return {
-                    async next() {
-                        if (checkAborted()) {
-                            void inner.return?.();
-                            return { done: true, value: undefined };
-                        }
-                        const result = await inner.next();
-                        if (!result.done) {
-                            if (firstChunk) {
-                                firstChunk = false;
-                                firstChunkTime = performance.now();
-                                queryLogger.log.debug(`TTFT ${(firstChunkTime - streamStart).toFixed(0)}ms`);
-                            }
-                        }
-                        if (result.done) {
-                            const now = performance.now();
-                            if (firstChunkTime > 0) {
-                                const genMs = now - firstChunkTime;
-                                queryLogger.log.debug(`generation ${genMs.toFixed(0)}ms (query total ${(now - queryStart).toFixed(0)}ms)`);
-                            }
-                            else {
-                                queryLogger.log.debug(`query total ${(performance.now() - queryStart).toFixed(0)}ms`);
-                            }
-                        }
-                        return result;
-                    },
-                    async return(value) {
-                        cleanupAbort();
-                        return inner.return?.(value) ?? { done: true, value: undefined };
-                    },
-                    async throw(e) {
-                        cleanupAbort();
-                        return inner.throw?.(e) ?? { done: true, value: undefined };
-                    },
-                };
-            },
-        };
-        const cacheController = this.cacheController;
-        const resultPromise = completion
-            .then(({ content, meta, error }) => {
-            const aborted = checkAborted();
-            if (error && !aborted) {
-                this.queryLogger.log.error('Stream error:', error.message);
-                throw error;
-            }
-            if (cacheController instanceof MlxCacheController && meta.prompt_tokens != null) {
-                cacheController.recordPromptTokens(meta.prompt_tokens, cacheTokensUsed);
-            }
-            if (meta.generation_tokens != null && firstChunkTime > 0) {
-                const genMs = performance.now() - firstChunkTime;
-                const actualTps = (meta.generation_tokens / genMs * 1000).toFixed(1);
-                this.queryLogger.log.debug(`${meta.generation_tokens} tokens, ${actualTps} tok/s`);
-            }
-            const hasTools = options?.tools && options.tools.length > 0;
-            const responseProcessor = selectResponseProcessor(this.model, this.runtimeInfo, { enableToolParsing: !!hasTools });
-            const parsed = responseProcessor(content);
-            const finalContent = parsed.content;
-            const thinkingContent = parsed.thinkingContent;
-            const toolCalls = parsed.toolCalls;
-            if (thinkingContent) {
-                this.queryLogger.log.verbose('Thinking content:', thinkingContent);
-            }
-            let structuredOutput;
-            if (prompt.metadata?.outputSchema && finalContent) {
-                const extracted = extractJSON(finalContent, { multiple: false });
-                if (extracted.source !== 'none' && extracted.data !== null) {
-                    structuredOutput = extracted.data;
-                }
-            }
-            const finishReason = aborted
-                ? 'error'
-                : toolCalls
-                    ? 'tool_calls'
-                    : 'stop';
-            return {
-                content: finalContent,
-                thinkingContent,
-                structuredOutput,
-                toolCalls,
-                finishReason,
-                usage: buildQueryUsage({
-                    promptTokens: meta.prompt_tokens,
-                    completionTokens: meta.generation_tokens,
-                    cacheReadTokens: cacheTokensUsed,
-                    cacheWriteTokens,
-                }),
-                ...this.queryLogger.collect(),
-            };
-        })
-            .finally(() => {
-            cleanupAbort();
-        });
-        return {
-            stream: wrappedIterable,
-            result: resultPromise,
-        };
+    set defaultOptions(value) {
+        super.defaultOptions = value ?? {};
     }
     /**
      * Get model capabilities (public API)
@@ -437,51 +60,62 @@ export class MlxDriver {
      */
     async getCapabilities() {
         await this.ensureInitialized();
-        if (!this.runtimeInfo) {
+        const runtimeInfo = this.getRuntimeInfo();
+        if (!runtimeInfo) {
             throw new Error('Failed to retrieve model capabilities');
         }
-        // Convert snake_case to camelCase
         return {
-            methods: this.runtimeInfo.methods,
-            specialTokens: this.runtimeInfo.special_tokens,
+            methods: runtimeInfo.methods,
+            specialTokens: runtimeInfo.special_tokens,
             features: {
-                hasChatTemplate: this.runtimeInfo.features.apply_chat_template,
-                vocabSize: this.runtimeInfo.features.vocab_size,
-                modelMaxLength: this.runtimeInfo.features.model_max_length,
-                chatTemplate: this.runtimeInfo.features.chat_template ? {
-                    supportedRoles: this.runtimeInfo.features.chat_template.supported_roles,
-                    preview: this.runtimeInfo.features.chat_template.preview,
-                    constraints: this.runtimeInfo.features.chat_template.constraints,
-                    toolCallFormat: this.runtimeInfo.features.chat_template.tool_call_format ? {
-                        toolParserType: this.runtimeInfo.features.chat_template.tool_call_format.tool_parser_type,
-                        callStart: this.runtimeInfo.features.chat_template.tool_call_format.call_start,
-                        callEnd: this.runtimeInfo.features.chat_template.tool_call_format.call_end,
-                        responseStart: this.runtimeInfo.features.chat_template.tool_call_format.response_start,
-                        responseEnd: this.runtimeInfo.features.chat_template.tool_call_format.response_end,
-                    } : undefined
-                } : undefined
+                hasChatTemplate: runtimeInfo.features.apply_chat_template,
+                vocabSize: runtimeInfo.features.vocab_size,
+                modelMaxLength: runtimeInfo.features.model_max_length,
+                chatTemplate: runtimeInfo.features.chat_template
+                    ? {
+                        supportedRoles: runtimeInfo.features.chat_template.supported_roles,
+                        preview: runtimeInfo.features.chat_template.preview,
+                        constraints: runtimeInfo.features.chat_template.constraints,
+                        toolCallFormat: runtimeInfo.features.chat_template.tool_call_format
+                            ? {
+                                toolParserType: runtimeInfo.features.chat_template.tool_call_format.tool_parser_type,
+                                callStart: runtimeInfo.features.chat_template.tool_call_format.call_start,
+                                callEnd: runtimeInfo.features.chat_template.tool_call_format.call_end,
+                                responseStart: runtimeInfo.features.chat_template.tool_call_format.response_start,
+                                responseEnd: runtimeInfo.features.chat_template.tool_call_format.response_end,
+                            }
+                            : undefined,
+                    }
+                    : undefined,
             },
-            chatRestrictions: this.runtimeInfo.chat_restrictions ? {
-                singleSystemAtStart: this.runtimeInfo.chat_restrictions.single_system_at_start,
-                maxSystemMessages: this.runtimeInfo.chat_restrictions.max_system_messages,
-                alternatingTurns: this.runtimeInfo.chat_restrictions.alternating_turns,
-                requiresUserLast: this.runtimeInfo.chat_restrictions.requires_user_last,
-                allowEmptyMessages: this.runtimeInfo.chat_restrictions.allow_empty_messages,
-            } : undefined
+            chatRestrictions: runtimeInfo.chat_restrictions
+                ? {
+                    singleSystemAtStart: runtimeInfo.chat_restrictions.single_system_at_start,
+                    maxSystemMessages: runtimeInfo.chat_restrictions.max_system_messages,
+                    alternatingTurns: runtimeInfo.chat_restrictions.alternating_turns,
+                    requiresUserLast: runtimeInfo.chat_restrictions.requires_user_last,
+                    allowEmptyMessages: runtimeInfo.chat_restrictions.allow_empty_messages,
+                }
+                : undefined,
         };
     }
+    async close() {
+        this.logCacheStats();
+        if (this.cacheDisabledForVlm) {
+            await this.cacheControllerRaw?.close();
+        }
+        await super.close();
+    }
     logCacheStats() {
-        if (!(this.cacheController instanceof MlxCacheController))
+        if (this.cacheDisabledForVlm)
+            return;
+        if (!(this.cacheControllerRaw instanceof MlxCacheController))
             return;
-        const s = this.cacheController.getStats();
+        const s = this.cacheControllerRaw.getStats();
         if (s.totalQueries === 0)
             return;
-        const queryBreakdown = s.incremental + s.fresh > 0
-            ? ` (incremental ${s.incremental}, fresh ${s.fresh})`
-            : '';
-        const parts = [
-            `cache stats: ${s.totalQueries} queries${queryBreakdown}`,
-        ];
+        const queryBreakdown = s.incremental + s.fresh > 0 ? ` (incremental ${s.incremental}, fresh ${s.fresh})` : '';
+        const parts = [`cache stats: ${s.totalQueries} queries${queryBreakdown}`];
         if (s.totalPromptTokens > 0) {
             const reusedRate = ((s.prefillReusedTokens / s.totalPromptTokens) * 100).toFixed(0);
             parts.push(`prompt ${s.totalPromptTokens} tokens, ${s.prefillReusedTokens} reused (${reusedRate}%)`);
@@ -491,13 +125,5 @@ export class MlxDriver {
         }
         this.queryLogger.log.verbose(parts.join(' | '));
     }
-    /**
-     * Close the process
-     */
-    async close() {
-        this.logCacheStats();
-        await this.cacheController?.close();
-        await this.process.exit();
-    }
 }
 //# sourceMappingURL=mlx-driver.js.map

package/dist/mlx-ml/mlx-driver.js.map

@@ -1 +1 @@
-{"version":3,"file":"mlx-driver.js","sourceRoot":"","sources":["../../src/mlx-ml/mlx-driver.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAGhD,OAAO,EAAE,4BAA4B,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACtF,OAAO,EAAE,uBAAuB,EAAE,MAAM,6BAA6B,CAAC;AAEtE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AACpG,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,eAAe,EACf,yBAAyB,EACzB,SAAS,EACT,gBAAgB,GACjB,MAAM,mBAAmB,CAAC;AAE3B,2EAA2E;AAC3E,2CAA2C;AAC3C,2EAA2E;AAE3E;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAsB;IACtD,MAAM,aAAa,GAAG,CAAC,QAAmB,EAAW,EAAE;QACrD,IAAI,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5B,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;YAC7B,MAAM,EAAE,GAAG,OAA4B,CAAC;YACxC,OAAO,EAAE,EAAE,IAAI,KAAK,SAAS,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC;QACvC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC;QAC1B,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAiCD,MAAM,WAAW,GAAG,eAAe,CAAC;AAEpC,SAAS,iBAAiB,CAAC,OAAe;IACxC,MAAM,GAAG,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAC7C,IAAI,GAAG,KAAK,CAAC,CAAC;QAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IACxD,IAAI,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;IACvE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IACtD,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAC3B,MAAgB,EAChB,SAAyB;IAKzB,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,iBAA8F,CAAC;IACnG,IAAI,OAAO,GAAG,KAAK,CAAC;IAEpB,MAAM,MAAM,GAAG,CAAC,KAAmB,EAAE,EAAE;QACrC,IAAI,OAAO;YAAE,OAAO;QACpB,OAAO,GAAG,IAAI,CAAC;QACf,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5B,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,SAAS,EAAE,EAAE,IAAI,KAAK,CAAC;QACvC,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,OAAO,CAA6D,CAAC,OAAO,EAAE,EAAE;QACrG,iBAAiB,GAAG,OAAO,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG;QACf,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;YAC3B,IAAI,CAAC;gBACH,IAAI,MAAM,GAAG,EAAE,CAAC;gBAChB,IAAI,WAAW,GAAG,KAAK,CAAC;gBACxB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACjC,IAAI,SAAS,EAAE,EAAE,EAAE,CAAC;wBAClB,MAAM;oBACR,CAAC;oBACD,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC7B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACjB,IAAI,WAAW;wBAAE,SAAS;oBAC1B,MAAM,IAAI,GAAG,CAAC;oBACd,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;oBAC9C,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;wBACrB,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;wBACxC,IAAI,IAAI;4BAAE,MAAM,IAAI,CAAC;wBACrB,WAAW,GAAG,IAAI,CAAC;oBACrB,CAAC;yBAAM,CAAC;wBACN,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBACzD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;4BAChB,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;4BAC/B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,WAAW,IAAI,MAAM;oBAAE,MAAM,MAAM,CAAC;YAC3C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAc,CAAC,CAAC;gBACvB,MAAM,OAAO,GAAG,SAAS,EAAE,EAAE,IAAI,KAAK,CAAC;gBACvC,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,MAAM,CAAC,IAAI,CAAC,CAAC;YACf,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,SAAS;IACZ,OAAO,CAAa;IACpB,KAAK,CAAS;IACd,eAAe,CAA6B;IAC5C,WAAW,GAA0B,IAAI,CAAC;IAC1C,cAAc,CAAC;IACf,gBAAgB,CAAmB;IACnC,YAAY,CAAS;IACrB,WAAW,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC,eAAe,CAAyB;IACxC,oBAAoB,GAAG,KAAK,CAAC;IAErC,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,IAAI,cAAc,CAAC,KAAiC;QAClD,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC;IAC/B,CAAC;IAED,YAAY,MAAuB;QACjC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;QACnD,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,GAAG,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE;YAC1C,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,cAAc,EAAE,MAAM,CAAC,cAAc;SACtC,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,4BAA4B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACjE,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC9C,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,kBAAkB,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB;QAC7B,gCAAgC;QAChC,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAEvC,2CAA2C;QAC3C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC;gBACH,IAAI,CAAC,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;gBAExD,gEAAgE;gBAChE,IAAI,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC;oBACpC,IAAI,CAAC,gBAAgB,CAAC,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC;gBACxE,CAAC;gBAED,8CAA8C;gBAC9C,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC;oBACpC,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;oBACpD,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,UAAU;iBACvC,CAAC,CAAC;gBAEH,sDAAsD;gBACtD,yEAAyE;gBACzE,0EAA0E;gBAC1E,iEAAiE;gBACjE,IAAI,IAAI,CAAC,eAAe,YAAY,kBAAkB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBACrF,IAAI,IAAI,CAAC,WAAW,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;wBAC1C,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;wBACjG,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;oBACnC,CAAC;yBAAM,CAAC;wBACN,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7B,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,gBAAgB,EACrB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAChE,CAAC;wBACF,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;oBACnC,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACxH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK;QACX,OAAO,IAAI,CAAC,WAAW,EAAE,UAAU,KAAK,KAAK,CAAC;IAChD,CAAC;IAED;;;OAGG;IACK,YAAY,CAAC,OAAsB;QACzC,OAAO,SAAS,CACd,OAAO,EAAE,WAAW,IAAI,MAAM,EAC9B,OAAO,EAAE,IAAI,EACb,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,mBAAmB,EAChD,IAAI,CAAC,cAAc,CAAC,sBAAsB,EAAE,CAC7C,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,oBAAoB;QAC1B,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,aAAa,EAAE,gBAAgB,EAAE,UAAU,CAAC;IACnF,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,YAAY,CACxB,MAAsB,EACtB,UAA6B,EAC7B,OAAsB;QAEtB,SAAS;QACT,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAEvC,UAAU;QACV,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEjF,4DAA4D;QAC5D,IAAI,eAAe,GAAG,MAAM,CAAC;QAC7B,IAAI,OAAO,EAAE,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,GAAG,KAAK,YAAY,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC,EAAE,CAAC;YACzG,MAAM,SAAS,GAAG,2BAA2B,CAC3C,OAAO,CAAC,KAAK,EACb,IAAI,CAAC,WAAW,EAAE,cAAc,EAChC,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,aAAa,EAAE,gBAAgB,CAC5D,CAAC;YACF,eAAe,GAAG;gBAChB,GAAG,MAAM;gBACT,YAAY,EAAE;oBACZ,GAAG,MAAM,CAAC,YAAY;oBACtB,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,SAAS,EAAE;iBAC9C;aACF,CAAC;QACJ,CAAC;QAED,4DAA4D;QAC5D,IAAI,CAAC,eAAe,EAAE,WAAW,EAAE,EAAE,CAAC;QAEtC,IAAI,MAAgB,CAAC;QACrB,IAAI,GAAG,KAAK,YAAY,EAAE,CAAC;YACzB,IAAI,eAAe,GAAG,sBAAsB,CAAC,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACrF,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,iCAAiC,CAAC,eAAe,CAAC,CAAC;YACzF,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QACtE,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,sBAAsB,CAAC,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChF,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,WAAW,GAAG,eAAe,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YACjD,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,2BAA2B,CAAC,WAAW,CAAC,CAAC;YAC3E,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,EAAE,IAAI,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;YACrF,MAAM,MAAM,GAAG,GAAG;gBAChB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/F,CAAC,CAAC,EAAE,CAAC;YAEP,2CAA2C;YAC3C,gDAAgD;YAChD,wEAAwE;YACxE,IAAI,SAA6B,CAAC;YAClC,IAAI,eAAmC,CAAC;YACxC,IAAI,gBAAgB,GAAG,CAAC,CAAC;YACzB,MAAM,iBAAiB,GAAG,IAAI,CAAC,eAAe,YAAY,kBAAkB;gBAC1E,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC,iBAAiB;gBACnD,CAAC,CAAC,CAAC,CAAC;YACN,MAAM,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;YACnD,IAAI,IAAI,CAAC,eAAe,IAAI,OAAO,EAAE,KAAK,KAAK,KAAK,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBACtF,MAAM,MAAM,GAAG,sBAAsB,CAAC,eAAe,CAAC,CAAC;gBACvD,MAAM,mBAAmB,GACvB,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC;oBAC9B,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAEzB,IAAI,mBAAmB,EAAE,CAAC;oBACxB,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;oBACrC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;wBAChD,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,IAAI,EAAE,MAAM,CAAC,IAAI;wBACjB,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,OAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;wBAC/C,eAAe,EAAE,OAAO,EAAE,eAAe;wBACzC,QAAQ,EAAE,OAAO,EAAE,KAAK,KAAK,WAAW;qBACzC,CAAC,CAAC;oBACH,SAAS,GAAG,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC;oBACpC,eAAe,GAAG,MAAM,CAAC,UAAU,CAAC;oBACpC,IAAI,SAAS,EAAE,CAAC;wBACd,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CACxB,iBAAiB,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAChE,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,KAAK,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EACzD,eAAe,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CACzD,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,IAAI,CAAC,eAAe,YAAY,kBAAkB,EAAE,CAAC;gBACvD,gBAAgB,GAAG,IAAI,CAAC,GAAG,CACzB,CAAC,EACD,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC,iBAAiB,GAAG,iBAAiB,CACtE,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;YAEnO,MAAM,eAAe,GAAG,SAAS;gBAC/B,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,IAAI,CAAC,eAAe,YAAY,kBAAkB;oBACvE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC,CAAC;YACN,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,gBAAgB,EAAE,CAAC;QACvD,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;IAC7D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,MAAsB,EAAE,OAAsB;QACxD,qDAAqD;QACrD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAEnE,2CAA2C;QAC3C,+FAA+F;QAC/F,6DAA6D;QAC7D,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,MAAM,EAAE,CAAC;YAClC,qEAAqE;QACvE,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IACD;;OAEG;IACH,KAAK,CAAC,WAAW,CACf,MAAsB,EACtB,OAAsB;QAEtB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAC/B,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,yBAAyB,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,EAAE,GAAG,EAAE;YACjD,cAAc,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;QACrC,CAAC,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,cAAc,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QAE/D,uDAAuD;QACvD,MAAM,UAAU,GAAsB;YACpC,GAAG,IAAI,CAAC,cAAc;YACtB,GAAG,CAAC,OAAO,EAAE,SAAS,KAAK,SAAS,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;YACzE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;YAC/E,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;YAC1D,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;SAC3D,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAqC,CAAC,CAAC;QAE7D,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACrC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,gBAAgB,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAE3G,IAAI,YAAY,EAAE,EAAE,CAAC;YACnB,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;QACrC,CAAC;QAED,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACtC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,WAAW,GAAG,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAE/E,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAE5E,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACrC,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,MAAM,eAAe,GAA0B;YAC7C,CAAC,MAAM,CAAC,aAAa,CAAC;gBACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC/C,IAAI,UAAU,GAAG,IAAI,CAAC;gBACtB,OAAO;oBACL,KAAK,CAAC,IAAI;wBACR,IAAI,YAAY,EAAE,EAAE,CAAC;4BACnB,KAAK,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;4BACtB,OAAO,EAAE,IAAI,EAAE,IAAa,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;wBACnD,CAAC;wBACD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;wBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;4BACjB,IAAI,UAAU,EAAE,CAAC;gCACf,UAAU,GAAG,KAAK,CAAC;gCACnB,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;gCACnC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,cAAc,GAAG,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;4BAC/E,CAAC;wBACH,CAAC;wBACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;4BAChB,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;4BAC9B,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;gCACvB,MAAM,KAAK,GAAG,GAAG,GAAG,cAAc,CAAC;gCACnC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,GAAG,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;4BAC7G,CAAC;iCAAM,CAAC;gCACN,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,eAAe,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;4BACxF,CAAC;wBACH,CAAC;wBACD,OAAO,MAAM,CAAC;oBAChB,CAAC;oBACD,KAAK,CAAC,MAAM,CAAC,KAAc;wBACzB,YAAY,EAAE,CAAC;wBACf,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,IAAa,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;oBAC5E,CAAC;oBACD,KAAK,CAAC,KAAK,CAAC,CAAW;wBACrB,YAAY,EAAE,CAAC;wBACf,OAAO,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAa,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;oBACvE,CAAC;iBACF,CAAC;YACJ,CAAC;SACF,CAAC;QAEF,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;QAC7C,MAAM,aAAa,GAAG,UAAU;aAC7B,IAAI,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;YACjC,MAAM,OAAO,GAAG,YAAY,EAAE,CAAC;YAC/B,IAAI,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;gBACtB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,eAAe,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC3D,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,eAAe,YAAY,kBAAkB,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;gBAChF,eAAe,CAAC,kBAAkB,CAAC,IAAI,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;YAC1E,CAAC;YAED,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;gBACzD,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC;gBACjD,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,iBAAiB,GAAG,KAAK,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACrE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CACxB,GAAG,IAAI,CAAC,iBAAiB,YAAY,SAAS,QAAQ,CACvD,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,OAAO,EAAE,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;YAC5D,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,iBAAiB,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;YACnH,MAAM,MAAM,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC1C,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC;YACpC,MAAM,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;YAC/C,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;YAEnC,IAAI,eAAe,EAAE,CAAC;gBACpB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC;YACrE,CAAC;YAED,IAAI,gBAAqC,CAAC;YAC1C,IAAI,MAAM,CAAC,QAAQ,EAAE,YAAY,IAAI,YAAY,EAAE,CAAC;gBAClD,MAAM,SAAS,GAAG,WAAW,CAAC,YAAY,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;gBACjE,IAAI,SAAS,CAAC,MAAM,KAAK,MAAM,IAAI,SAAS,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;oBAC3D,gBAAgB,GAAG,SAAS,CAAC,IAAI,CAAC;gBACpC,CAAC;YACH,CAAC;YAED,MAAM,YAAY,GAAiB,OAAO;gBACxC,CAAC,CAAC,OAAO;gBACT,CAAC,CAAC,SAAS;oBACT,CAAC,CAAC,YAAY;oBACd,CAAC,CAAC,MAAM,CAAC;YAEb,OAAO;gBACL,OAAO,EAAE,YAAY;gBACrB,eAAe;gBACf,gBAAgB;gBAChB,SAAS;gBACT,YAAY;gBACZ,KAAK,EAAE,eAAe,CAAC;oBACrB,YAAY,EAAE,IAAI,CAAC,aAAa;oBAChC,gBAAgB,EAAE,IAAI,CAAC,iBAAiB;oBACxC,eAAe,EAAE,eAAe;oBAChC,gBAAgB;iBACjB,CAAC;gBACF,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;aAC9B,CAAC;QACJ,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACZ,YAAY,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QAEL,OAAO;YACL,MAAM,EAAE,eAAe;YACvB,MAAM,EAAE,aAAa;SACtB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe;QACnB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,kCAAkC;QAClC,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO;YACjC,aAAa,EAAE,IAAI,CAAC,WAAW,CAAC,cAAc;YAC9C,QAAQ,EAAE;gBACR,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,mBAAmB;gBAC9D,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,UAAU;gBAC/C,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,gBAAgB;gBAC1D,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;oBACtD,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,eAAe;oBACvE,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO;oBACxD,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW;oBAChE,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC;wBACzE,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,gBAAgB;wBACzF,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,UAAU;wBAC9E,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,QAAQ;wBAC1E,aAAa,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,cAAc;wBACtF,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,YAAY;qBACnF,CAAC,CAAC,CAAC,SAAS;iBACd,CAAC,CAAC,CAAC,SAAS;aACd;YACD,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,CAAC;gBACrD,mBAAmB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,sBAAsB;gBAC9E,iBAAiB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,mBAAmB;gBACzE,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,iBAAiB;gBACtE,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,kBAAkB;gBACvE,kBAAkB,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,oBAAoB;aAC5E,CAAC,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,YAAY,kBAAkB,CAAC;YAAE,OAAO;QAClE,MAAM,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;QAC1C,IAAI,CAAC,CAAC,YAAY,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,cAAc,GAAG,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC;YAChD,CAAC,CAAC,iBAAiB,CAAC,CAAC,WAAW,WAAW,CAAC,CAAC,KAAK,GAAG;YACrD,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,KAAK,GAAa;YACtB,gBAAgB,CAAC,CAAC,YAAY,WAAW,cAAc,EAAE;SAC1D,CAAC;QACF,IAAI,CAAC,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,mBAAmB,GAAG,CAAC,CAAC,iBAAiB,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACpF,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,iBAAiB,YAAY,CAAC,CAAC,mBAAmB,YAAY,UAAU,IAAI,CAAC,CAAC;QACvG,CAAC;QACD,IAAI,CAAC,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,iBAAiB,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,MAAM,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;QACpC,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF"}
+{"version":3,"file":"mlx-driver.js","sourceRoot":"","sources":["../../src/mlx-ml/mlx-driver.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAKvE,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,yBAAyB,EAAE,MAAM,mCAAmC,CAAC;AAC9E,OAAO,EACL,gCAAgC,EAChC,qBAAqB,GACtB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAE/D,OAAO,EAAE,iBAAiB,EAAE,CAAC;AAuB7B;;;GAGG;AACH,MAAM,OAAO,SAAU,SAAQ,oBAAoB;IACzC,kBAAkB,CAAyB;IAC3C,mBAAmB,GAAG,KAAK,CAAC;IACnB,iBAAiB,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAEtD,YAAY,MAAuB;QACjC,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE;YAC9C,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,cAAc,EAAE,MAAM,CAAC,cAAc;SACtC,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe;YACzC,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC,eAAe,CAAC;YAC/C,CAAC,CAAC,SAAS,CAAC;QAEd,KAAK,CAAC;YACJ,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO,EAAE,UAAU;YACnB,QAAQ,EAAE,yBAAyB;YACnC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,YAAY,EAAE,KAAK;YACnB,KAAK,EAAE,YAAY;YACnB,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE;gBAC/C,IAAI,CAAC,YAAY;oBAAE,OAAO;gBAC1B,MAAM,gCAAgC,CACpC,YAAY,EACZ,IAAI,CAAC,iBAAiB,EACtB,WAAW,EACX,GAAG,EACH,GAAG,EAAE;oBACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CACvB,qEAAqE,CACtE,CAAC;oBACF,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;oBAChC,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC7B,CAAC,CACF,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,eAAe,CAAC;QAEjD,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,kBAAkB,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,KAAK,CAAC,cAA0C,CAAC;IAC1D,CAAC;IAED,IAAI,cAAc,CAAC,KAA+B;QAChD,KAAK,CAAC,cAAc,GAAG,KAAK,IAAI,EAAE,CAAC;IACrC,CAAC;IAED;;;;OAIG;IACH,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;oBAC9C,CAAC,CAAC;wBACE,cAAc,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,eAAe;wBAClE,OAAO,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO;wBACnD,WAAW,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW;wBAC3D,cAAc,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB;4BACjE,CAAC,CAAC;gCACE,cAAc,EACZ,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,gBAAgB;gCACtE,SAAS,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,UAAU;gCACzE,OAAO,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,QAAQ;gCACrE,aAAa,EACX,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,cAAc;gCACpE,WAAW,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,YAAY;6BAC9E;4BACH,CAAC,CAAC,SAAS;qBACd;oBACH,CAAC,CAAC,SAAS;aACd;YACD,gBAAgB,EAAE,WAAW,CAAC,iBAAiB;gBAC7C,CAAC,CAAC;oBACE,mBAAmB,EAAE,WAAW,CAAC,iBAAiB,CAAC,sBAAsB;oBACzE,iBAAiB,EAAE,WAAW,CAAC,iBAAiB,CAAC,mBAAmB;oBACpE,gBAAgB,EAAE,WAAW,CAAC,iBAAiB,CAAC,iBAAiB;oBACjE,gBAAgB,EAAE,WAAW,CAAC,iBAAiB,CAAC,kBAAkB;oBAClE,kBAAkB,EAAE,WAAW,CAAC,iBAAiB,CAAC,oBAAoB;iBACvE;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEQ,KAAK,CAAC,KAAK;QAClB,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,kBAAkB,EAAE,KAAK,EAAE,CAAC;QACzC,CAAC;QACD,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAEO,aAAa;QACnB,IAAI,IAAI,CAAC,mBAAmB;YAAE,OAAO;QACrC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,YAAY,kBAAkB,CAAC;YAAE,OAAO;QACrE,MAAM,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC;QAC7C,IAAI,CAAC,CAAC,YAAY,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,cAAc,GAClB,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,WAAW,WAAW,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACzF,MAAM,KAAK,GAAa,CAAC,gBAAgB,CAAC,CAAC,YAAY,WAAW,cAAc,EAAE,CAAC,CAAC;QACpF,IAAI,CAAC,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,mBAAmB,GAAG,CAAC,CAAC,iBAAiB,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACpF,KAAK,CAAC,IAAI,CACR,UAAU,CAAC,CAAC,iBAAiB,YAAY,CAAC,CAAC,mBAAmB,YAAY,UAAU,IAAI,CACzF,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,iBAAiB,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,CAAC;CACF"}

package/dist/mlx-ml/mlx-local-inference-adapters.d.ts

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

package/dist/mlx-ml/mlx-local-inference-adapters.d.ts.map

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