@modular-prompt/driver 0.4.5

package/dist/mlx-ml/process/model-handlers.js

@@ -0,0 +1,197 @@
+/**
+ * モデル固有の処理ハンドラー
+ *
+ * 各モデル特有のメッセージ変換やプロンプト生成処理を管理
+ */
+/**
+ * システムメッセージをマージ（末尾の改行をtrim）
+ */
+export function mergeSystemMessages(msgs) {
+    let systemContent = '';
+    const conversation = [];
+    for (const msg of msgs) {
+        if (msg.role === 'system') {
+            systemContent += msg.content + '\n\n';
+        }
+        else {
+            conversation.push(msg);
+        }
+    }
+    if (systemContent) {
+        const systemMessage = { role: 'system', content: systemContent.trim() };
+        return [systemMessage, ...conversation];
+    }
+    return conversation;
+}
+/**
+ * システムメッセージをマージ（末尾の改行を保持）
+ */
+function mergeSystemMessagesRaw(msgs) {
+    const systemMessages = [];
+    const conversation = [];
+    for (const msg of msgs) {
+        if (msg.role === 'system') {
+            systemMessages.push(msg.content);
+        }
+        else {
+            conversation.push(msg);
+        }
+    }
+    if (systemMessages.length > 0) {
+        const systemMessage = {
+            role: 'system',
+            content: systemMessages.join('\n\n')
+        };
+        return [systemMessage, ...conversation];
+    }
+    return conversation;
+}
+/**
+ * Tanuki-8B-dpo-v1用のChat処理
+ */
+export function processTanukiChat(messages) {
+    // まずsystemメッセージをマージ（改行を保持）
+    const mergedMessages = mergeSystemMessagesRaw(messages);
+    const result = [];
+    // systemメッセージにプレフィックスを追加
+    if (mergedMessages.length > 0 && mergedMessages[0].role === 'system') {
+        result.push({
+            role: 'system',
+            content: '以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。\n\n' + mergedMessages[0].content
+        });
+        result.push(...mergedMessages.slice(1));
+    }
+    else {
+        // systemメッセージがない場合
+        result.push({
+            role: 'system',
+            content: '以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。\n'
+        });
+        result.push(...mergedMessages);
+    }
+    // 最後にuserメッセージを追加
+    result.push({
+        role: 'user',
+        content: 'systemプロンプトで説明されたタスクを正確に実行し、Output Sectionに書かれるべき内容を出力してください。\n'
+    });
+    return result;
+}
+/**
+ * CodeLlama用のChat処理
+ */
+export function processCodeLlamaChat(messages) {
+    const processedMessages = mergeSystemMessages(messages);
+    // userがない場合は必要
+    if (processedMessages[processedMessages.length - 1].role !== 'user') {
+        processedMessages.push({
+            role: 'user',
+            content: 'Read the system prompt and output the appropriate content.',
+        });
+    }
+    return processedMessages;
+}
+/**
+ * Gemma-3用のChat処理
+ */
+export function processGemmaChat(messages) {
+    const processedMessages = mergeSystemMessages(messages);
+    // (system) => user => assistant => user => ... と、きれいに並んでいないとエラーになる。
+    if (processedMessages[processedMessages.length - 1].role !== 'user') {
+        processedMessages.push({
+            role: 'user',
+            content: 'Read the system prompt and output the appropriate content.',
+        });
+    }
+    return processedMessages;
+}
+/**
+ * llm-jp-3.1用のCompletion処理
+ * chat_templateに厳密に準拠した形式
+ * Note: llm-jp-3.1はforce-completion設定のため、chat APIは使用されない
+ */
+export function processLlmJpCompletion(prompt) {
+    // llm-jpは複数言語の混ぜ書きに弱い傾向があるため、特定の言語を優先しないよう明示的に指示
+    const llmJpSpecificInstruction = '指示は英語・日本語の混ぜ書きになっているが、どちらの言語も同等の指示として十分に理解すること。\n出力は与えられたタスクに対してふさわしい言語を選択する。\n\n';
+    // chat_templateのsystemメッセージと同じ固定文字列（良好な回答を得るために必要な固定プロンプト）
+    return '<s>以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。' +
+        '\n\n### 指示:\n' +
+        llmJpSpecificInstruction +
+        prompt +
+        '\n\n### 応答:';
+}
+/**
+ * Tanuki-8B用のCompletion処理
+ */
+export function processTanukiCompletion(prompt) {
+    // completion APIではブロック化トークンを使用可能
+    return `### システム:\n${prompt}\n\n### 応答:\n`;
+}
+/**
+ * Gemma-3用のCompletion処理
+ */
+export function processGemmaCompletion(prompt) {
+    // Gemma-3はchat形式に依存しないcompletion処理が可能
+    return `<start_of_turn>user\n${prompt}<end_of_turn>\n<start_of_turn>model\n`;
+}
+/**
+ * ELYZA (Llama 2ベース) 用のCompletion処理
+ *
+ * ELYZAはLlama 2のinstruction formatを使用:
+ * <s>[INST] <<SYS>>
+ * システムプロンプト
+ * <</SYS>>
+ *
+ * ユーザープロンプト [/INST]
+ *
+ * Note: <<SYS>>, [INST]などはspecial tokensではなく、複数トークンに分割される
+ */
+export function processElyzaCompletion(prompt) {
+    const B_INST = '[INST]';
+    const E_INST = '[/INST]';
+    const B_SYS = '<<SYS>>\n';
+    const E_SYS = '\n<</SYS>>\n\n';
+    const DEFAULT_SYSTEM = 'あなたは誠実で優秀な日本人のアシスタントです。';
+    // Llama 2 instruction format
+    // <s>は自動的に追加されるため含めない（add_bos_token: true）
+    return `${B_INST} ${B_SYS}${DEFAULT_SYSTEM}${E_SYS}${prompt} ${E_INST} `;
+}
+/**
+ * モデル名に基づいてChat処理を選択
+ */
+export function selectChatProcessor(modelName) {
+    // Note: llm-jp-3.1はforce-completion設定のため、ここには含めない
+    if (modelName.includes('Tanuki-8B-dpo-v1')) {
+        return processTanukiChat;
+    }
+    if (modelName.includes('mlx-community/CodeLlama')) {
+        return processCodeLlamaChat;
+    }
+    if (modelName.includes('mlx-community/gemma-3')) {
+        return processGemmaChat;
+    }
+    return null;
+}
+/**
+ * モデル名に基づいてCompletion処理を選択
+ */
+export function selectCompletionProcessor(modelName) {
+    if (modelName.includes('llm-jp-3.1')) {
+        return processLlmJpCompletion;
+    }
+    if (modelName.includes('Tanuki-8B')) {
+        return processTanukiCompletion;
+    }
+    if (modelName.includes('CodeLlama')) {
+        // CodeLlamaはそのまま返す
+        return (prompt) => prompt;
+    }
+    if (modelName.includes('gemma-3')) {
+        return processGemmaCompletion;
+    }
+    // ELYZA models (elyza/ELYZA-japanese-Llama-2-*)
+    if (modelName.toLowerCase().includes('elyza')) {
+        return processElyzaCompletion;
+    }
+    return null;
+}
+//# sourceMappingURL=model-handlers.js.map

package/dist/mlx-ml/process/model-handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-handlers.js","sourceRoot":"","sources":["../../../src/mlx-ml/process/model-handlers.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,IAAkB;IACpD,IAAI,aAAa,GAAG,EAAE,CAAC;IACvB,MAAM,YAAY,GAAiB,EAAE,CAAC;IAEtC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,aAAa,IAAI,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,aAAa,GAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,IAAI,EAAE,EAAE,CAAC;QACpF,OAAO,CAAC,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAS,sBAAsB,CAAC,IAAkB;IAChD,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,MAAM,YAAY,GAAiB,EAAE,CAAC;IAEtC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,MAAM,aAAa,GAAe;YAChC,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC;SACrC,CAAC;QACF,OAAO,CAAC,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,QAAsB;IACtD,2BAA2B;IAC3B,MAAM,cAAc,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAExD,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,yBAAyB;IACzB,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACrE,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,yCAAyC,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO;SAC/E,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC;SAAM,CAAC;QACN,mBAAmB;QACnB,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,uCAAuC;SACjD,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IACjC,CAAC;IAED,kBAAkB;IAClB,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,iEAAiE;KAC3E,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,QAAsB;IACzD,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;IACxD,eAAe;IACf,IAAI,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACpE,iBAAiB,CAAC,IAAI,CAAC;YACrB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,4DAA4D;SAC7D,CAAC,CAAC;IACd,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,QAAsB;IACrD,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;IACxD,oEAAoE;IACpE,IAAI,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACpE,iBAAiB,CAAC,IAAI,CAAC;YACrB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,4DAA4D;SAC7D,CAAC,CAAC;IACd,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,iDAAiD;IACjD,MAAM,wBAAwB,GAAG,mFAAmF,CAAC;IAErH,2DAA2D;IAC3D,OAAO,wCAAwC;QAC7C,eAAe;QACf,wBAAwB;QACxB,MAAM;QACN,aAAa,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,MAAc;IACpD,iCAAiC;IACjC,OAAO,cAAc,MAAM,eAAe,CAAC;AAC7C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,sCAAsC;IACtC,OAAO,wBAAwB,MAAM,uCAAuC,CAAC;AAC/E,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,MAAM,MAAM,GAAG,QAAQ,CAAC;IACxB,MAAM,MAAM,GAAG,SAAS,CAAC;IACzB,MAAM,KAAK,GAAG,WAAW,CAAC;IAC1B,MAAM,KAAK,GAAG,gBAAgB,CAAC;IAC/B,MAAM,cAAc,GAAG,yBAAyB,CAAC;IAEjD,6BAA6B;IAC7B,2CAA2C;IAC3C,OAAO,GAAG,MAAM,IAAI,KAAK,GAAG,cAAc,GAAG,KAAK,GAAG,MAAM,IAAI,MAAM,GAAG,CAAC;AAC3E,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,SAAiB;IACnD,kDAAkD;IAClD,IAAI,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC3C,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IACD,IAAI,SAAS,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC;QAClD,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IACD,IAAI,SAAS,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE,CAAC;QAChD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB,CAAC,SAAiB;IACzD,IAAI,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,OAAO,sBAAsB,CAAC;IAChC,CAAC;IACD,IAAI,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QACpC,OAAO,uBAAuB,CAAC;IACjC,CAAC;IACD,IAAI,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QACpC,mBAAmB;QACnB,OAAO,CAAC,MAAc,EAAE,EAAE,CAAC,MAAM,CAAC;IACpC,CAAC;IACD,IAAI,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAClC,OAAO,sBAAsB,CAAC;IAChC,CAAC;IACD,gDAAgD;IAChD,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9C,OAAO,sBAAsB,CAAC;IAChC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/mlx-ml/process/model-specific.d.ts

@@ -0,0 +1,35 @@
+/**
+ * MLX Driver モデル固有処理
+ *
+ * モデルごとの特殊なフォーマット処理、メッセージ変換、プロンプト生成を管理
+ * Python側のapply_model_specific_processingをTypeScript側に移行
+ */
+import type { MlxMessage } from './types.js';
+export interface ModelSpecificProcessor {
+    /**
+     * Chat API用のモデル固有処理
+     * メッセージ配列を受け取り、モデルに最適化されたメッセージ配列を返す
+     */
+    applyChatSpecificProcessing(messages: MlxMessage[]): MlxMessage[];
+    /**
+     * Completion API用のモデル固有処理（文字列ベース）
+     * プロンプト文字列を受け取り、モデルに最適化されたプロンプトを返す
+     */
+    applyCompletionSpecificProcessing(prompt: string): string;
+}
+export declare class DefaultModelSpecificProcessor implements ModelSpecificProcessor {
+    private modelName;
+    constructor(modelName: string);
+    /**
+     * Chat API用のモデル固有処理
+     * モデルごとのチャットフォーマットに合わせるための処理
+     */
+    applyChatSpecificProcessing(messages: MlxMessage[]): MlxMessage[];
+    /**
+     * Completion API用のモデル固有処理
+     * モデルごとにブロック化トークンやプロンプトフォーマットを適用
+     */
+    applyCompletionSpecificProcessing(prompt: string): string;
+}
+export declare function createModelSpecificProcessor(modelName: string): ModelSpecificProcessor;
+//# sourceMappingURL=model-specific.d.ts.map

package/dist/mlx-ml/process/model-specific.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-specific.d.ts","sourceRoot":"","sources":["../../../src/mlx-ml/process/model-specific.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAG7C,MAAM,WAAW,sBAAsB;IACrC;;;OAGG;IACH,2BAA2B,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,UAAU,EAAE,CAAC;IAElE;;;OAGG;IACH,iCAAiC,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;CAC3D;AAGD,qBAAa,6BAA8B,YAAW,sBAAsB;IAC9D,OAAO,CAAC,SAAS;gBAAT,SAAS,EAAE,MAAM;IAGrC;;;OAGG;IACH,2BAA2B,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,UAAU,EAAE;IAKjE;;;OAGG;IACH,iCAAiC,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;CAI1D;AAGD,wBAAgB,4BAA4B,CAC1C,SAAS,EAAE,MAAM,GAChB,sBAAsB,CAGxB"}

package/dist/mlx-ml/process/model-specific.js

@@ -0,0 +1,35 @@
+/**
+ * MLX Driver モデル固有処理
+ *
+ * モデルごとの特殊なフォーマット処理、メッセージ変換、プロンプト生成を管理
+ * Python側のapply_model_specific_processingをTypeScript側に移行
+ */
+import { selectChatProcessor, selectCompletionProcessor } from './model-handlers.js';
+export class DefaultModelSpecificProcessor {
+    modelName;
+    constructor(modelName) {
+        this.modelName = modelName;
+    }
+    /**
+     * Chat API用のモデル固有処理
+     * モデルごとのチャットフォーマットに合わせるための処理
+     */
+    applyChatSpecificProcessing(messages) {
+        const processor = selectChatProcessor(this.modelName);
+        return processor ? processor(messages) : messages;
+    }
+    /**
+     * Completion API用のモデル固有処理
+     * モデルごとにブロック化トークンやプロンプトフォーマットを適用
+     */
+    applyCompletionSpecificProcessing(prompt) {
+        const processor = selectCompletionProcessor(this.modelName);
+        return processor ? processor(prompt) : prompt;
+    }
+}
+// ファクトリー関数
+export function createModelSpecificProcessor(modelName) {
+    // 将来的に異なるプロセッサーを返すことも可能
+    return new DefaultModelSpecificProcessor(modelName);
+}
+//# sourceMappingURL=model-specific.js.map

package/dist/mlx-ml/process/model-specific.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-specific.js","sourceRoot":"","sources":["../../../src/mlx-ml/process/model-specific.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,mBAAmB,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAiBrF,MAAM,OAAO,6BAA6B;IACpB;IAApB,YAAoB,SAAiB;QAAjB,cAAS,GAAT,SAAS,CAAQ;IACrC,CAAC;IAED;;;OAGG;IACH,2BAA2B,CAAC,QAAsB;QAChD,MAAM,SAAS,GAAG,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtD,OAAO,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IACpD,CAAC;IAED;;;OAGG;IACH,iCAAiC,CAAC,MAAc;QAC9C,MAAM,SAAS,GAAG,yBAAyB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5D,OAAO,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAChD,CAAC;CACF;AAED,WAAW;AACX,MAAM,UAAU,4BAA4B,CAC1C,SAAiB;IAEjB,wBAAwB;IACxB,OAAO,IAAI,6BAA6B,CAAC,SAAS,CAAC,CAAC;AACtD,CAAC"}

package/dist/mlx-ml/process/parameter-mapper.d.ts

@@ -0,0 +1,17 @@
+/**
+ * MLX Driver パラメータマッピング
+ *
+ * TypeScript側のパラメータ名をPython側のパラメータ名に変換し、
+ * サポートされているパラメータのみをフィルタリングする
+ */
+import type { MlxMlModelOptions } from './types.js';
+/**
+ * TypeScript側のオプションをPython側のオプションに変換
+ *
+ * @param options TypeScript側のオプション（キャメルケース）
+ * @param strict 厳密モード（不明なパラメータをエラーとして扱う）
+ * @returns Python側で使用可能なオプション（スネークケース）
+ * @throws {Error} バリデーションエラーが発生した場合
+ */
+export declare function mapOptionsToPython(options?: MlxMlModelOptions, strict?: boolean): MlxMlModelOptions | undefined;
+//# sourceMappingURL=parameter-mapper.d.ts.map

package/dist/mlx-ml/process/parameter-mapper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parameter-mapper.d.ts","sourceRoot":"","sources":["../../../src/mlx-ml/process/parameter-mapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAsCpD;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAChC,OAAO,CAAC,EAAE,iBAAiB,EAC3B,MAAM,GAAE,OAAc,GACrB,iBAAiB,GAAG,SAAS,CAiD/B"}

package/dist/mlx-ml/process/parameter-mapper.js

@@ -0,0 +1,91 @@
+/**
+ * MLX Driver パラメータマッピング
+ *
+ * TypeScript側のパラメータ名をPython側のパラメータ名に変換し、
+ * サポートされているパラメータのみをフィルタリングする
+ */
+import { validateOptions, formatValidationResult } from './parameter-validator.js';
+/**
+ * キャメルケースをスネークケースに変換
+ *
+ * @param str キャメルケースの文字列
+ * @returns スネークケースの文字列
+ */
+function camelToSnakeCase(str) {
+    return str.replace(/[A-Z]/g, letter => `_${letter.toLowerCase()}`);
+}
+/**
+ * MLX stream_generateでサポートされているパラメータ
+ * TypeScript側 (キャメルケース) -> Python側 (スネークケース) のマッピング
+ *
+ * 注: MLXは内部でmake_samplerを使用しており、
+ * stream_generateは**kwargsで受け取ったパラメータを処理する
+ */
+const SPECIAL_MAPPINGS = {
+// 現在は特殊なマッピングなし
+// temperatureはそのままtemperatureとして送る
+};
+/**
+ * MLXでサポートされているパラメータのリスト
+ * キャメルケースで定義（TypeScript側の名前）
+ */
+const SUPPORTED_PARAMS = new Set([
+    'maxTokens', // -> max_tokens
+    'temperature', // -> temp (特殊ケース)
+    'topP', // -> top_p
+    'topK', // -> top_k
+    'repetitionPenalty', // -> repetition_penalty
+    'repetitionContextSize', // -> repetition_context_size
+]);
+/**
+ * TypeScript側のオプションをPython側のオプションに変換
+ *
+ * @param options TypeScript側のオプション（キャメルケース）
+ * @param strict 厳密モード（不明なパラメータをエラーとして扱う）
+ * @returns Python側で使用可能なオプション（スネークケース）
+ * @throws {Error} バリデーションエラーが発生した場合
+ */
+export function mapOptionsToPython(options, strict = true) {
+    if (!options) {
+        return undefined;
+    }
+    // まずバリデーションを実行
+    const validationResult = validateOptions(options, strict);
+    // バリデーションエラーがある場合
+    if (!validationResult.valid) {
+        const errorMessage = formatValidationResult(validationResult);
+        console.error(`[MLX Driver] ${errorMessage}`);
+        if (strict) {
+            // 厳密モードではエラーをスロー
+            throw new Error(`MLX Driver parameter validation failed:\n${errorMessage}`);
+        }
+        // 非厳密モードでは処理を続行（エラーのあったパラメータはスキップ）
+    }
+    // 警告がある場合は出力
+    if (validationResult.warnings.length > 0) {
+        const warningMessage = formatValidationResult({
+            valid: true,
+            errors: [],
+            warnings: validationResult.warnings,
+            validatedOptions: validationResult.validatedOptions
+        });
+        console.warn(`[MLX Driver] ${warningMessage}`);
+    }
+    // バリデーション済みのオプションを使用
+    const validatedOptions = validationResult.validatedOptions || {};
+    const mappedOptions = {};
+    for (const [key, value] of Object.entries(validatedOptions)) {
+        // 特殊なマッピングがある場合はそれを使用
+        let pythonKey;
+        if (key in SPECIAL_MAPPINGS) {
+            pythonKey = SPECIAL_MAPPINGS[key];
+        }
+        else {
+            // 通常のキャメルケース -> スネークケース変換
+            pythonKey = camelToSnakeCase(key);
+        }
+        mappedOptions[pythonKey] = value;
+    }
+    return Object.keys(mappedOptions).length > 0 ? mappedOptions : undefined;
+}
+//# sourceMappingURL=parameter-mapper.js.map

package/dist/mlx-ml/process/parameter-mapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parameter-mapper.js","sourceRoot":"","sources":["../../../src/mlx-ml/process/parameter-mapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAEnF;;;;;GAKG;AACH,SAAS,gBAAgB,CAAC,GAAW;IACnC,OAAO,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;AACrE,CAAC;AAED;;;;;;GAMG;AACH,MAAM,gBAAgB,GAA2B;AAC/C,gBAAgB;AAChB,mCAAmC;CACpC,CAAC;AAEF;;;GAGG;AACH,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAC/B,WAAW,EAAS,gBAAgB;IACpC,aAAa,EAAO,kBAAkB;IACtC,MAAM,EAAa,WAAW;IAC9B,MAAM,EAAa,WAAW;IAC9B,mBAAmB,EAAM,wBAAwB;IACjD,uBAAuB,EAAE,6BAA6B;CACvD,CAAC,CAAC;AAEH;;;;;;;GAOG;AACH,MAAM,UAAU,kBAAkB,CAChC,OAA2B,EAC3B,SAAkB,IAAI;IAEtB,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,eAAe;IACf,MAAM,gBAAgB,GAAG,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAE1D,kBAAkB;IAClB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC5B,MAAM,YAAY,GAAG,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;QAC9D,OAAO,CAAC,KAAK,CAAC,gBAAgB,YAAY,EAAE,CAAC,CAAC;QAE9C,IAAI,MAAM,EAAE,CAAC;YACX,iBAAiB;YACjB,MAAM,IAAI,KAAK,CAAC,4CAA4C,YAAY,EAAE,CAAC,CAAC;QAC9E,CAAC;QACD,mCAAmC;IACrC,CAAC;IAED,aAAa;IACb,IAAI,gBAAgB,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,cAAc,GAAG,sBAAsB,CAAC;YAC5C,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,EAAE;YACV,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;YACnC,gBAAgB,EAAE,gBAAgB,CAAC,gBAAgB;SACpD,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,gBAAgB,cAAc,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,qBAAqB;IACrB,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,gBAAgB,IAAI,EAAE,CAAC;IACjE,MAAM,aAAa,GAAsB,EAAE,CAAC;IAE5C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC5D,sBAAsB;QACtB,IAAI,SAAiB,CAAC;QACtB,IAAI,GAAG,IAAI,gBAAgB,EAAE,CAAC;YAC5B,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,0BAA0B;YAC1B,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;QAED,aAAa,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC;IACnC,CAAC;IAED,OAAO,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3E,CAAC"}

package/dist/mlx-ml/process/parameter-validator.d.ts

@@ -0,0 +1,55 @@
+/**
+ * MLX Driver パラメータバリデーター
+ *
+ * オプションパラメータの存在チェックと値の妥当性検証を行う
+ */
+import type { MlxMlModelOptions } from './types.js';
+/**
+ * バリデーション結果
+ */
+export interface ValidationResult {
+    valid: boolean;
+    errors: ValidationError[];
+    warnings: ValidationWarning[];
+    validatedOptions?: MlxMlModelOptions;
+}
+/**
+ * バリデーションエラー
+ */
+export interface ValidationError {
+    field: string;
+    value: any;
+    message: string;
+    type: 'unknown_parameter' | 'invalid_type' | 'out_of_range';
+}
+/**
+ * バリデーション警告
+ */
+export interface ValidationWarning {
+    field: string;
+    value: any;
+    message: string;
+    correctedValue?: any;
+}
+/**
+ * オプションパラメータのバリデーション
+ *
+ * @param options 検証するオプション
+ * @param strict 厳密モード（不明なパラメータをエラーとして扱う）
+ * @returns バリデーション結果
+ */
+export declare function validateOptions(options: Record<string, any> | undefined, strict?: boolean): ValidationResult;
+/**
+ * バリデーション結果をフォーマットして出力
+ *
+ * @param result バリデーション結果
+ * @returns フォーマットされたエラーメッセージ
+ */
+export declare function formatValidationResult(result: ValidationResult): string;
+/**
+ * 利用可能なパラメータのヘルプテキストを生成
+ *
+ * @returns パラメータのヘルプテキスト
+ */
+export declare function getParameterHelp(): string;
+//# sourceMappingURL=parameter-validator.d.ts.map

package/dist/mlx-ml/process/parameter-validator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parameter-validator.d.ts","sourceRoot":"","sources":["../../../src/mlx-ml/process/parameter-validator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEpD;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,eAAe,EAAE,CAAC;IAC1B,QAAQ,EAAE,iBAAiB,EAAE,CAAC;IAC9B,gBAAgB,CAAC,EAAE,iBAAiB,CAAC;CACtC;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,GAAG,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,mBAAmB,GAAG,cAAc,GAAG,cAAc,CAAC;CAC7D;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,GAAG,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,GAAG,CAAC;CACtB;AA8DD;;;;;;GAMG;AACH,wBAAgB,eAAe,CAC7B,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS,EACxC,MAAM,GAAE,OAAc,GACrB,gBAAgB,CAuGlB;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,gBAAgB,GAAG,MAAM,CAsBvE;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,IAAI,MAAM,CAkBzC"}

package/dist/mlx-ml/process/parameter-validator.js

@@ -0,0 +1,203 @@
+/**
+ * MLX Driver パラメータバリデーター
+ *
+ * オプションパラメータの存在チェックと値の妥当性検証を行う
+ */
+/**
+ * MLXパラメータの制約定義
+ * キャメルケース形式のみで定義（APIインターフェース用）
+ */
+const PARAMETER_CONSTRAINTS = {
+    maxTokens: {
+        type: 'number',
+        min: 1,
+        max: 100000,
+        default: 1000,
+        description: '生成する最大トークン数'
+    },
+    temperature: {
+        type: 'number',
+        min: 0.0,
+        max: 2.0,
+        default: 1.0,
+        description: '生成のランダム性を制御する温度パラメータ'
+    },
+    topP: {
+        type: 'number',
+        min: 0.0,
+        max: 1.0,
+        default: 1.0,
+        description: 'Top-pサンプリングのしきい値'
+    },
+    topK: {
+        type: 'number',
+        min: 1,
+        max: 1000,
+        default: 40,
+        description: 'Top-kサンプリングのk値'
+    },
+    repetitionPenalty: {
+        type: 'number',
+        min: 0.0,
+        max: 2.0,
+        default: 1.0,
+        description: '繰り返しペナルティの強度'
+    },
+    repetitionContextSize: {
+        type: 'number',
+        min: 1,
+        max: 10000,
+        default: 20,
+        description: '繰り返し検出のコンテキストサイズ'
+    }
+};
+/**
+ * オプションパラメータのバリデーション
+ *
+ * @param options 検証するオプション
+ * @param strict 厳密モード（不明なパラメータをエラーとして扱う）
+ * @returns バリデーション結果
+ */
+export function validateOptions(options, strict = true) {
+    const errors = [];
+    const warnings = [];
+    const validatedOptions = {};
+    if (!options) {
+        return { valid: true, errors: [], warnings: [], validatedOptions: undefined };
+    }
+    // 各パラメータをチェック
+    for (const [key, value] of Object.entries(options)) {
+        const constraint = PARAMETER_CONSTRAINTS[key];
+        // 不明なパラメータ
+        if (!constraint) {
+            // 利用可能なパラメータから重複を除く（camelCaseのみ表示）
+            const availableParams = ['maxTokens', 'temperature', 'topP', 'topK', 'repetitionPenalty', 'repetitionContextSize'];
+            if (strict) {
+                errors.push({
+                    field: key,
+                    value,
+                    message: `Unknown parameter '${key}'. Available parameters: ${availableParams.join(', ')}`,
+                    type: 'unknown_parameter'
+                });
+            }
+            else {
+                warnings.push({
+                    field: key,
+                    value,
+                    message: `Unknown parameter '${key}' will be ignored`
+                });
+            }
+            continue;
+        }
+        // 型チェック
+        const actualType = typeof value;
+        if (actualType !== constraint.type) {
+            // 数値への変換を試みる
+            if (constraint.type === 'number' && actualType === 'string') {
+                const numValue = parseFloat(value);
+                if (!isNaN(numValue)) {
+                    warnings.push({
+                        field: key,
+                        value,
+                        message: `Parameter '${key}' was converted from string to number`,
+                        correctedValue: numValue
+                    });
+                    validatedOptions[key] = numValue;
+                    continue;
+                }
+            }
+            errors.push({
+                field: key,
+                value,
+                message: `Parameter '${key}' must be of type ${constraint.type}, got ${actualType}`,
+                type: 'invalid_type'
+            });
+            continue;
+        }
+        // 範囲チェック（数値型の場合）
+        if (constraint.type === 'number') {
+            let finalValue = value;
+            // 最小値チェック
+            if (constraint.min !== undefined && value < constraint.min) {
+                warnings.push({
+                    field: key,
+                    value,
+                    message: `Parameter '${key}' value ${value} is below minimum ${constraint.min}, using minimum value`,
+                    correctedValue: constraint.min
+                });
+                finalValue = constraint.min;
+            }
+            // 最大値チェック
+            if (constraint.max !== undefined && value > constraint.max) {
+                warnings.push({
+                    field: key,
+                    value,
+                    message: `Parameter '${key}' value ${value} exceeds maximum ${constraint.max}, using maximum value`,
+                    correctedValue: constraint.max
+                });
+                finalValue = constraint.max;
+            }
+            validatedOptions[key] = finalValue;
+        }
+        else {
+            // その他の型はそのまま設定
+            validatedOptions[key] = value;
+        }
+    }
+    const valid = errors.length === 0;
+    return {
+        valid,
+        errors,
+        warnings,
+        validatedOptions: valid ? validatedOptions : undefined
+    };
+}
+/**
+ * バリデーション結果をフォーマットして出力
+ *
+ * @param result バリデーション結果
+ * @returns フォーマットされたエラーメッセージ
+ */
+export function formatValidationResult(result) {
+    const messages = [];
+    if (!result.valid) {
+        messages.push('❌ Validation failed:');
+        for (const error of result.errors) {
+            messages.push(`  - ${error.message}`);
+        }
+    }
+    if (result.warnings.length > 0) {
+        messages.push('⚠️  Warnings:');
+        for (const warning of result.warnings) {
+            let msg = `  - ${warning.message}`;
+            if (warning.correctedValue !== undefined) {
+                msg += ` (corrected to: ${warning.correctedValue})`;
+            }
+            messages.push(msg);
+        }
+    }
+    return messages.join('\n');
+}
+/**
+ * 利用可能なパラメータのヘルプテキストを生成
+ *
+ * @returns パラメータのヘルプテキスト
+ */
+export function getParameterHelp() {
+    const lines = ['Available MLX parameters (use camelCase format):'];
+    for (const [key, constraint] of Object.entries(PARAMETER_CONSTRAINTS)) {
+        let line = `  - ${key} (${constraint.type}): ${constraint.description}`;
+        if (constraint.type === 'number' && (constraint.min !== undefined || constraint.max !== undefined)) {
+            const range = [];
+            if (constraint.min !== undefined)
+                range.push(`min: ${constraint.min}`);
+            if (constraint.max !== undefined)
+                range.push(`max: ${constraint.max}`);
+            line += ` [${range.join(', ')}]`;
+        }
+        line += ` (default: ${constraint.default})`;
+        lines.push(line);
+    }
+    return lines.join('\n');
+}
+//# sourceMappingURL=parameter-validator.js.map