@modular-prompt/driver 0.6.3 → 0.8.1

package/dist/mlx-ml/tool-call-parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-parser.d.ts","sourceRoot":"","sources":["../../src/mlx-ml/tool-call-parser.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAE5E,MAAM,WAAW,mBAAmB;IAClC,wBAAwB;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,sBAAsB;IACtB,SAAS,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED;;;;;;;;GAQG;AACH,wBAAgB,cAAc,CAC5B,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,cAAc,GAAG,IAAI,GACjC,mBAAmB,CA+BrB;AA6GD;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,KAAK,EAAE,cAAc,EAAE,EACvB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,gBAAgB,CAAC,EAC/D,cAAc,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,GAC1D,MAAM,CAqCR"}

package/dist/mlx-ml/tool-call-parser.js

@@ -0,0 +1,163 @@
+/**
+ * モデル出力からtool callを検出・パース
+ *
+ * 検出戦略:
+ * 1. runtimeInfo.features.chat_template.tool_call_format が存在 → テンプレート由来のデリミタで検出
+ * 2. runtimeInfo.special_tokens.tool_call が存在 → 特殊トークンによるデリミタで検出
+ * 3. ```json:toolCall コードブロック → ラベル付きコードブロック検出
+ * 4. 汎用フォールバック → JSON形式のtool call検出
+ */
+export function parseToolCalls(text, runtimeInfo) {
+    // 1. tool_call_format（テンプレート由来）による検出
+    const toolCallFormat = runtimeInfo?.features?.chat_template?.tool_call_format;
+    if (toolCallFormat?.call_start && toolCallFormat?.call_end) {
+        const result = parseWithDelimiters(text, toolCallFormat.call_start, toolCallFormat.call_end);
+        if (result.toolCalls.length > 0) {
+            return result;
+        }
+    }
+    // 2. 特殊トークンによる検出
+    const toolCallToken = runtimeInfo?.special_tokens?.tool_call;
+    if (toolCallToken && typeof toolCallToken === 'object' && 'start' in toolCallToken) {
+        const result = parseWithDelimiters(text, toolCallToken.start.text, toolCallToken.end.text);
+        if (result.toolCalls.length > 0) {
+            return result;
+        }
+    }
+    // 3. ```json:toolCall コードブロック検出
+    const codeBlockResult = parseCodeBlockToolCalls(text);
+    if (codeBlockResult.toolCalls.length > 0) {
+        return codeBlockResult;
+    }
+    // 4. 汎用フォールバック
+    return parseGenericToolCalls(text);
+}
+function parseWithDelimiters(text, startDelimiter, endDelimiter) {
+    const toolCalls = [];
+    let content = text;
+    let callIndex = 0;
+    // startDelimiter...endDelimiter のパターンを繰り返し検索
+    const regex = new RegExp(escapeRegExp(startDelimiter) + '([\\s\\S]*?)' + escapeRegExp(endDelimiter), 'g');
+    let match;
+    while ((match = regex.exec(text)) !== null) {
+        const jsonStr = match[1].trim();
+        try {
+            const parsed = JSON.parse(jsonStr);
+            toolCalls.push({
+                id: `call_${callIndex++}`,
+                name: parsed.name,
+                arguments: parsed.arguments || parsed.parameters || {}
+            });
+        }
+        catch {
+            // JSONパース失敗 → スキップ
+        }
+    }
+    if (toolCalls.length > 0) {
+        // tool call部分をテキストから除去
+        content = text.replace(regex, '').trim();
+    }
+    return { content, toolCalls };
+}
+function parseCodeBlockToolCalls(text) {
+    const toolCalls = [];
+    let content = text;
+    let callIndex = 0;
+    // ```json:toolCall ... ``` パターンを検出
+    const codeBlockRegex = /```json:toolCall\s*\n([\s\S]*?)```/g;
+    let match;
+    while ((match = codeBlockRegex.exec(text)) !== null) {
+        const jsonStr = match[1].trim();
+        try {
+            const parsed = JSON.parse(jsonStr);
+            toolCalls.push({
+                id: `call_${callIndex++}`,
+                name: parsed.name,
+                arguments: parsed.arguments || parsed.parameters || {}
+            });
+        }
+        catch {
+            // JSONパース失敗 → スキップ
+        }
+    }
+    if (toolCalls.length > 0) {
+        content = text.replace(codeBlockRegex, '').trim();
+    }
+    return { content, toolCalls };
+}
+function parseGenericToolCalls(text) {
+    const toolCalls = [];
+    let content = text;
+    let callIndex = 0;
+    // 汎用パターン: {"name": "...", "arguments": {...}} を検出
+    // 行頭からJSONオブジェクトが始まるか、テキスト末尾のJSONブロックを検出
+    const jsonPattern = /\{[\s\S]*?"name"\s*:\s*"[^"]+?"[\s\S]*?(?:"arguments"|"parameters")\s*:\s*\{[\s\S]*?\}\s*\}/g;
+    let match;
+    while ((match = jsonPattern.exec(text)) !== null) {
+        try {
+            const parsed = JSON.parse(match[0]);
+            if (parsed.name && (parsed.arguments || parsed.parameters)) {
+                toolCalls.push({
+                    id: `call_${callIndex++}`,
+                    name: parsed.name,
+                    arguments: parsed.arguments || parsed.parameters || {}
+                });
+            }
+        }
+        catch {
+            // JSONパース失敗 → スキップ
+        }
+    }
+    if (toolCalls.length > 0) {
+        // tool call部分をテキストから除去（最後のJSON部分のみ）
+        for (const match2 of [...text.matchAll(jsonPattern)]) {
+            content = content.replace(match2[0], '').trim();
+        }
+    }
+    return { content, toolCalls };
+}
+function escapeRegExp(str) {
+    return str.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+/**
+ * tool定義をテキスト形式にフォーマット
+ * tool_call_formatまたは特殊トークンがある場合はそのフォーマットに合わせた指示を生成
+ */
+export function formatToolDefinitionsAsText(tools, specialTokens, toolCallFormat) {
+    const lines = ['## Available Tools', ''];
+    for (const tool of tools) {
+        lines.push(`### ${tool.name}`);
+        if (tool.description) {
+            lines.push(tool.description);
+        }
+        if (tool.parameters) {
+            lines.push(`Parameters: ${JSON.stringify(tool.parameters)}`);
+        }
+        lines.push('');
+    }
+    // tool call出力フォーマットの指示
+    if (toolCallFormat?.call_start && toolCallFormat?.call_end) {
+        lines.push('To call a tool, respond with:');
+        lines.push(toolCallFormat.call_start);
+        lines.push('{"name": "tool_name", "arguments": {"param": "value"}}');
+        lines.push(toolCallFormat.call_end);
+    }
+    else {
+        const toolCallToken = specialTokens?.tool_call;
+        if (toolCallToken && 'start' in toolCallToken && 'end' in toolCallToken) {
+            const pair = toolCallToken;
+            lines.push('To call a tool, respond with:');
+            lines.push(`${pair.start.text}`);
+            lines.push('{"name": "tool_name", "arguments": {"param": "value"}}');
+            lines.push(`${pair.end.text}`);
+        }
+        else {
+            lines.push('To call a tool, respond with:');
+            lines.push('```json:toolCall');
+            lines.push('{"name": "tool_name", "arguments": {"param": "value"}}');
+            lines.push('```');
+        }
+    }
+    return lines.join('\n');
+}
+//# sourceMappingURL=tool-call-parser.js.map

package/dist/mlx-ml/tool-call-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-parser.js","sourceRoot":"","sources":["../../src/mlx-ml/tool-call-parser.ts"],"names":[],"mappings":"AAYA;;;;;;;;GAQG;AACH,MAAM,UAAU,cAAc,CAC5B,IAAY,EACZ,WAAkC;IAElC,qCAAqC;IACrC,MAAM,cAAc,GAAG,WAAW,EAAE,QAAQ,EAAE,aAAa,EAAE,gBAAgB,CAAC;IAC9E,IAAI,cAAc,EAAE,UAAU,IAAI,cAAc,EAAE,QAAQ,EAAE,CAAC;QAC3D,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC7F,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IAED,iBAAiB;IACjB,MAAM,aAAa,GAAG,WAAW,EAAE,cAAc,EAAE,SAAS,CAAC;IAC7D,IAAI,aAAa,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,OAAO,IAAI,aAAa,EAAE,CAAC;QACnF,MAAM,MAAM,GAAG,mBAAmB,CAChC,IAAI,EACJ,aAAa,CAAC,KAAK,CAAC,IAAI,EACxB,aAAa,CAAC,GAAG,CAAC,IAAI,CACvB,CAAC;QACF,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IAED,gCAAgC;IAChC,MAAM,eAAe,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;IACtD,IAAI,eAAe,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,eAAe;IACf,OAAO,qBAAqB,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC;AAED,SAAS,mBAAmB,CAC1B,IAAY,EACZ,cAAsB,EACtB,YAAoB;IAEpB,MAAM,SAAS,GAAe,EAAE,CAAC;IACjC,IAAI,OAAO,GAAG,IAAI,CAAC;IACnB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,6CAA6C;IAC7C,MAAM,KAAK,GAAG,IAAI,MAAM,CACtB,YAAY,CAAC,cAAc,CAAC,GAAG,cAAc,GAAG,YAAY,CAAC,YAAY,CAAC,EAC1E,GAAG,CACJ,CAAC;IAEF,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC3C,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,SAAS,CAAC,IAAI,CAAC;gBACb,EAAE,EAAE,QAAQ,SAAS,EAAE,EAAE;gBACzB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,IAAI,EAAE;aACvD,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAuB;QACvB,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAY;IAC3C,MAAM,SAAS,GAAe,EAAE,CAAC;IACjC,IAAI,OAAO,GAAG,IAAI,CAAC;IACnB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,mCAAmC;IACnC,MAAM,cAAc,GAAG,qCAAqC,CAAC;IAE7D,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACpD,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,SAAS,CAAC,IAAI,CAAC;gBACb,EAAE,EAAE,QAAQ,SAAS,EAAE,EAAE;gBACzB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,IAAI,EAAE;aACvD,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACpD,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,qBAAqB,CAAC,IAAY;IACzC,MAAM,SAAS,GAAe,EAAE,CAAC;IACjC,IAAI,OAAO,GAAG,IAAI,CAAC;IACnB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,kDAAkD;IAClD,yCAAyC;IACzC,MAAM,WAAW,GAAG,8FAA8F,CAAC;IAEnH,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACjD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC3D,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,QAAQ,SAAS,EAAE,EAAE;oBACzB,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,IAAI,EAAE;iBACvD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,oCAAoC;QACpC,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC;YACrD,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClD,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,YAAY,CAAC,GAAW;IAC/B,OAAO,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AACpD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,2BAA2B,CACzC,KAAuB,EACvB,aAA+D,EAC/D,cAA2D;IAE3D,MAAM,KAAK,GAAa,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;IAEnD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,uBAAuB;IACvB,IAAI,cAAc,EAAE,UAAU,IAAI,cAAc,EAAE,QAAQ,EAAE,CAAC;QAC3D,KAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QAC5C,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;QACrE,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;SAAM,CAAC;QACN,MAAM,aAAa,GAAG,aAAa,EAAE,SAAS,CAAC;QAC/C,IAAI,aAAa,IAAI,OAAO,IAAI,aAAa,IAAI,KAAK,IAAI,aAAa,EAAE,CAAC;YACxE,MAAM,IAAI,GAAG,aAAiC,CAAC;YAC/C,KAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAC5C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YACjC,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACrE,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAC5C,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACrE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/mlx-ml/types.d.ts

@@ -26,6 +26,16 @@ export interface MlxMlModelOptions {
     [key: string]: number | undefined;
 }
 export type { ChatRestrictions, ApiStrategy };
+/**
+ * ツール呼び出しフォーマット
+ */
+export interface ToolCallFormat {
+    toolParserType?: string;
+    callStart?: string;
+    callEnd?: string;
+    responseStart?: string;
+    responseEnd?: string;
+}
 /**
  * チャットテンプレート情報
  */
@@ -34,6 +44,7 @@ export interface ChatTemplateInfo {
     supportedRoles: string[];
     preview?: string;
     constraints: Record<string, unknown>;
+    toolCallFormat?: ToolCallFormat;
 }
 /**
  * モデルの機能情報

package/dist/mlx-ml/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/mlx-ml/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC5E,OAAO,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAE3E;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACnC;AAGD,YAAY,EAAE,gBAAgB,EAAE,WAAW,EAAE,CAAC;AAE9C;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,wBAAwB;IACxB,eAAe,EAAE,OAAO,CAAC;IAEzB,YAAY;IACZ,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB,cAAc;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,mBAAmB;IACnB,YAAY,CAAC,EAAE,gBAAgB,CAAC;CACjC;AAED;;;;GAIG;AACH,MAAM,WAAW,oBAAoB;IACnC,kBAAkB;IAClB,OAAO,EAAE,MAAM,EAAE,CAAC;IAElB,aAAa;IACb,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,gBAAgB,CAAC,CAAC;IAE/D,aAAa;IACb,QAAQ,EAAE,aAAa,CAAC;IAExB,iCAAiC;IACjC,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;CACrC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/mlx-ml/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC5E,OAAO,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAE3E;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACnC;AAGD,YAAY,EAAE,gBAAgB,EAAE,WAAW,EAAE,CAAC;AAE9C;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,wBAAwB;IACxB,eAAe,EAAE,OAAO,CAAC;IAEzB,YAAY;IACZ,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB,cAAc;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,mBAAmB;IACnB,YAAY,CAAC,EAAE,gBAAgB,CAAC;CACjC;AAED;;;;GAIG;AACH,MAAM,WAAW,oBAAoB;IACnC,kBAAkB;IAClB,OAAO,EAAE,MAAM,EAAE,CAAC;IAElB,aAAa;IACb,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,gBAAgB,CAAC,CAAC;IAE/D,aAAa;IACb,QAAQ,EAAE,aAAa,CAAC;IAExB,iCAAiC;IACjC,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;CACrC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@modular-prompt/driver",
-  "version": "0.6.3",
+  "version": "0.8.1",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -13,6 +13,7 @@
   "files": [
     "dist",
     "scripts",
+    "skills",
     "src/mlx-ml/python"
   ],
   "dependencies": {
@@ -23,8 +24,8 @@
     "google-auth-library": "^9.15.1",
     "js-yaml": "^4.1.0",
     "openai": "^5.19.1",
-    "@modular-prompt/core": "0.1.12",
-    "@modular-prompt/utils": "0.2.3"
+    "@modular-prompt/core": "0.1.13",
+    "@modular-prompt/utils": "0.2.4"
   },
   "devDependencies": {
     "@eslint/js": "^9.35.0",
@@ -54,7 +55,8 @@
     "download-model": "node scripts/download-model.js",
     "lint": "eslint src/**/*.ts",
     "typecheck": "tsc --noEmit",
-    "clean": "rm -rf dist tsconfig.tsbuildinfo",
+    "copy-skills": "mkdir -p skills/driver-usage && cp ../../skills/driver-usage/SKILL.md skills/driver-usage/SKILL.md",
+    "clean": "rm -rf dist skills tsconfig.tsbuildinfo",
     "postinstall": "node scripts/setup-mlx.js || true",
     "setup-mlx": "node scripts/setup-mlx.js"
   }

package/skills/driver-usage/SKILL.md

@@ -0,0 +1,432 @@
+---
+name: driver-usage
+description: modular-promptのドライバー（AIDriver）の使い方ガイド。各ドライバーの初期化、Config、query/streamQuery、ツール定義、構造化出力、AIServiceによるモデル選択を参照する。
+---
+
+# ドライバー使い方ガイド
+
+## ドライバーとは
+
+`@modular-prompt/driver` は、コンパイル済みプロンプト（CompiledPrompt）をAIモデルに送信し、結果を受け取るための統一インターフェースを提供する。各AIサービスのAPI差異をドライバー層が吸収するため、プロンプト側のコードを変えずにモデルを切り替えられる。
+
+### 基本的な使い方
+
+```typescript
+import { compile } from '@modular-prompt/core';
+import { OpenAIDriver } from '@modular-prompt/driver';
+
+const driver = new OpenAIDriver({ model: 'gpt-4o' });
+const compiled = compile(myModule, context);
+
+// 通常クエリ
+const result = await driver.query(compiled);
+console.log(result.content);
+
+// ストリーミング
+const { stream, result: resultPromise } = await driver.streamQuery(compiled);
+for await (const chunk of stream) {
+  process.stdout.write(chunk);
+}
+const finalResult = await resultPromise;
+
+await driver.close();
+```
+
+## AIDriver インターフェース
+
+全ドライバーが実装する共通インターフェース:
+
+```typescript
+interface AIDriver {
+  query(prompt: CompiledPrompt, options?: QueryOptions): Promise<QueryResult>;
+  streamQuery(prompt: CompiledPrompt, options?: QueryOptions): Promise<StreamResult>;
+  close(): Promise<void>;
+}
+```
+
+### QueryOptions
+
+```typescript
+interface QueryOptions {
+  temperature?: number;
+  maxTokens?: number;
+  topP?: number;
+  stream?: boolean;
+  tools?: ToolDefinition[];
+  toolChoice?: ToolChoice;
+}
+```
+
+### QueryResult
+
+```typescript
+interface QueryResult {
+  content: string;               // テキストレスポンス
+  structuredOutput?: unknown;    // 構造化出力（schema指定時）
+  usage?: {
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+  };
+  toolCalls?: ToolCall[];        // ツール呼び出し
+  finishReason?: FinishReason;   // 'stop' | 'length' | 'error' | 'tool_calls'
+}
+```
+
+### StreamResult
+
+```typescript
+interface StreamResult {
+  stream: AsyncIterable<string>;  // テキストチャンクのストリーム
+  result: Promise<QueryResult>;   // 最終結果（ストリーム完了後に解決）
+}
+```
+
+## 各ドライバーのConfig
+
+### OpenAIDriver
+
+```typescript
+import { OpenAIDriver } from '@modular-prompt/driver';
+
+const driver = new OpenAIDriver({
+  apiKey: process.env.OPENAI_API_KEY,  // 環境変数で代替可
+  model: 'gpt-4o-mini',               // デフォルト: 'gpt-4o-mini'
+  baseURL: 'https://...',             // カスタムエンドポイント（オプション）
+  organization: '...',                // Organization ID（オプション）
+  defaultOptions: {
+    temperature: 0.7,
+    maxTokens: 2000,
+    frequencyPenalty: 0,               // OpenAI固有
+    presencePenalty: 0,                // OpenAI固有
+    stop: ['---'],                     // 停止シーケンス
+    responseFormat: { type: 'json_object' },
+    seed: 42
+  }
+});
+```
+
+### AnthropicDriver
+
+```typescript
+import { AnthropicDriver } from '@modular-prompt/driver';
+
+const driver = new AnthropicDriver({
+  apiKey: process.env.ANTHROPIC_API_KEY,  // 環境変数で代替可
+  model: 'claude-3-5-sonnet-20241022',    // デフォルト
+  defaultOptions: {
+    maxTokens: 4096,
+    temperature: 0.7,
+    topK: 40,                              // Anthropic固有
+    stopSequences: ['---']
+  }
+});
+```
+
+### VertexAIDriver
+
+```typescript
+import { VertexAIDriver } from '@modular-prompt/driver';
+
+const driver = new VertexAIDriver({
+  project: 'my-gcp-project',     // 環境変数 GOOGLE_CLOUD_PROJECT で代替可
+  location: 'us-central1',       // デフォルト: 'us-central1'
+  model: 'gemini-2.0-flash-001', // デフォルト
+  temperature: 0.05,
+  defaultOptions: {
+    maxTokens: 1000,
+    topP: 0.95,
+    topK: 40
+  }
+});
+```
+
+Google Cloud認証（ADCまたはサービスアカウント）が必要。
+
+### GoogleGenAIDriver
+
+```typescript
+import { GoogleGenAIDriver } from '@modular-prompt/driver';
+
+const driver = new GoogleGenAIDriver({
+  apiKey: process.env.GOOGLE_GENAI_API_KEY,  // 必須
+  model: 'gemini-2.0-flash-exp',
+  temperature: 0.7,
+  defaultOptions: {
+    maxTokens: 2048,
+    topP: 0.95,
+    topK: 40,
+    thinkingConfig: { thinkingLevel: 'HIGH' }  // GoogleGenAI固有
+  }
+});
+```
+
+APIキーのみで利用可能（Google AI Studioから取得）。
+
+### OllamaDriver
+
+```typescript
+import { OllamaDriver } from '@modular-prompt/driver';
+
+const driver = new OllamaDriver({
+  baseURL: 'http://localhost:11434/v1',  // デフォルト
+  model: 'llama3.2'                      // デフォルト
+});
+```
+
+OpenAI互換APIでローカルLLMにアクセス。
+
+### MlxDriver
+
+```typescript
+import { MlxDriver } from '@modular-prompt/driver';
+
+const driver = new MlxDriver({
+  model: 'mlx-community/Llama-3.2-3B-Instruct-4bit',  // 必須
+  defaultOptions: {
+    temperature: 0.7,
+    maxTokens: 500,
+    repetitionPenalty: 1.1,     // MLX固有
+    repetitionContextSize: 20   // MLX固有
+  }
+});
+
+// 使用後は必ずclose()（Pythonサブプロセス終了）
+await driver.close();
+```
+
+Apple Silicon専用。Python 3.11以上が必要。
+
+### テスト・デバッグ用ドライバー
+
+```typescript
+import { TestDriver, EchoDriver } from '@modular-prompt/driver';
+
+// TestDriver: モックレスポンス
+const testDriver = new TestDriver({
+  responses: ['応答1', '応答2'],    // キューから順に返す
+  delay: 100                        // レイテンシのシミュレート（ms）
+});
+
+// レスポンスプロバイダ関数
+const testDriver2 = new TestDriver({
+  responses: (prompt, options) => {
+    if (prompt.metadata?.outputSchema) {
+      return JSON.stringify({ result: 'ok' });
+    }
+    return 'テキスト応答';
+  }
+});
+
+// EchoDriver: フォーマット済みプロンプトをそのまま返す（AI呼び出しなし）
+const echoDriver = new EchoDriver({
+  format: 'debug',        // 'text' | 'messages' | 'raw' | 'both' | 'debug'
+  includeMetadata: true
+});
+```
+
+## ツール定義（Function Calling）
+
+### ToolDefinition
+
+```typescript
+const tools: ToolDefinition[] = [
+  {
+    name: 'get_weather',
+    description: '指定都市の天気を取得',
+    parameters: {
+      type: 'object',
+      properties: {
+        city: { type: 'string', description: '都市名' },
+        unit: { type: 'string', enum: ['celsius', 'fahrenheit'] }
+      },
+      required: ['city']
+    }
+  }
+];
+```
+
+### ToolChoice
+
+```typescript
+type ToolChoice =
+  | 'auto'             // モデルが自動判断（デフォルト）
+  | 'none'             // ツール使用禁止
+  | 'required'         // 必ず1つ以上のツールを使用
+  | { name: string };  // 特定ツールを強制
+```
+
+### ツール呼び出しの処理
+
+```typescript
+const result = await driver.query(compiled, { tools, toolChoice: 'auto' });
+
+if (result.toolCalls) {
+  for (const call of result.toolCalls) {
+    console.log(call.name);       // 関数名
+    console.log(call.id);         // 呼び出しID
+    console.log(call.arguments);  // 引数オブジェクト
+  }
+}
+```
+
+対応ドライバー: OpenAI、Anthropic、VertexAI、GoogleGenAI
+
+### ツール結果の返し方（会話ループ）
+
+ツール呼び出し結果をモデルに返す会話ループは利用者側で実装する。`QueryOptions.messages` にツール結果を含めて再クエリする。
+
+```typescript
+const result1 = await driver.query(compiled, { tools, toolChoice: 'auto' });
+
+if (result1.toolCalls) {
+  // ツールを実行して結果を収集
+  const toolResults = await Promise.all(
+    result1.toolCalls.map(async (tc) => {
+      const data = await executeFunction(tc.name, tc.arguments);
+      return {
+        role: 'tool' as const,
+        toolCallId: tc.id,
+        name: tc.name,
+        kind: 'data' as const,   // 'text' | 'data' | 'error'
+        value: data
+      };
+    })
+  );
+
+  // ツール結果を含めて再クエリ
+  const result2 = await driver.query(compiled, {
+    tools,
+    messages: [
+      { role: 'assistant', content: result1.content, toolCalls: result1.toolCalls },
+      ...toolResults
+    ]
+  });
+}
+```
+
+### ToolResultKind
+
+ツール結果の種類を示すタグ:
+- `'text'` - プレーンテキスト
+- `'data'` - 構造化データ（オブジェクト等）
+- `'error'` - エラー情報
+
+## 構造化出力
+
+プロンプトの `schema` セクションに JSONElement を定義すると、ドライバーが自動的に構造化出力を処理する。
+
+```typescript
+const myModule: PromptModule = {
+  objective: ['ユーザー情報を抽出する'],
+  schema: [{
+    type: 'json',
+    content: {
+      type: 'object',
+      properties: {
+        name: { type: 'string' },
+        age: { type: 'number' }
+      },
+      required: ['name', 'age']
+    }
+  }]
+};
+
+const result = await driver.query(compile(myModule, ctx));
+const data = result.structuredOutput as { name: string; age: number };
+```
+
+ドライバーごとの実装方式:
+- **ネイティブサポート**: OpenAI（`response_format`）、VertexAI / GoogleGenAI（`responseSchema`）
+- **JSON抽出型**: Anthropic、MLX（プロンプト指示 + レスポンスからJSON抽出）
+
+## AIService（モデル選択）
+
+複数モデルを登録し、能力（capabilities）ベースで最適なモデルを自動選択する。
+
+### 設定
+
+```typescript
+import { AIService } from '@modular-prompt/driver';
+
+const service = new AIService({
+  models: [
+    {
+      model: 'gpt-4o',
+      provider: 'openai',
+      capabilities: ['streaming', 'japanese', 'tools', 'structured'],
+      priority: 10,
+      cost: { input: 0.01, output: 0.03 }
+    },
+    {
+      model: 'claude-3-5-sonnet-20241022',
+      provider: 'anthropic',
+      capabilities: ['streaming', 'japanese', 'tools', 'reasoning'],
+      priority: 8
+    }
+  ],
+  drivers: {
+    openai: { apiKey: process.env.OPENAI_API_KEY },
+    anthropic: { apiKey: process.env.ANTHROPIC_API_KEY }
+  },
+  defaultOptions: {
+    temperature: 0.7,
+    maxTokens: 2048
+  }
+});
+```
+
+### ModelSpec
+
+```typescript
+interface ModelSpec {
+  model: string;
+  provider: DriverProvider;
+  capabilities: DriverCapability[];
+  priority?: number;              // 高いほど優先
+  disabled?: boolean;             // 無効化フラグ
+  maxInputTokens?: number;
+  maxOutputTokens?: number;
+  maxTotalTokens?: number;
+  tokensPerMinute?: number;       // TPM制限
+  requestsPerMinute?: number;     // RPM制限
+  cost?: { input: number; output: number };
+  metadata?: Record<string, unknown>;
+}
+```
+
+### DriverCapability（能力フラグ）
+
+| 能力 | 説明 |
+|------|------|
+| `streaming` | ストリーミング応答 |
+| `local` | ローカル実行 |
+| `fast` | 高速応答 |
+| `large-context` | 大規模コンテキスト |
+| `multilingual` | 多言語対応 |
+| `japanese` | 日本語特化 |
+| `coding` | コーディング特化 |
+| `reasoning` | 推論・思考特化 |
+| `chat` | チャット特化 |
+| `tools` | ツール使用 |
+| `vision` | 画像認識 |
+| `audio` | 音声処理 |
+| `structured` | 構造化出力 |
+| `json` | JSON出力 |
+| `function-calling` | 関数呼び出し |
+
+### モデル選択
+
+```typescript
+// 能力ベースでドライバーを自動作成
+const driver = await service.createDriverFromCapabilities(
+  ['japanese', 'streaming'],
+  {
+    preferLocal: true,           // ローカル優先
+    preferProvider: 'anthropic', // 特定プロバイダー優先
+    excludeProviders: ['openai'],
+    preferFast: true,            // 高速優先
+    lenient: true                // 条件緩和モード（条件を後ろから減らして再検索）
+  }
+);
+```