@modular-prompt/driver 0.4.5

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

@@ -0,0 +1,71 @@
+import type { CompiledPrompt } from '@modular-prompt/core';
+import type { AIDriver, QueryOptions, QueryResult, StreamResult } from '../types.js';
+/**
+ * OpenAI driver configuration
+ */
+export interface OpenAIDriverConfig {
+    apiKey?: string;
+    baseURL?: string;
+    model?: string;
+    organization?: string;
+    defaultOptions?: Partial<OpenAIQueryOptions>;
+}
+/**
+ * OpenAI-specific query options
+ */
+export interface OpenAIQueryOptions extends QueryOptions {
+    model?: string;
+    temperature?: number;
+    maxTokens?: number;
+    topP?: number;
+    frequencyPenalty?: number;
+    presencePenalty?: number;
+    stop?: string | string[];
+    n?: number;
+    logprobs?: boolean;
+    topLogprobs?: number;
+    responseFormat?: {
+        type: 'json_object' | 'text';
+    };
+    seed?: number;
+    tools?: Array<{
+        type: 'function';
+        function: {
+            name: string;
+            description?: string;
+            parameters?: Record<string, unknown>;
+        };
+    }>;
+    toolChoice?: 'none' | 'auto' | {
+        type: 'function';
+        function: {
+            name: string;
+        };
+    };
+}
+/**
+ * OpenAI API driver
+ */
+export declare class OpenAIDriver implements AIDriver {
+    private client;
+    private defaultModel;
+    private defaultOptions;
+    constructor(config?: OpenAIDriverConfig);
+    /**
+     * Convert CompiledPrompt to OpenAI messages
+     */
+    private compiledPromptToMessages;
+    /**
+     * Query the AI model
+     */
+    query(prompt: CompiledPrompt, options?: QueryOptions): Promise<QueryResult>;
+    /**
+     * Stream query implementation with both stream and result
+     */
+    streamQuery(prompt: CompiledPrompt, options?: QueryOptions): Promise<StreamResult>;
+    /**
+     * Close the client
+     */
+    close(): Promise<void>;
+}
+//# sourceMappingURL=openai-driver.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai-driver.d.ts","sourceRoot":"","sources":["../../src/openai/openai-driver.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAW,MAAM,sBAAsB,CAAC;AACpE,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAMrF;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,cAAc,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC,CAAC;CAC9C;AAED;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,YAAY;IACtD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IACzB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE;QAAE,IAAI,EAAE,aAAa,GAAG,MAAM,CAAA;KAAE,CAAC;IAClD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,UAAU,CAAC;QACjB,QAAQ,EAAE;YACR,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,CAAC,EAAE,MAAM,CAAC;YACrB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;SACtC,CAAC;KACH,CAAC,CAAC;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;CACjF;AAED;;GAEG;AACH,qBAAa,YAAa,YAAW,QAAQ;IAC3C,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,cAAc,CAA8B;gBAExC,MAAM,GAAE,kBAAuB;IAW3C;;OAEG;IACH,OAAO,CAAC,wBAAwB;IA8EhC;;OAEG;IACG,KAAK,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAMjF;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IA4HxF;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG7B"}

package/dist/openai/openai-driver.js

@@ -0,0 +1,230 @@
+import OpenAI from 'openai';
+/**
+ * OpenAI API driver
+ */
+export class OpenAIDriver {
+    client;
+    defaultModel;
+    defaultOptions;
+    constructor(config = {}) {
+        this.client = new OpenAI({
+            apiKey: config.apiKey || process.env.OPENAI_API_KEY,
+            baseURL: config.baseURL,
+            organization: config.organization
+        });
+        this.defaultModel = config.model || 'gpt-4o-mini';
+        this.defaultOptions = config.defaultOptions || {};
+    }
+    /**
+     * Convert CompiledPrompt to OpenAI messages
+     */
+    compiledPromptToMessages(prompt) {
+        const messages = [];
+        // Helper to process elements
+        const processElements = (elements) => {
+            const content = [];
+            for (const element of elements) {
+                if (typeof element === 'string') {
+                    content.push(element);
+                }
+                else if (typeof element === 'object' && element !== null && 'type' in element) {
+                    const el = element;
+                    if (el.type === 'text') {
+                        content.push(el.content);
+                    }
+                    else if (el.type === 'message') {
+                        // Handle message elements separately
+                        messages.push({
+                            role: el.role,
+                            content: typeof el.content === 'string' ? el.content : JSON.stringify(el.content)
+                        });
+                    }
+                    else if (el.type === 'section' || el.type === 'subsection') {
+                        // Process section content
+                        if (el.title)
+                            content.push(`## ${el.title}`);
+                        if (el.items) {
+                            for (const item of el.items) {
+                                if (typeof item === 'string') {
+                                    content.push(item);
+                                }
+                                else if (typeof item === 'object' && item !== null && 'type' in item && item.type === 'subsection') {
+                                    if (item.title)
+                                        content.push(`### ${item.title}`);
+                                    if ('items' in item && item.items) {
+                                        content.push(...item.items.filter((i) => typeof i === 'string'));
+                                    }
+                                }
+                            }
+                        }
+                    }
+                    else {
+                        // Default formatting for other elements
+                        content.push(JSON.stringify(el));
+                    }
+                }
+            }
+            return content.join('\n');
+        };
+        // Process instructions as system message
+        if (prompt.instructions && prompt.instructions.length > 0) {
+            const instructionContent = processElements(prompt.instructions);
+            if (instructionContent) {
+                messages.push({ role: 'system', content: instructionContent });
+            }
+        }
+        // Process data as user message
+        if (prompt.data && prompt.data.length > 0) {
+            const dataContent = processElements(prompt.data);
+            if (dataContent) {
+                messages.push({ role: 'user', content: dataContent });
+            }
+        }
+        // Process output as user message (continuation)
+        if (prompt.output && prompt.output.length > 0) {
+            const outputContent = processElements(prompt.output);
+            if (outputContent) {
+                messages.push({ role: 'user', content: outputContent });
+            }
+        }
+        // Ensure at least one message
+        if (messages.length === 0) {
+            messages.push({ role: 'user', content: 'Please respond.' });
+        }
+        return messages;
+    }
+    /**
+     * Query the AI model
+     */
+    async query(prompt, options) {
+        // Use streamQuery for consistency
+        const { result } = await this.streamQuery(prompt, options);
+        return result;
+    }
+    /**
+     * Stream query implementation with both stream and result
+     */
+    async streamQuery(prompt, options) {
+        try {
+            const openaiOptions = options || {};
+            const mergedOptions = { ...this.defaultOptions, ...openaiOptions };
+            const messages = this.compiledPromptToMessages(prompt);
+            const params = {
+                model: mergedOptions.model || this.defaultModel,
+                messages,
+                temperature: mergedOptions.temperature,
+                max_tokens: mergedOptions.maxTokens,
+                top_p: mergedOptions.topP,
+                frequency_penalty: mergedOptions.frequencyPenalty,
+                presence_penalty: mergedOptions.presencePenalty,
+                stop: mergedOptions.stop,
+                response_format: prompt.metadata?.outputSchema ? { type: 'json_object' } : undefined,
+                stream: true
+            };
+            // Remove undefined values
+            Object.keys(params).forEach(key => {
+                if (params[key] === undefined) {
+                    delete params[key];
+                }
+            });
+            const openaiStream = await this.client.chat.completions.create(params);
+            // Shared state for accumulating content and metadata
+            let fullContent = '';
+            let usage;
+            let finishReason = 'stop';
+            let streamConsumed = false;
+            const chunks = [];
+            // Process the OpenAI stream and cache chunks
+            const processStream = async () => {
+                try {
+                    for await (const chunk of openaiStream) {
+                        const content = chunk.choices[0]?.delta?.content;
+                        if (content) {
+                            fullContent += content;
+                            chunks.push(content);
+                        }
+                        // Update finish reason if provided
+                        if (chunk.choices[0]?.finish_reason) {
+                            const reason = chunk.choices[0].finish_reason;
+                            if (reason === 'length') {
+                                finishReason = 'length';
+                            }
+                            else if (reason === 'content_filter') {
+                                finishReason = 'error';
+                            }
+                        }
+                        // Accumulate usage if provided (usually in the final chunk)
+                        if (chunk.usage) {
+                            usage = {
+                                promptTokens: chunk.usage.prompt_tokens,
+                                completionTokens: chunk.usage.completion_tokens,
+                                totalTokens: chunk.usage.total_tokens
+                            };
+                        }
+                    }
+                }
+                catch {
+                    finishReason = 'error';
+                }
+                streamConsumed = true;
+            };
+            // Start processing the stream
+            const processingPromise = processStream();
+            // Create the stream generator that yields cached chunks
+            const streamGenerator = async function* () {
+                let index = 0;
+                while (!streamConsumed || index < chunks.length) {
+                    if (index < chunks.length) {
+                        yield chunks[index++];
+                    }
+                    else {
+                        // Wait a bit for more chunks
+                        await new Promise(resolve => setTimeout(resolve, 10));
+                    }
+                }
+            };
+            // Create result promise
+            const resultPromise = processingPromise.then(() => {
+                // If response_format was used, the content should already be JSON
+                let structuredOutput;
+                if (prompt.metadata?.outputSchema && params.response_format?.type === 'json_object') {
+                    try {
+                        structuredOutput = JSON.parse(fullContent);
+                    }
+                    catch {
+                        // Keep as undefined if parsing fails
+                    }
+                }
+                return {
+                    content: fullContent,
+                    structuredOutput,
+                    usage,
+                    finishReason
+                };
+            });
+            return {
+                stream: streamGenerator(),
+                result: resultPromise
+            };
+        }
+        catch {
+            // Return error state
+            return {
+                stream: (async function* () {
+                    // Empty stream
+                })(),
+                result: Promise.resolve({
+                    content: '',
+                    finishReason: 'error'
+                })
+            };
+        }
+    }
+    /**
+     * Close the client
+     */
+    async close() {
+        // OpenAI client doesn't need explicit closing
+    }
+}
+//# sourceMappingURL=openai-driver.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai-driver.js","sourceRoot":"","sources":["../../src/openai/openai-driver.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AA8C5B;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,MAAM,CAAS;IACf,YAAY,CAAS;IACrB,cAAc,CAA8B;IAEpD,YAAY,SAA6B,EAAE;QACzC,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;YACnD,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,YAAY,EAAE,MAAM,CAAC,YAAY;SAClC,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,KAAK,IAAI,aAAa,CAAC;QAClD,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,wBAAwB,CAAC,MAAsB;QACrD,MAAM,QAAQ,GAAiC,EAAE,CAAC;QAElD,6BAA6B;QAC7B,MAAM,eAAe,GAAG,CAAC,QAAmB,EAAE,EAAE;YAC9C,MAAM,OAAO,GAAa,EAAE,CAAC;YAE7B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAChC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACxB,CAAC;qBAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,IAAI,IAAI,MAAM,IAAI,OAAO,EAAE,CAAC;oBAChF,MAAM,EAAE,GAAG,OAAkB,CAAC;oBAE9B,IAAI,EAAE,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACvB,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;oBAC3B,CAAC;yBAAM,IAAI,EAAE,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;wBACjC,qCAAqC;wBACrC,QAAQ,CAAC,IAAI,CAAC;4BACZ,IAAI,EAAE,EAAE,CAAC,IAAuC;4BAChD,OAAO,EAAE,OAAO,EAAE,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,CAAC;yBAClF,CAAC,CAAC;oBACL,CAAC;yBAAM,IAAI,EAAE,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBAC7D,0BAA0B;wBAC1B,IAAI,EAAE,CAAC,KAAK;4BAAE,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;wBAC7C,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC;4BACb,KAAK,MAAM,IAAI,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC;gCAC5B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oCAC7B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gCACrB,CAAC;qCAAM,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oCACrG,IAAI,IAAI,CAAC,KAAK;wCAAE,OAAO,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;oCAClD,IAAI,OAAO,IAAI,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;wCAClC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC;oCACnE,CAAC;gCACH,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,wCAAwC;wBACxC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;oBACnC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC;QAEF,yCAAyC;QACzC,IAAI,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,MAAM,kBAAkB,GAAG,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAChE,IAAI,kBAAkB,EAAE,CAAC;gBACvB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,+BAA+B;QAC/B,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,WAAW,EAAE,CAAC;gBAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAED,gDAAgD;QAChD,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACrD,IAAI,aAAa,EAAE,CAAC;gBAClB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QAED,8BAA8B;QAC9B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,MAAsB,EAAE,OAAsB;QACxD,kCAAkC;QAClC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,MAAsB,EAAE,OAAsB;QAC9D,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,OAA6B,IAAI,EAAE,CAAC;YAC1D,MAAM,aAAa,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,GAAG,aAAa,EAAE,CAAC;YACnE,MAAM,QAAQ,GAAG,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;YAEvD,MAAM,MAAM,GAA+B;gBACzC,KAAK,EAAE,aAAa,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY;gBAC/C,QAAQ;gBACR,WAAW,EAAE,aAAa,CAAC,WAAW;gBACtC,UAAU,EAAE,aAAa,CAAC,SAAS;gBACnC,KAAK,EAAE,aAAa,CAAC,IAAI;gBACzB,iBAAiB,EAAE,aAAa,CAAC,gBAAgB;gBACjD,gBAAgB,EAAE,aAAa,CAAC,eAAe;gBAC/C,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,eAAe,EAAE,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;gBACpF,MAAM,EAAE,IAAI;aACb,CAAC;YAEF,0BAA0B;YAC1B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;gBAChC,IAAI,MAAM,CAAC,GAA0B,CAAC,KAAK,SAAS,EAAE,CAAC;oBACrD,OAAO,MAAM,CAAC,GAA0B,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEvE,qDAAqD;YACrD,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,KAAuC,CAAC;YAC5C,IAAI,YAAY,GAAgC,MAAM,CAAC;YACvD,IAAI,cAAc,GAAG,KAAK,CAAC;YAC3B,MAAM,MAAM,GAAa,EAAE,CAAC;YAE5B,6CAA6C;YAC7C,MAAM,aAAa,GAAG,KAAK,IAAI,EAAE;gBAC/B,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC;wBACjD,IAAI,OAAO,EAAE,CAAC;4BACZ,WAAW,IAAI,OAAO,CAAC;4BACvB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;wBACvB,CAAC;wBAED,mCAAmC;wBACnC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;4BACpC,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;4BAC9C,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gCACxB,YAAY,GAAG,QAAQ,CAAC;4BAC1B,CAAC;iCAAM,IAAI,MAAM,KAAK,gBAAgB,EAAE,CAAC;gCACvC,YAAY,GAAG,OAAO,CAAC;4BACzB,CAAC;wBACH,CAAC;wBAED,4DAA4D;wBAC5D,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;4BAChB,KAAK,GAAG;gCACN,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,aAAa;gCACvC,gBAAgB,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB;gCAC/C,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;6BACtC,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,YAAY,GAAG,OAAO,CAAC;gBACzB,CAAC;gBACD,cAAc,GAAG,IAAI,CAAC;YACxB,CAAC,CAAC;YAEF,8BAA8B;YAC9B,MAAM,iBAAiB,GAAG,aAAa,EAAE,CAAC;YAE1C,wDAAwD;YACxD,MAAM,eAAe,GAAG,KAAK,SAAS,CAAC;gBACrC,IAAI,KAAK,GAAG,CAAC,CAAC;gBACd,OAAO,CAAC,cAAc,IAAI,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;oBAChD,IAAI,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;wBAC1B,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;oBACxB,CAAC;yBAAM,CAAC;wBACN,6BAA6B;wBAC7B,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC;YACH,CAAC,CAAC;YAEF,wBAAwB;YACxB,MAAM,aAAa,GAAG,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE;gBAChD,kEAAkE;gBAClE,IAAI,gBAAqC,CAAC;gBAC1C,IAAI,MAAM,CAAC,QAAQ,EAAE,YAAY,IAAI,MAAM,CAAC,eAAe,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;oBACpF,IAAI,CAAC;wBACH,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;oBAC7C,CAAC;oBAAC,MAAM,CAAC;wBACP,qCAAqC;oBACvC,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,OAAO,EAAE,WAAW;oBACpB,gBAAgB;oBAChB,KAAK;oBACL,YAAY;iBACb,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,MAAM,EAAE,eAAe,EAAE;gBACzB,MAAM,EAAE,aAAa;aACtB,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,qBAAqB;YACrB,OAAO;gBACL,MAAM,EAAE,CAAC,KAAK,SAAS,CAAC;oBACtB,eAAe;gBACjB,CAAC,CAAC,EAAE;gBACJ,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC;oBACtB,OAAO,EAAE,EAAE;oBACX,YAAY,EAAE,OAAgB;iBAC/B,CAAC;aACH,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,8CAA8C;IAChD,CAAC;CACF"}

package/dist/test-driver.d.ts

@@ -0,0 +1,78 @@
+import type { CompiledPrompt } from '@modular-prompt/core';
+import type { AIDriver, QueryOptions, QueryResult, StreamResult } from './types.js';
+/**
+ * Mock response configuration
+ */
+export interface MockResponse {
+    content: string;
+    finishReason?: 'stop' | 'length' | 'error';
+}
+/**
+ * Response provider function type
+ * A function that generates responses dynamically based on the prompt and options.
+ * Called each time query() or streamQuery() is invoked.
+ *
+ * @param prompt - The compiled prompt being executed
+ * @param options - Query options (temperature, maxTokens, etc.)
+ * @returns The response string or MockResponse object
+ */
+export type ResponseProvider = (prompt: CompiledPrompt, options?: QueryOptions) => string | MockResponse | Promise<string | MockResponse>;
+/**
+ * Test driver options
+ */
+export interface TestDriverOptions {
+    /**
+     * Mock responses for the driver.
+     * Can be either:
+     * - An array of strings: responses are consumed sequentially (queue pattern)
+     * - An array of MockResponse objects: responses with finishReason control
+     * - A function: called for each query to generate dynamic responses
+     */
+    responses?: (string | MockResponse)[] | ResponseProvider;
+    /**
+     * Delay in milliseconds to simulate API latency.
+     * Applied before returning each response.
+     */
+    delay?: number;
+}
+/**
+ * Test driver for unit testing and mocking AI responses.
+ *
+ * @example
+ * // Using response queue (array)
+ * const driver = new TestDriver({
+ *   responses: ['First', 'Second', 'Third']
+ * });
+ * await driver.query(prompt); // Returns 'First'
+ * await driver.query(prompt); // Returns 'Second'
+ * await driver.query(prompt); // Returns 'Third'
+ * await driver.query(prompt); // Throws: No more responses available
+ *
+ * @example
+ * // Using response provider (function)
+ * const driver = new TestDriver({
+ *   responses: (prompt, options) => {
+ *     if (prompt.metadata?.outputSchema) {
+ *       return JSON.stringify({ result: 'structured' });
+ *     }
+ *     return 'Plain text';
+ *   }
+ * });
+ */
+export declare class TestDriver implements AIDriver {
+    /** Queue of responses when using array mode */
+    private responseQueue;
+    /** Function to generate responses dynamically */
+    private responseProvider?;
+    /** Delay in milliseconds to simulate latency */
+    private delay;
+    constructor(options?: TestDriverOptions);
+    query(prompt: CompiledPrompt, options?: QueryOptions): Promise<QueryResult>;
+    /**
+     * Stream a response character by character.
+     * Consumes one response from the queue or calls the provider function.
+     */
+    streamQuery(prompt: CompiledPrompt, options?: QueryOptions): Promise<StreamResult>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=test-driver.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"test-driver.d.ts","sourceRoot":"","sources":["../src/test-driver.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAGpF;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,GAAG,QAAQ,GAAG,OAAO,CAAC;CAC5C;AAED;;;;;;;;GAQG;AACH,MAAM,MAAM,gBAAgB,GAAG,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,YAAY,KAAK,MAAM,GAAG,YAAY,GAAG,OAAO,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC;AAE1I;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,CAAC,MAAM,GAAG,YAAY,CAAC,EAAE,GAAG,gBAAgB,CAAC;IAEzD;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AASD;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,qBAAa,UAAW,YAAW,QAAQ;IACzC,+CAA+C;IAC/C,OAAO,CAAC,aAAa,CAAiB;IACtC,iDAAiD;IACjD,OAAO,CAAC,gBAAgB,CAAC,CAAmB;IAC5C,gDAAgD;IAChD,OAAO,CAAC,KAAK,CAAS;gBAEV,OAAO,GAAE,iBAAsB;IAgBrC,KAAK,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAiFjF;;;OAGG;IACG,WAAW,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAkElF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG7B"}

package/dist/test-driver.js

@@ -0,0 +1,193 @@
+import { extractJSON } from '@modular-prompt/utils';
+/**
+ * Simple token estimation (roughly 4 characters per token)
+ */
+function estimateTokens(text) {
+    return Math.ceil(text.length / 4);
+}
+/**
+ * Test driver for unit testing and mocking AI responses.
+ *
+ * @example
+ * // Using response queue (array)
+ * const driver = new TestDriver({
+ *   responses: ['First', 'Second', 'Third']
+ * });
+ * await driver.query(prompt); // Returns 'First'
+ * await driver.query(prompt); // Returns 'Second'
+ * await driver.query(prompt); // Returns 'Third'
+ * await driver.query(prompt); // Throws: No more responses available
+ *
+ * @example
+ * // Using response provider (function)
+ * const driver = new TestDriver({
+ *   responses: (prompt, options) => {
+ *     if (prompt.metadata?.outputSchema) {
+ *       return JSON.stringify({ result: 'structured' });
+ *     }
+ *     return 'Plain text';
+ *   }
+ * });
+ */
+export class TestDriver {
+    /** Queue of responses when using array mode */
+    responseQueue;
+    /** Function to generate responses dynamically */
+    responseProvider;
+    /** Delay in milliseconds to simulate latency */
+    delay;
+    constructor(options = {}) {
+        if (typeof options.responses === 'function') {
+            // Function mode: generate responses dynamically
+            this.responseProvider = options.responses;
+            this.responseQueue = [];
+        }
+        else {
+            // Array mode: use responses as a queue (FIFO)
+            // Normalize string responses to MockResponse objects
+            this.responseQueue = options.responses
+                ? options.responses.map(r => typeof r === 'string' ? { content: r, finishReason: 'stop' } : r)
+                : [];
+            this.responseProvider = undefined;
+        }
+        this.delay = options.delay || 0;
+    }
+    async query(prompt, options) {
+        // Create a simple formatted prompt for token counting
+        const formattedPrompt = [
+            prompt.instructions,
+            prompt.data,
+            prompt.output
+        ].filter(Boolean).join('\n\n');
+        // If we have a response provider function, use it
+        if (this.responseProvider) {
+            // Dynamic response generation mode
+            const response = await this.responseProvider(prompt, options);
+            // Normalize response to MockResponse
+            const mockResponse = typeof response === 'string'
+                ? { content: response, finishReason: 'stop' }
+                : response;
+            // Simulate API latency if configured
+            if (this.delay > 0) {
+                await new Promise(resolve => setTimeout(resolve, this.delay));
+            }
+            // Handle structured outputs if schema is provided
+            // This attempts to extract JSON from the response if outputSchema is defined
+            let structuredOutput;
+            if (prompt.metadata?.outputSchema && mockResponse.content) {
+                const extracted = extractJSON(mockResponse.content, { multiple: false });
+                if (extracted.source !== 'none' && extracted.data !== null) {
+                    structuredOutput = extracted.data;
+                }
+            }
+            return {
+                content: mockResponse.content,
+                structuredOutput,
+                usage: {
+                    promptTokens: estimateTokens(formattedPrompt),
+                    completionTokens: estimateTokens(mockResponse.content),
+                    totalTokens: estimateTokens(formattedPrompt) + estimateTokens(mockResponse.content)
+                },
+                finishReason: mockResponse.finishReason || 'stop'
+            };
+        }
+        // Queue mode: consume responses sequentially
+        if (this.responseQueue.length === 0) {
+            throw new Error('No more responses available');
+        }
+        // Simulate API latency if configured
+        if (this.delay > 0) {
+            await new Promise(resolve => setTimeout(resolve, this.delay));
+        }
+        // Take the first response from the queue (FIFO pattern)
+        const mockResponse = this.responseQueue.shift();
+        // Handle structured outputs if schema is provided
+        // This attempts to extract JSON from the response if outputSchema is defined
+        let structuredOutput;
+        if (prompt.metadata?.outputSchema && mockResponse.content) {
+            const extracted = extractJSON(mockResponse.content, { multiple: false });
+            if (extracted.source !== 'none' && extracted.data !== null) {
+                structuredOutput = extracted.data;
+            }
+        }
+        return {
+            content: mockResponse.content,
+            structuredOutput,
+            usage: {
+                promptTokens: estimateTokens(formattedPrompt),
+                completionTokens: estimateTokens(mockResponse.content),
+                totalTokens: estimateTokens(formattedPrompt) + estimateTokens(mockResponse.content)
+            },
+            finishReason: mockResponse.finishReason || 'stop'
+        };
+    }
+    /**
+     * Stream a response character by character.
+     * Consumes one response from the queue or calls the provider function.
+     */
+    async streamQuery(prompt, options) {
+        // Create a simple formatted prompt for token counting
+        const formattedPrompt = [
+            prompt.instructions,
+            prompt.data,
+            prompt.output
+        ].filter(Boolean).join('\n\n');
+        // Get the response (either from provider function or queue)
+        let mockResponse;
+        if (this.responseProvider) {
+            // Dynamic mode: generate response
+            const response = await this.responseProvider(prompt, options);
+            mockResponse = typeof response === 'string'
+                ? { content: response, finishReason: 'stop' }
+                : response;
+        }
+        else {
+            // Queue mode: consume one response
+            if (this.responseQueue.length === 0) {
+                throw new Error('No more responses available');
+            }
+            mockResponse = this.responseQueue.shift();
+        }
+        // Create stream generator that yields characters one by one
+        const delay = this.delay;
+        const content = mockResponse.content;
+        async function* streamGenerator() {
+            // Stream response character by character to simulate streaming
+            for (const char of content) {
+                if (delay > 0) {
+                    // Distribute delay across all characters
+                    await new Promise(resolve => setTimeout(resolve, delay / content.length));
+                }
+                yield char;
+            }
+        }
+        // Handle structured outputs if schema is provided
+        // Extract JSON from the complete response (not streamed)
+        let structuredOutput;
+        if (prompt.metadata?.outputSchema && mockResponse.content) {
+            const extracted = extractJSON(mockResponse.content, { multiple: false });
+            if (extracted.source !== 'none' && extracted.data !== null) {
+                structuredOutput = extracted.data;
+            }
+        }
+        // Create result promise
+        const resultPromise = Promise.resolve({
+            content: mockResponse.content,
+            structuredOutput,
+            usage: {
+                promptTokens: estimateTokens(formattedPrompt),
+                completionTokens: estimateTokens(mockResponse.content),
+                totalTokens: estimateTokens(formattedPrompt) + estimateTokens(mockResponse.content)
+            },
+            finishReason: mockResponse.finishReason || 'stop'
+        });
+        return {
+            stream: streamGenerator(),
+            result: resultPromise
+        };
+    }
+    async close() {
+        // No resources to clean up
+    }
+}
+//# sourceMappingURL=test-driver.js.map

package/dist/test-driver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-driver.js","sourceRoot":"","sources":["../src/test-driver.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAyCpD;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpC,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,OAAO,UAAU;IACrB,+CAA+C;IACvC,aAAa,CAAiB;IACtC,iDAAiD;IACzC,gBAAgB,CAAoB;IAC5C,gDAAgD;IACxC,KAAK,CAAS;IAEtB,YAAY,UAA6B,EAAE;QACzC,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;YAC5C,gDAAgD;YAChD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC;YAC1C,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,8CAA8C;YAC9C,qDAAqD;YACrD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,SAAS;gBACpC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,YAAY,EAAE,MAAe,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBACvG,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAsB,EAAE,OAAsB;QACxD,sDAAsD;QACtD,MAAM,eAAe,GAAG;YACtB,MAAM,CAAC,YAAY;YACnB,MAAM,CAAC,IAAI;YACX,MAAM,CAAC,MAAM;SACd,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE/B,kDAAkD;QAClD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,mCAAmC;YACnC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAE9D,qCAAqC;YACrC,MAAM,YAAY,GAAiB,OAAO,QAAQ,KAAK,QAAQ;gBAC7D,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE;gBAC7C,CAAC,CAAC,QAAQ,CAAC;YAEb,qCAAqC;YACrC,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;gBACnB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YAChE,CAAC;YAED,kDAAkD;YAClD,6EAA6E;YAC7E,IAAI,gBAAqC,CAAC;YAC1C,IAAI,MAAM,CAAC,QAAQ,EAAE,YAAY,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBAC1D,MAAM,SAAS,GAAG,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzE,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,OAAO;gBACL,OAAO,EAAE,YAAY,CAAC,OAAO;gBAC7B,gBAAgB;gBAChB,KAAK,EAAE;oBACL,YAAY,EAAE,cAAc,CAAC,eAAe,CAAC;oBAC7C,gBAAgB,EAAE,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC;oBACtD,WAAW,EAAE,cAAc,CAAC,eAAe,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC;iBACpF;gBACD,YAAY,EAAE,YAAY,CAAC,YAAY,IAAI,MAAM;aAClD,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,qCAAqC;QACrC,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAChE,CAAC;QAED,wDAAwD;QACxD,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAG,CAAC;QAEjD,kDAAkD;QAClD,6EAA6E;QAC7E,IAAI,gBAAqC,CAAC;QAC1C,IAAI,MAAM,CAAC,QAAQ,EAAE,YAAY,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;YAC1D,MAAM,SAAS,GAAG,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACzE,IAAI,SAAS,CAAC,MAAM,KAAK,MAAM,IAAI,SAAS,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC3D,gBAAgB,GAAG,SAAS,CAAC,IAAI,CAAC;YACpC,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,YAAY,CAAC,OAAO;YAC7B,gBAAgB;YAChB,KAAK,EAAE;gBACL,YAAY,EAAE,cAAc,CAAC,eAAe,CAAC;gBAC7C,gBAAgB,EAAE,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC;gBACtD,WAAW,EAAE,cAAc,CAAC,eAAe,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC;aACpF;YACD,YAAY,EAAE,YAAY,CAAC,YAAY,IAAI,MAAM;SAClD,CAAC;IACJ,CAAC;IAGD;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAC,MAAsB,EAAE,OAAsB;QAC9D,sDAAsD;QACtD,MAAM,eAAe,GAAG;YACtB,MAAM,CAAC,YAAY;YACnB,MAAM,CAAC,IAAI;YACX,MAAM,CAAC,MAAM;SACd,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE/B,4DAA4D;QAC5D,IAAI,YAA0B,CAAC;QAC/B,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,kCAAkC;YAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC9D,YAAY,GAAG,OAAO,QAAQ,KAAK,QAAQ;gBACzC,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE;gBAC7C,CAAC,CAAC,QAAQ,CAAC;QACf,CAAC;aAAM,CAAC;YACN,mCAAmC;YACnC,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;YACD,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAG,CAAC;QAC7C,CAAC;QAED,4DAA4D;QAC5D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;QACrC,KAAK,SAAS,CAAC,CAAC,eAAe;YAC7B,+DAA+D;YAC/D,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;gBAC3B,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;oBACd,yCAAyC;oBACzC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC5E,CAAC;gBACD,MAAM,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,kDAAkD;QAClD,yDAAyD;QACzD,IAAI,gBAAqC,CAAC;QAC1C,IAAI,MAAM,CAAC,QAAQ,EAAE,YAAY,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;YAC1D,MAAM,SAAS,GAAG,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACzE,IAAI,SAAS,CAAC,MAAM,KAAK,MAAM,IAAI,SAAS,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC3D,gBAAgB,GAAG,SAAS,CAAC,IAAI,CAAC;YACpC,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;YACpC,OAAO,EAAE,YAAY,CAAC,OAAO;YAC7B,gBAAgB;YAChB,KAAK,EAAE;gBACL,YAAY,EAAE,cAAc,CAAC,eAAe,CAAC;gBAC7C,gBAAgB,EAAE,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC;gBACtD,WAAW,EAAE,cAAc,CAAC,eAAe,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC;aACpF;YACD,YAAY,EAAE,YAAY,CAAC,YAAY,IAAI,MAAM;SAClD,CAAC,CAAC;QAEH,OAAO;YACL,MAAM,EAAE,eAAe,EAAE;YACzB,MAAM,EAAE,aAAa;SACtB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,2BAA2B;IAC7B,CAAC;CACF"}