@blackbox_ai/blackbox-cli-core 0.0.9 → 0.8.1

package/dist/src/encrypt/streaming-client.js

@@ -0,0 +1,232 @@
+import { serializePublicKeyPem, encryptAndSign, decryptAndVerify } from './crypto-utils.js';
+import {} from './types.js';
+import { ENCRYPTION_CONFIG } from './config.js';
+import { truncateHistory, getHistorySummary, HISTORY_CONFIG } from './history-manager.js';
+/**
+ * StreamingClient handles the complete encrypted streaming protocol
+ */
+export class StreamingClient {
+    baseUrl;
+    initialized = false;
+    nonce = 1000;
+    history = [];
+    attestationData;
+    constructor(baseUrl, _attestationUrl, systemPrompt) {
+        this.baseUrl = baseUrl;
+        if (systemPrompt) {
+            this.history.push({ role: 'system', content: systemPrompt });
+        }
+    }
+    /**
+     * Initialize from existing session data (avoids re-attestation)
+     * Now uses the session management layer
+     */
+    initializeFromSession(sessionData) {
+        // sessionId is stored in attestationData, no need to store separately
+        this.nonce = sessionData.nonce;
+        this.history = sessionData.history || [];
+        this.attestationData = sessionData.attestationData;
+        this.initialized = true;
+    }
+    async sendMessageStream(userText, fileData, retryAttempt = 0) {
+        if (!this.initialized || !this.attestationData) {
+            throw new Error('StreamingClient not initialized with attestation data. Call initializeFromSession() first.');
+        }
+        // Validate user text is not empty
+        if (!userText || userText.trim().length === 0) {
+            throw new Error('User message cannot be empty');
+        }
+        // Process file data and build context
+        let processedMessage = userText;
+        const contextParts = [];
+        // Process file data if provided
+        if (fileData) {
+            // Handle PDF/text content
+            if (fileData.fileText) {
+                contextParts.push(fileData.fileText);
+            }
+            // Build message with context
+            if (contextParts.length > 0) {
+                const contextText = contextParts.join('\n\n');
+                processedMessage = `Context:\n\n${contextText}\n\nQuery: ${userText}`;
+            }
+        }
+        // Create message object
+        const messageObj = {
+            role: 'user',
+            content: processedMessage
+        };
+        // Process image if provided
+        if (fileData?.imagesData && fileData.imagesData.length > 0) {
+            // Use the first image for now (following the sample pattern)
+            const firstImage = fileData.imagesData[0];
+            if (firstImage?.contents) {
+                // Extract base64 data (remove data:image/...;base64, prefix if present)
+                const base64Data = firstImage.contents.includes(',')
+                    ? firstImage.contents.split(',')[1]
+                    : firstImage.contents;
+                messageObj.image_url = base64Data;
+            }
+        }
+        // Ensure we have a valid history array - add the current message to history for encryption
+        const currentHistory = [...this.history, messageObj];
+        // Validate history is not empty
+        if (currentHistory.length === 0) {
+            throw new Error('Message history cannot be empty');
+        }
+        // Truncate history to prevent payload size issues (413 errors)
+        let truncatedHistory = truncateHistory(currentHistory);
+        // If this is a retry, apply even more aggressive truncation
+        if (retryAttempt > 0) {
+            const { truncateToSize } = await import('./history-manager.js');
+            // Reduce limit by half for each retry: 100KB → 50KB → 25KB
+            const retryLimit = HISTORY_CONFIG.maxHistoryBytes / (2 ** retryAttempt);
+            console.log(`🔄 Retry attempt ${retryAttempt}: Using ${(retryLimit / 1024).toFixed(1)}KB limit`);
+            truncatedHistory = truncateToSize(truncatedHistory, retryLimit);
+        }
+        // Log truncation if it occurred
+        if (truncatedHistory.length < currentHistory.length) {
+            const summary = getHistorySummary(currentHistory, truncatedHistory);
+            console.log(`📦 ${summary}`);
+        }
+        const historyJson = JSON.stringify(truncatedHistory);
+        const payloadSize = new TextEncoder().encode(historyJson).length;
+        console.log(`📊 Payload size before encryption: ${(payloadSize / 1024).toFixed(1)}KB (${truncatedHistory.length} messages)`);
+        // Warn if payload is large
+        if (payloadSize > 600_000) {
+            console.warn(`⚠️ Large payload: ${(payloadSize / 1024).toFixed(1)}KB - may cause 413 error`);
+        }
+        const localKeys = this.attestationData.localKeys;
+        // Encrypt with all shared keys for attestation
+        const encryptedPayloads = {};
+        for (const sharedKeyData of this.attestationData.sharedKeys) {
+            // Use the imported encryptAndSign function for each shared key
+            const payload = await encryptAndSign(historyJson, sharedKeyData.sharedKey, localKeys.ecdsaPrivate, this.nonce);
+            const peerPubPem = await serializePublicKeyPem(localKeys.ecdsaPublic);
+            encryptedPayloads[sharedKeyData.id] = {
+                peer_public_key: peerPubPem,
+                payload
+            };
+        }
+        // Build request with all encrypted payloads
+        const request = {
+            payloads: encryptedPayloads
+        };
+        // Send to /message_stream endpoint
+        const url = `${this.baseUrl.replace(/\/$/, '')}/message_stream`;
+        const response = await fetch(url, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(request),
+            signal: AbortSignal.timeout(ENCRYPTION_CONFIG.streamingTimeout)
+        });
+        if (!response.ok) {
+            // Provide helpful error messages for common failure cases
+            if (response.status === 401 || response.status === 403) {
+                throw new Error(`Authentication failed (${response.status}). ` +
+                    `Please ensure BLACKBOX_API_KEY is set correctly. ` +
+                    `Run 'blackbox configure' to set up your API key.`);
+            }
+            // Handle 413 Payload Too Large with retry logic
+            if (response.status === 413 && retryAttempt < 3) {
+                console.warn(`⚠️ 413 Payload Too Large error (attempt ${retryAttempt + 1}/3)`);
+                console.log(`🔄 Retrying with more aggressive history truncation...`);
+                // Retry with more aggressive truncation
+                return this.sendMessageStream(userText, fileData, retryAttempt + 1);
+            }
+            // If all retries failed or other error
+            if (response.status === 413) {
+                throw new Error(`Payload too large even after ${retryAttempt + 1} retry attempts. ` +
+                    `The conversation history is too large for the encrypted service. ` +
+                    `Please start a new conversation.`);
+            }
+            throw new Error(`Streaming request failed: ${response.status}`);
+        }
+        // Process streaming response using ReadableStream (NDJSON format)
+        let buffer = '';
+        let fullResponse = '';
+        const attestationData = this.attestationData;
+        // FIX: Use arrow functions to preserve 'this' context
+        const transformStream = new TransformStream({
+            transform: async (chunk, controller) => {
+                try {
+                    buffer += new TextDecoder().decode(chunk);
+                    // Process complete lines
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop() || ''; // Keep incomplete line in buffer
+                    for (const line of lines) {
+                        const trimmed = line.trim();
+                        if (!trimmed)
+                            continue;
+                        try {
+                            const jsonData = JSON.parse(trimmed);
+                            // Check for end-of-stream marker
+                            if (jsonData.eos === true) {
+                                // Add the complete response to history when stream ends
+                                if (fullResponse.trim().length > 0) {
+                                    // 'this' correctly refers to the class instance here
+                                    this.history.push({ role: 'assistant', content: fullResponse });
+                                    this.nonce = this.nonce + 1;
+                                }
+                                controller.terminate();
+                                return;
+                            }
+                            // The server should indicate which key was used for encryption
+                            // Try to decrypt with the appropriate key based on server response
+                            let decryptedChunk = null;
+                            // If server indicates which key was used
+                            if (jsonData.key_id) {
+                                const keyData = attestationData.sharedKeys.find((k) => k.id === jsonData.key_id);
+                                const serverKey = attestationData.serverPublicKeys.find((k) => k.id === jsonData.key_id);
+                                if (keyData && serverKey) {
+                                    try {
+                                        decryptedChunk = await decryptAndVerify(jsonData, keyData.sharedKey, serverKey.publicKeySpki);
+                                    }
+                                    catch (_error) {
+                                        // Failed to decrypt with specified key, will try all keys
+                                    }
+                                }
+                            }
+                            // If no key_id specified or decryption failed, try all keys
+                            if (!decryptedChunk) {
+                                for (let i = 0; i < attestationData.sharedKeys.length; i++) {
+                                    const keyData = attestationData.sharedKeys[i];
+                                    const serverKey = attestationData.serverPublicKeys[i];
+                                    try {
+                                        decryptedChunk = await decryptAndVerify(jsonData, keyData.sharedKey, serverKey.publicKeySpki);
+                                        break;
+                                    }
+                                    catch (_error) {
+                                        // Try next key
+                                        continue;
+                                    }
+                                }
+                            }
+                            if (!decryptedChunk) {
+                                throw new Error('Failed to decrypt chunk with any available key');
+                            }
+                            fullResponse += decryptedChunk;
+                            // Enqueue the decrypted chunk
+                            controller.enqueue(new TextEncoder().encode(decryptedChunk));
+                        }
+                        catch (_e) {
+                            // Error processing chunk, skip this line
+                        }
+                    }
+                }
+                catch (error) {
+                    controller.error(error);
+                }
+            },
+            flush: () => {
+                // Handle case where stream ends without explicit EOS marker
+                if (fullResponse.trim().length > 0) {
+                    this.history.push({ role: 'assistant', content: fullResponse });
+                    this.nonce = this.nonce + 1;
+                }
+            }
+        });
+        return response.body.pipeThrough(transformStream);
+    }
+}
+//# sourceMappingURL=streaming-client.js.map

package/dist/src/encrypt/streaming-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-client.js","sourceRoot":"","sources":["../../../src/encrypt/streaming-client.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,cAAc,EACd,gBAAgB,EAEjB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAgD,MAAM,YAAY,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAkB1F;;GAEG;AACH,MAAM,OAAO,eAAe;IAClB,OAAO,CAAS;IAChB,WAAW,GAAY,KAAK,CAAC;IAC7B,KAAK,GAAW,IAAI,CAAC;IACrB,OAAO,GAAkB,EAAE,CAAC;IAC5B,eAAe,CAA4B;IAEnD,YAAY,OAAe,EAAE,eAAuB,EAAE,YAAqB;QACzE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,qBAAqB,CAAC,WAKrB;QACC,sEAAsE;QACtE,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,IAAI,EAAE,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,QAAmB,EAAE,eAAuB,CAAC;QACrF,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,4FAA4F,CAAC,CAAC;QAChH,CAAC;QAED,kCAAkC;QAClC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,sCAAsC;QACtC,IAAI,gBAAgB,GAAG,QAAQ,CAAC;QAChC,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,gCAAgC;QAChC,IAAI,QAAQ,EAAE,CAAC;YACb,0BAA0B;YAC1B,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACvC,CAAC;YAED,6BAA6B;YAC7B,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC9C,gBAAgB,GAAG,eAAe,WAAW,cAAc,QAAQ,EAAE,CAAC;YACxE,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,MAAM,UAAU,GAAgB;YAC9B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,gBAAgB;SAC1B,CAAC;QAEF,4BAA4B;QAC5B,IAAI,QAAQ,EAAE,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3D,6DAA6D;YAC7D,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1C,IAAI,UAAU,EAAE,QAAQ,EAAE,CAAC;gBACzB,wEAAwE;gBACxE,MAAM,UAAU,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;oBAClD,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACnC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC;gBACxB,UAAU,CAAC,SAAS,GAAG,UAAU,CAAC;YACpC,CAAC;QACH,CAAC;QAED,2FAA2F;QAC3F,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAErD,gCAAgC;QAChC,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,+DAA+D;QAC/D,IAAI,gBAAgB,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAEvD,4DAA4D;QAC5D,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAChE,2DAA2D;YAC3D,MAAM,UAAU,GAAG,cAAc,CAAC,eAAe,GAAG,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,oBAAoB,YAAY,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACjG,gBAAgB,GAAG,cAAc,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;QAClE,CAAC;QAED,gCAAgC;QAChC,IAAI,gBAAgB,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC;YACpD,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACpE,OAAO,CAAC,GAAG,CAAC,MAAM,OAAO,EAAE,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACrD,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,gBAAgB,CAAC,MAAM,YAAY,CAAC,CAAC;QAE7H,2BAA2B;QAC3B,IAAI,WAAW,GAAG,OAAO,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CAAC,qBAAqB,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAC/F,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC;QAEjD,+CAA+C;QAC/C,MAAM,iBAAiB,GAA4E,EAAE,CAAC;QACtG,KAAK,MAAM,aAAa,IAAI,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;YAC5D,+DAA+D;YAC/D,MAAM,OAAO,GAAG,MAAM,cAAc,CAClC,WAAW,EAAE,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,KAAK,CACzE,CAAC;YAEF,MAAM,UAAU,GAAG,MAAM,qBAAqB,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACtE,iBAAiB,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG;gBACpC,eAAe,EAAE,UAAU;gBAC3B,OAAO;aACR,CAAC;QACJ,CAAC;QAED,4CAA4C;QAC5C,MAAM,OAAO,GAAqB;YAChC,QAAQ,EAAE,iBAAiB;SAC5B,CAAC;QAGF,mCAAmC;QACnC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC;SAChE,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,0DAA0D;YAC1D,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvD,MAAM,IAAI,KAAK,CACb,0BAA0B,QAAQ,CAAC,MAAM,KAAK;oBAC9C,mDAAmD;oBACnD,kDAAkD,CACnD,CAAC;YACJ,CAAC;YAED,gDAAgD;YAChD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;gBAChD,OAAO,CAAC,IAAI,CAAC,2CAA2C,YAAY,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC/E,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;gBAEtE,wCAAwC;gBACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;YACtE,CAAC;YAED,uCAAuC;YACvC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CACb,gCAAgC,YAAY,GAAG,CAAC,mBAAmB;oBACnE,mEAAmE;oBACnE,kCAAkC,CACnC,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAClE,CAAC;QAGD,kEAAkE;QAClE,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,YAAY,GAAG,EAAE,CAAC;QAEtB,MAAM,eAAe,GAAG,IAAI,CAAC,eAAgB,CAAC;QAE9C,sDAAsD;QACtD,MAAM,eAAe,GAAG,IAAI,eAAe,CAAyB;YAClE,SAAS,EAAE,KAAK,EAAE,KAAiB,EAAE,UAAwD,EAAE,EAAE;gBAC/F,IAAI,CAAC;oBACH,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAE1C,yBAAyB;oBACzB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,iCAAiC;oBAE7D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;wBAC5B,IAAI,CAAC,OAAO;4BAAE,SAAS;wBAEvB,IAAI,CAAC;4BACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAErC,iCAAiC;4BACjC,IAAI,QAAQ,CAAC,GAAG,KAAK,IAAI,EAAE,CAAC;gCAC1B,wDAAwD;gCACxD,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oCACnC,qDAAqD;oCACrD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;oCAChE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;gCAC9B,CAAC;gCACD,UAAU,CAAC,SAAS,EAAE,CAAC;gCACvB,OAAO;4BACT,CAAC;4BAED,+DAA+D;4BAC/D,mEAAmE;4BACnE,IAAI,cAAc,GAAkB,IAAI,CAAC;4BAEzC,yCAAyC;4BACzC,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gCACpB,MAAM,OAAO,GAAG,eAAe,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAwC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC;gCACxH,MAAM,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAA4C,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC;gCAEpI,IAAI,OAAO,IAAI,SAAS,EAAE,CAAC;oCACzB,IAAI,CAAC;wCACH,cAAc,GAAG,MAAM,gBAAgB,CACrC,QAA4B,EAAE,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,aAAa,CACzE,CAAC;oCACJ,CAAC;oCAAC,OAAO,MAAM,EAAE,CAAC;wCAChB,0DAA0D;oCAC5D,CAAC;gCACH,CAAC;4BACH,CAAC;4BAED,4DAA4D;4BAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;gCACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oCAC3D,MAAM,OAAO,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;oCAC9C,MAAM,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;oCAEtD,IAAI,CAAC;wCACH,cAAc,GAAG,MAAM,gBAAgB,CACrC,QAA4B,EAAE,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,aAAa,CACzE,CAAC;wCACF,MAAM;oCACR,CAAC;oCAAC,OAAO,MAAM,EAAE,CAAC;wCAChB,eAAe;wCACf,SAAS;oCACX,CAAC;gCACH,CAAC;4BACH,CAAC;4BAED,IAAI,CAAC,cAAc,EAAE,CAAC;gCACpB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;4BACpE,CAAC;4BAED,YAAY,IAAI,cAAc,CAAC;4BAE/B,8BAA8B;4BAC9B,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;wBAE/D,CAAC;wBAAC,OAAO,EAAW,EAAE,CAAC;4BACrB,yCAAyC;wBAC3C,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAc,EAAE,CAAC;oBACxB,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;YAED,KAAK,EAAE,GAAG,EAAE;gBACV,4DAA4D;gBAC5D,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACnC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;oBAChE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAK,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;IACrD,CAAC;CACF"}

package/dist/src/encrypt/tool-formatter.d.ts

@@ -0,0 +1,36 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { FunctionDeclaration } from '@google/genai';
+export interface ToolCall {
+    name: string;
+    arguments: Record<string, unknown>;
+}
+export interface ChatMessage {
+    role: 'system' | 'user' | 'assistant' | 'tool';
+    content: string;
+    tool_calls?: ToolCall[];
+    image_url?: string;
+}
+/**
+ * Format tools for system prompt based on model type
+ * Matches the Rust implementation's format_tools_for_system_prompt function
+ */
+export declare function formatToolsForSystemPrompt(tools: FunctionDeclaration[], modelType?: string): string;
+/**
+ * Format messages with tool calls based on model type
+ * Matches the Rust implementation's format_messages function
+ */
+export declare function formatMessages(messages: ChatMessage[], modelType?: string, history?: ChatMessage[]): ChatMessage[];
+/**
+ * Parse tool calls from text response
+ * Supports MiniMax, Qwen and DeepSeek formats
+ * Enhanced version with better error handling
+ */
+export declare function parseToolCallsFromText(text: string, modelType?: string): ToolCall[];
+/**
+ * Detect model type from response text
+ */
+export declare function detectModelType(text: string): string;

package/dist/src/encrypt/tool-formatter.js

@@ -0,0 +1,353 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Format tools for system prompt based on model type
+ * Matches the Rust implementation's format_tools_for_system_prompt function
+ */
+export function formatToolsForSystemPrompt(tools, modelType = 'minimax') {
+    if (!tools || tools.length === 0) {
+        return '';
+    }
+    let result = '';
+    switch (modelType.toLowerCase()) {
+        case 'minimax': {
+            result += '\n\n# Tools\n';
+            result += 'You may call one or more tools to assist with the user query.\n';
+            result += 'Here are the tools available in JSONSchema format:\n\n';
+            result += '<tools>\n';
+            for (const tool of tools) {
+                const toolFunction = {
+                    name: tool.name,
+                    description: tool.description || '',
+                    parameters: tool.parametersJsonSchema || {}
+                };
+                result += `<tool>${JSON.stringify(toolFunction, null, 0)}</tool>\n`;
+            }
+            result += '</tools>\n\n';
+            result += 'When making tool calls, use XML format to invoke tools and pass parameters:\n\n';
+            result += '<minimax:tool_call>\n';
+            result += '<invoke name="tool-name-1">\n';
+            result += '<parameter name="param-key-1">param-value-1</parameter>\n';
+            result += '<parameter name="param-key-2">param-value-2</parameter>\n';
+            result += '...\n';
+            result += '</invoke>\n';
+            result += '</minimax:tool_call>\n\n';
+            break;
+        }
+        case 'deepseek': {
+            result += '\n\n## Tools\n\nYou have access to the following tools. Use them when needed to help the user:\n\n';
+            for (const tool of tools) {
+                const name = tool.name || '';
+                const description = tool.description || '';
+                const parameters = tool.parametersJsonSchema || {};
+                result += `**${name}**: ${description}\n`;
+                if (parameters && typeof parameters === 'object' && Object.keys(parameters).length > 0) {
+                    result += `Parameters: ${JSON.stringify(parameters, null, 2)}\n`;
+                }
+                result += '\n';
+            }
+            result += '**CRITICAL TOOL USAGE RULES:**\n';
+            result += '1. NEVER output tool descriptions, parameters, or schemas in your response\n';
+            result += '2. NEVER show tool invocation payloads or debug information\n';
+            result += '3. Use tools silently - the user should only see the results\n';
+            result += '4. Use this EXACT format for tool calls:\n';
+            result += '   <｜tool▁calls▁begin｜><｜tool▁call▁begin｜>tool_name<｜tool▁sep｜>{"param":"value"}<｜tool▁call▁end｜><｜tool▁calls▁end｜>\n\n';
+            result += '5. Replace tool_name with the exact tool name and provide valid JSON arguments\n';
+            result += '6. Do NOT explain what tools you are using - just use them and show results\n\n';
+            break;
+        }
+        case 'qwen': { // Qwen format
+            result += '\n\n## Tools\n\nYou have access to the following tools. Use them when needed to help the user:\n\n';
+            for (const tool of tools) {
+                const name = tool.name || '';
+                const description = tool.description || '';
+                const parameters = tool.parametersJsonSchema || {};
+                result += `**${name}**: ${description}\n`;
+                if (parameters && typeof parameters === 'object' && Object.keys(parameters).length > 0) {
+                    result += `Parameters: ${JSON.stringify(parameters, null, 2)}\n`;
+                }
+                result += '\n';
+            }
+            result += '**CRITICAL TOOL USAGE RULES:**\n';
+            result += '1. NEVER output tool descriptions, parameters, or schemas in your response\n';
+            result += '2. NEVER show tool invocation payloads or debug information\n';
+            result += '3. Use tools silently - the user should only see the results\n';
+            result += '4. Use this EXACT format for tool calls:\n';
+            result += '   <tool_call>\n   {"name": "tool_name", "arguments": {"param": "value"}}\n   </tool_call>\n\n';
+            result += '5. Replace tool_name with the exact tool name and provide valid JSON arguments\n';
+            result += '6. Do NOT explain what tools you are using - just use them and show results\n\n';
+            break;
+        }
+        default: {
+            result += '\n\n## Tools\n\nYou have access to the following tools:\n\n';
+            for (const tool of tools) {
+                const name = tool.name || '';
+                const description = tool.description || '';
+                const parameters = tool.parametersJsonSchema || {};
+                result += `### ${name}\n`;
+                result += `Description: ${description}\n\n`;
+                result += `Parameters: ${JSON.stringify(parameters)}\n\n`;
+            }
+            result += 'IMPORTANT: ALWAYS adhere to this exact format for tool use:\n';
+            result += '<｜tool▁calls▁begin｜><｜tool▁call▁begin｜>tool_call_name<｜tool▁sep｜>tool_call_arguments<｜tool▁call▁end｜>{additional_tool_calls}<｜tool▁calls▁end｜>\n\n';
+            result += 'Where:\n';
+            result += '- `tool_call_name` must be an exact match to one of the available tools\n';
+            result += '- `tool_call_arguments` must be valid JSON that strictly follows the tool\'s Parameters Schema\n';
+            result += '- For multiple tool calls, chain them directly without separators or spaces\n';
+            break;
+        }
+    }
+    return result;
+}
+/**
+ * Format messages with tool calls based on model type
+ * Matches the Rust implementation's format_messages function
+ */
+export function formatMessages(messages, modelType = 'deepseek', history = []) {
+    // Helper function to create a message key for deduplication
+    const messageKey = (msg) => `${msg.role}:${msg.content}`;
+    const formattedMessages = [];
+    const seenMessages = new Set();
+    // Extract messages from input (skip system messages)
+    const requestMessages = messages.filter(msg => msg.role !== 'system');
+    // Process messages and format tool calls
+    for (const msg of requestMessages) {
+        const formattedMsg = {
+            role: msg.role,
+            content: msg.content,
+            image_url: msg.image_url
+        };
+        // Format tool calls if present
+        if (msg.tool_calls && msg.tool_calls.length > 0) {
+            let toolCallsStr = '';
+            switch (modelType.toLowerCase()) {
+                case 'minimax': {
+                    toolCallsStr = '<minimax:tool_call>\n';
+                    for (const tc of msg.tool_calls) {
+                        toolCallsStr += `<invoke name="${tc.name}">\n`;
+                        for (const [key, value] of Object.entries(tc.arguments)) {
+                            const valueStr = typeof value === 'string' ? value : JSON.stringify(value);
+                            toolCallsStr += `<parameter name="${key}">${valueStr}</parameter>\n`;
+                        }
+                        toolCallsStr += '</invoke>\n';
+                    }
+                    toolCallsStr += '</minimax:tool_call>';
+                    break;
+                }
+                case 'deepseek': {
+                    const formattedCalls = msg.tool_calls.map(tc => {
+                        const argsStr = JSON.stringify(tc.arguments);
+                        return `<｜tool▁call▁begin｜>${tc.name}<｜tool▁sep｜>${argsStr}<｜tool▁call▁end｜>`;
+                    });
+                    if (formattedCalls.length > 0) {
+                        toolCallsStr = `<｜tool▁calls▁begin｜>${formattedCalls.join('')}<｜tool▁calls▁end｜>`;
+                    }
+                    break;
+                }
+                case 'qwen': { // Qwen format
+                    toolCallsStr = msg.tool_calls.map(tc => {
+                        const argsStr = JSON.stringify(tc.arguments);
+                        return `<tool_call>\n{"name": "${tc.name}", "arguments": ${argsStr}}\n</tool_call>`;
+                    }).join('\n');
+                    break;
+                }
+                default: {
+                    const formattedCalls = msg.tool_calls.map(tc => {
+                        const argsStr = JSON.stringify(tc.arguments);
+                        return `<｜tool▁call▁begin｜>${tc.name}<｜tool▁sep｜>${argsStr}<｜tool▁call▁end｜>`;
+                    });
+                    if (formattedCalls.length > 0) {
+                        toolCallsStr = `<｜tool▁calls▁begin｜>${formattedCalls.join('')}<｜tool▁calls▁end｜>`;
+                    }
+                    break;
+                }
+            }
+            // Combine content and tool calls
+            if (formattedMsg.content && toolCallsStr) {
+                formattedMsg.content = `${formattedMsg.content}\n${toolCallsStr}`;
+            }
+            else {
+                formattedMsg.content = formattedMsg.content + toolCallsStr;
+            }
+        }
+        // Add to formatted messages if not seen before
+        const key = messageKey(formattedMsg);
+        if (!seenMessages.has(key)) {
+            seenMessages.add(key);
+            formattedMessages.push(formattedMsg);
+        }
+    }
+    // Merge with history (simplified version - can be enhanced based on needs)
+    const result = [];
+    // Add system messages from history first
+    for (const msg of history) {
+        if (msg.role === 'system') {
+            const key = messageKey(msg);
+            if (!seenMessages.has(key)) {
+                seenMessages.add(key);
+                result.push(msg);
+            }
+        }
+    }
+    // Add formatted messages
+    result.push(...formattedMessages);
+    return result;
+}
+/**
+ * Parse tool calls from text response
+ * Supports MiniMax, Qwen and DeepSeek formats
+ * Enhanced version with better error handling
+ */
+export function parseToolCallsFromText(text, modelType = 'minimax') {
+    const toolCalls = [];
+    switch (modelType.toLowerCase()) {
+        case 'minimax': {
+            // Parse MiniMax format with XML-style tool calls
+            const minimaxRegex = /<minimax:tool_call>([\s\S]*?)<\/minimax:tool_call>/g;
+            let minimaxMatch;
+            while ((minimaxMatch = minimaxRegex.exec(text)) !== null) {
+                const toolCallContent = minimaxMatch[1];
+                // Parse individual invoke blocks
+                const invokeRegex = /<invoke name="([^"]+)">([\s\S]*?)<\/invoke>/g;
+                let invokeMatch;
+                while ((invokeMatch = invokeRegex.exec(toolCallContent)) !== null) {
+                    const toolName = invokeMatch[1];
+                    const parametersContent = invokeMatch[2];
+                    // Parse parameters
+                    const parameterRegex = /<parameter name="([^"]+)">([\s\S]*?)<\/parameter>/g;
+                    const toolArguments = {};
+                    let paramMatch;
+                    while ((paramMatch = parameterRegex.exec(parametersContent)) !== null) {
+                        const paramName = paramMatch[1];
+                        const paramValue = paramMatch[2];
+                        // Try to parse as JSON, fallback to string
+                        try {
+                            toolArguments[paramName] = JSON.parse(paramValue);
+                        }
+                        catch {
+                            toolArguments[paramName] = paramValue;
+                        }
+                    }
+                    toolCalls.push({
+                        name: toolName,
+                        arguments: toolArguments
+                    });
+                }
+            }
+            break;
+        }
+        case 'deepseek': {
+            // Parse DeepSeek format with more robust regex that handles complex JSON
+            const deepseekRegex = /<｜tool▁call▁begin｜>(.*?)<｜tool▁sep｜>(.*?)<｜tool▁call▁end｜>/gs;
+            let deepseekMatch;
+            while ((deepseekMatch = deepseekRegex.exec(text)) !== null) {
+                try {
+                    const name = deepseekMatch[1].trim();
+                    const argsJson = deepseekMatch[2].trim();
+                    const args = parseJsonWithFallback(argsJson);
+                    toolCalls.push({ name, arguments: args });
+                }
+                catch (error) {
+                    console.warn('Failed to parse DeepSeek tool call:', error);
+                }
+            }
+            break;
+        }
+        default: { // Qwen format
+            // Parse Qwen format with more robust regex that handles multiline JSON
+            const qwenRegex = /<tool_call>\s*([\s\S]*?)\s*<\/tool_call>/g;
+            let qwenMatch;
+            while ((qwenMatch = qwenRegex.exec(text)) !== null) {
+                try {
+                    const jsonStr = qwenMatch[1].trim();
+                    const toolCallData = parseJsonWithFallback(jsonStr);
+                    if (toolCallData['name'] && toolCallData['arguments']) {
+                        toolCalls.push({
+                            name: toolCallData['name'],
+                            arguments: toolCallData['arguments']
+                        });
+                    }
+                }
+                catch (error) {
+                    console.warn('Failed to parse Qwen tool call:', error);
+                }
+            }
+            break;
+        }
+    }
+    return toolCalls;
+}
+/**
+ * Parse JSON with fallback error handling
+ */
+function parseJsonWithFallback(jsonStr) {
+    try {
+        return JSON.parse(jsonStr);
+    }
+    catch (_error) {
+        // Try to fix common JSON issues
+        let fixedJson = jsonStr;
+        // Fix trailing commas
+        fixedJson = fixedJson.replace(/,(\s*[}\]])/g, '$1');
+        // Try parsing again
+        try {
+            return JSON.parse(fixedJson);
+        }
+        catch (_secondError) {
+            // If still fails, try to extract key-value pairs manually
+            return extractKeyValuePairs(jsonStr);
+        }
+    }
+}
+/**
+ * Extract key-value pairs from malformed JSON as fallback
+ */
+function extractKeyValuePairs(jsonStr) {
+    const result = {};
+    // Simple regex to extract "key": "value" or "key": value patterns
+    const kvRegex = /"([^"]+)"\s*:\s*("([^"]*)"|([^,}\]]+))/g;
+    let match;
+    while ((match = kvRegex.exec(jsonStr)) !== null) {
+        const key = match[1];
+        const value = match[3] !== undefined ? match[3] : match[4]?.trim();
+        if (value !== undefined) {
+            // Try to parse value as number or boolean
+            if (value === 'true') {
+                result[key] = true;
+            }
+            else if (value === 'false') {
+                result[key] = false;
+            }
+            else if (!isNaN(Number(value))) {
+                result[key] = Number(value);
+            }
+            else {
+                result[key] = value;
+            }
+        }
+    }
+    return result;
+}
+/**
+ * Detect model type from response text
+ */
+export function detectModelType(text) {
+    // Check for MiniMax format tokens
+    if (text.includes('<minimax:tool_call>') || text.includes('<invoke name=')) {
+        return 'minimax';
+    }
+    // Check for DeepSeek format tokens
+    if (text.includes('<｜tool▁calls▁begin｜>') || text.includes('<｜tool▁call▁begin｜>')) {
+        return 'deepseek';
+    }
+    // Check for Qwen format tokens
+    if (text.includes('<tool_call>')) {
+        return 'qwen';
+    }
+    // Default to MiniMax format
+    return 'minimax';
+}
+//# sourceMappingURL=tool-formatter.js.map

package/dist/src/encrypt/tool-formatter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-formatter.js","sourceRoot":"","sources":["../../../src/encrypt/tool-formatter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgBH;;;GAGG;AACH,MAAM,UAAU,0BAA0B,CACxC,KAA4B,EAC5B,YAAoB,SAAS;IAE7B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,QAAQ,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC;QAChC,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,MAAM,IAAI,eAAe,CAAC;YAC1B,MAAM,IAAI,iEAAiE,CAAC;YAC5E,MAAM,IAAI,wDAAwD,CAAC;YACnE,MAAM,IAAI,WAAW,CAAC;YAEtB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,YAAY,GAAG;oBACnB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;oBACnC,UAAU,EAAE,IAAI,CAAC,oBAAoB,IAAI,EAAE;iBAC5C,CAAC;gBACF,MAAM,IAAI,SAAS,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,WAAW,CAAC;YACtE,CAAC;YAED,MAAM,IAAI,cAAc,CAAC;YACzB,MAAM,IAAI,iFAAiF,CAAC;YAC5F,MAAM,IAAI,uBAAuB,CAAC;YAClC,MAAM,IAAI,+BAA+B,CAAC;YAC1C,MAAM,IAAI,2DAA2D,CAAC;YACtE,MAAM,IAAI,2DAA2D,CAAC;YACtE,MAAM,IAAI,OAAO,CAAC;YAClB,MAAM,IAAI,aAAa,CAAC;YACxB,MAAM,IAAI,0BAA0B,CAAC;YACrC,MAAM;QACR,CAAC;QAED,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,IAAI,oGAAoG,CAAC;YAE/G,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;gBAC7B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC;gBAEnD,MAAM,IAAI,KAAK,IAAI,OAAO,WAAW,IAAI,CAAC;gBAC1C,IAAI,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvF,MAAM,IAAI,eAAe,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC;gBACnE,CAAC;gBACD,MAAM,IAAI,IAAI,CAAC;YACjB,CAAC;YAED,MAAM,IAAI,kCAAkC,CAAC;YAC7C,MAAM,IAAI,8EAA8E,CAAC;YACzF,MAAM,IAAI,+DAA+D,CAAC;YAC1E,MAAM,IAAI,gEAAgE,CAAC;YAC3E,MAAM,IAAI,4CAA4C,CAAC;YACvD,MAAM,IAAI,yHAAyH,CAAC;YACpI,MAAM,IAAI,kFAAkF,CAAC;YAC7F,MAAM,IAAI,iFAAiF,CAAC;YAC5F,MAAM;QACR,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc;YAC3B,MAAM,IAAI,oGAAoG,CAAC;YAE/G,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;gBAC7B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC;gBAEnD,MAAM,IAAI,KAAK,IAAI,OAAO,WAAW,IAAI,CAAC;gBAC1C,IAAI,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvF,MAAM,IAAI,eAAe,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC;gBACnE,CAAC;gBACD,MAAM,IAAI,IAAI,CAAC;YACjB,CAAC;YAED,MAAM,IAAI,kCAAkC,CAAC;YAC7C,MAAM,IAAI,8EAA8E,CAAC;YACzF,MAAM,IAAI,+DAA+D,CAAC;YAC1E,MAAM,IAAI,gEAAgE,CAAC;YAC3E,MAAM,IAAI,4CAA4C,CAAC;YACvD,MAAM,IAAI,gGAAgG,CAAC;YAC3G,MAAM,IAAI,kFAAkF,CAAC;YAC7F,MAAM,IAAI,iFAAiF,CAAC;YAC5F,MAAM;QACR,CAAC;QAED,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,IAAI,6DAA6D,CAAC;YAExE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;gBAC7B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC;gBAEnD,MAAM,IAAI,OAAO,IAAI,IAAI,CAAC;gBAC1B,MAAM,IAAI,gBAAgB,WAAW,MAAM,CAAC;gBAC5C,MAAM,IAAI,eAAe,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC;YAC5D,CAAC;YAED,MAAM,IAAI,+DAA+D,CAAC;YAC1E,MAAM,IAAI,oJAAoJ,CAAC;YAC/J,MAAM,IAAI,UAAU,CAAC;YACrB,MAAM,IAAI,2EAA2E,CAAC;YACtF,MAAM,IAAI,kGAAkG,CAAC;YAC7G,MAAM,IAAI,+EAA+E,CAAC;YAC1F,MAAM;QACR,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,cAAc,CAC5B,QAAuB,EACvB,YAAoB,UAAU,EAC9B,UAAyB,EAAE;IAE3B,4DAA4D;IAC5D,MAAM,UAAU,GAAG,CAAC,GAAgB,EAAU,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;IAE9E,MAAM,iBAAiB,GAAkB,EAAE,CAAC;IAC5C,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;IAEvC,qDAAqD;IACrD,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IAEtE,yCAAyC;IACzC,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QAClC,MAAM,YAAY,GAAgB;YAChC,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,SAAS,EAAE,GAAG,CAAC,SAAS;SACzB,CAAC;QAEF,+BAA+B;QAC/B,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,IAAI,YAAY,GAAG,EAAE,CAAC;YAEtB,QAAQ,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC;gBAChC,KAAK,SAAS,CAAC,CAAC,CAAC;oBACf,YAAY,GAAG,uBAAuB,CAAC;oBACvC,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;wBAChC,YAAY,IAAI,iBAAiB,EAAE,CAAC,IAAI,MAAM,CAAC;wBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;4BACxD,MAAM,QAAQ,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;4BAC3E,YAAY,IAAI,oBAAoB,GAAG,KAAK,QAAQ,gBAAgB,CAAC;wBACvE,CAAC;wBACD,YAAY,IAAI,aAAa,CAAC;oBAChC,CAAC;oBACD,YAAY,IAAI,sBAAsB,CAAC;oBACvC,MAAM;gBACR,CAAC;gBAED,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,MAAM,cAAc,GAAG,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;wBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;wBAC7C,OAAO,sBAAsB,EAAE,CAAC,IAAI,eAAe,OAAO,mBAAmB,CAAC;oBAChF,CAAC,CAAC,CAAC;oBAEH,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC9B,YAAY,GAAG,uBAAuB,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,oBAAoB,CAAC;oBACpF,CAAC;oBACD,MAAM;gBACR,CAAC;gBAED,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc;oBAC3B,YAAY,GAAG,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;wBACrC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;wBAC7C,OAAO,0BAA0B,EAAE,CAAC,IAAI,mBAAmB,OAAO,iBAAiB,CAAC;oBACtF,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,MAAM;gBACR,CAAC;gBAED,OAAO,CAAC,CAAC,CAAC;oBACR,MAAM,cAAc,GAAG,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;wBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;wBAC7C,OAAO,sBAAsB,EAAE,CAAC,IAAI,eAAe,OAAO,mBAAmB,CAAC;oBAChF,CAAC,CAAC,CAAC;oBAEH,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC9B,YAAY,GAAG,uBAAuB,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,oBAAoB,CAAC;oBACpF,CAAC;oBACD,MAAM;gBACR,CAAC;YACH,CAAC;YAED,iCAAiC;YACjC,IAAI,YAAY,CAAC,OAAO,IAAI,YAAY,EAAE,CAAC;gBACzC,YAAY,CAAC,OAAO,GAAG,GAAG,YAAY,CAAC,OAAO,KAAK,YAAY,EAAE,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,GAAG,YAAY,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,+CAA+C;QAC/C,MAAM,GAAG,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;QACrC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACtB,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,MAAM,MAAM,GAAkB,EAAE,CAAC;IAEjC,yCAAyC;IACzC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC1B,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3B,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACtB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,MAAM,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC,CAAC;IAElC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,sBAAsB,CAAC,IAAY,EAAE,YAAoB,SAAS;IAChF,MAAM,SAAS,GAAe,EAAE,CAAC;IAEjC,QAAQ,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC;QAChC,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,iDAAiD;YACjD,MAAM,YAAY,GAAG,qDAAqD,CAAC;YAC3E,IAAI,YAAY,CAAC;YAEjB,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;gBACzD,MAAM,eAAe,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;gBAExC,iCAAiC;gBACjC,MAAM,WAAW,GAAG,8CAA8C,CAAC;gBACnE,IAAI,WAAW,CAAC;gBAEhB,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;oBAClE,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;oBAChC,MAAM,iBAAiB,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;oBAEzC,mBAAmB;oBACnB,MAAM,cAAc,GAAG,oDAAoD,CAAC;oBAC5E,MAAM,aAAa,GAA4B,EAAE,CAAC;oBAClD,IAAI,UAAU,CAAC;oBAEf,OAAO,CAAC,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;wBACtE,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;wBAChC,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;wBAEjC,2CAA2C;wBAC3C,IAAI,CAAC;4BACH,aAAa,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;wBACpD,CAAC;wBAAC,MAAM,CAAC;4BACP,aAAa,CAAC,SAAS,CAAC,GAAG,UAAU,CAAC;wBACxC,CAAC;oBACH,CAAC;oBAED,SAAS,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,QAAQ;wBACd,SAAS,EAAE,aAAa;qBACzB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,yEAAyE;YACzE,MAAM,aAAa,GAAG,8DAA8D,CAAC;YACrF,IAAI,aAAa,CAAC;YAElB,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC3D,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACrC,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACzC,MAAM,IAAI,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;oBAE7C,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;gBAC7D,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QAED,OAAO,CAAC,CAAC,CAAC,CAAC,cAAc;YACvB,uEAAuE;YACvE,MAAM,SAAS,GAAG,2CAA2C,CAAC;YAC9D,IAAI,SAAS,CAAC;YAEd,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;gBACnD,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACpC,MAAM,YAAY,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;oBAEpD,IAAI,YAAY,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;wBACtD,SAAS,CAAC,IAAI,CAAC;4BACb,IAAI,EAAE,YAAY,CAAC,MAAM,CAAW;4BACpC,SAAS,EAAE,YAAY,CAAC,WAAW,CAA4B;yBAChE,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,OAAe;IAC5C,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,MAAM,EAAE,CAAC;QAChB,gCAAgC;QAChC,IAAI,SAAS,GAAG,OAAO,CAAC;QAExB,sBAAsB;QACtB,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAEpD,oBAAoB;QACpB,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,YAAY,EAAE,CAAC;YACtB,0DAA0D;YAC1D,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAAC,OAAe;IAC3C,MAAM,MAAM,GAA4B,EAAE,CAAC;IAE3C,kEAAkE;IAClE,MAAM,OAAO,GAAG,yCAAyC,CAAC;IAE1D,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAChD,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACrB,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QAEnE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,0CAA0C;YAC1C,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;gBACrB,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;YACrB,CAAC;iBAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;gBAC7B,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACtB,CAAC;iBAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjC,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,IAAY;IAC1C,kCAAkC;IAClC,IAAI,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QAC3E,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,mCAAmC;IACnC,IAAI,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC;QAClF,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,+BAA+B;IAC/B,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QACjC,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,4BAA4B;IAC5B,OAAO,SAAS,CAAC;AACnB,CAAC"}