centient 2.20.0

package/dist/utils/rlvr/RewardHistoryStore.js

@@ -0,0 +1,428 @@
+import { randomUUID } from "crypto";
+import { OpenAI } from "openai";
+import { getQdrantClient } from "../QdrantConnectionManager.js";
+const COLLECTION_NAME = "rlvr-reward-history";
+const VECTOR_SIZE = 3072;
+const MAX_REWARDS_PER_PATTERN = 100;
+const inMemoryStore = new Map();
+export class RewardHistoryStore {
+    openai = null;
+    config;
+    initialized = false;
+    useQdrant = false;
+    constructor(config = {}) {
+        this.config = config;
+        if (!config.inMemoryOnly) {
+            const openaiKey = config.openaiApiKey || process.env.OPENAI_API_KEY;
+            if (openaiKey) {
+                this.openai = new OpenAI({ apiKey: openaiKey });
+            }
+        }
+    }
+    async initialize() {
+        if (this.initialized) {
+            return {
+                mode: this.useQdrant ? "qdrant" : "memory",
+                message: "Already initialized",
+            };
+        }
+        if (this.config.inMemoryOnly) {
+            this.initialized = true;
+            this.useQdrant = false;
+            return { mode: "memory", message: "Using in-memory store (configured)" };
+        }
+        try {
+            const client = await getQdrantClient();
+            if (!client) {
+                this.initialized = true;
+                this.useQdrant = false;
+                return {
+                    mode: "memory",
+                    message: "Qdrant unavailable, using in-memory fallback",
+                };
+            }
+            const collections = await client.getCollections();
+            const exists = collections.collections.some((c) => c.name === COLLECTION_NAME);
+            if (!exists) {
+                await client.createCollection(COLLECTION_NAME, {
+                    vectors: {
+                        size: VECTOR_SIZE,
+                        distance: "Cosine",
+                    },
+                });
+                await client.createPayloadIndex(COLLECTION_NAME, {
+                    field_name: "patternId",
+                    field_schema: "keyword",
+                });
+                await client.createPayloadIndex(COLLECTION_NAME, {
+                    field_name: "timestamp",
+                    field_schema: "datetime",
+                });
+                await client.createPayloadIndex(COLLECTION_NAME, {
+                    field_name: "passed",
+                    field_schema: "bool",
+                });
+                console.error(`[RewardHistoryStore] Created collection: ${COLLECTION_NAME}`);
+            }
+            this.initialized = true;
+            this.useQdrant = true;
+            return {
+                mode: "qdrant",
+                message: `Using Qdrant collection: ${COLLECTION_NAME}`,
+            };
+        }
+        catch (error) {
+            console.error(`[RewardHistoryStore] Qdrant init failed, using in-memory: ${error}`);
+            this.initialized = true;
+            this.useQdrant = false;
+            return {
+                mode: "memory",
+                message: `Qdrant error: ${error}, using in-memory fallback`,
+            };
+        }
+    }
+    async saveReward(signal) {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        if (this.useQdrant) {
+            await this.saveToQdrant(signal);
+        }
+        this.updateInMemory(signal);
+    }
+    async getHistory(patternId) {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        const memHistory = inMemoryStore.get(patternId);
+        if (memHistory) {
+            return memHistory;
+        }
+        if (this.useQdrant) {
+            return this.loadFromQdrant(patternId);
+        }
+        return undefined;
+    }
+    async listPatterns() {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        if (this.useQdrant) {
+            try {
+                const client = await getQdrantClient();
+                if (client) {
+                    const result = await client.scroll(COLLECTION_NAME, {
+                        limit: 1000,
+                        with_payload: ["patternId"],
+                    });
+                    const patternIds = new Set();
+                    for (const point of result.points) {
+                        const payload = point.payload;
+                        if (payload?.patternId) {
+                            patternIds.add(payload.patternId);
+                        }
+                    }
+                    return Array.from(patternIds);
+                }
+            }
+            catch (error) {
+                console.error(`[RewardHistoryStore] listPatterns error: ${error}`);
+            }
+        }
+        return Array.from(inMemoryStore.keys());
+    }
+    async getTopPatterns(limit = 10, minScore = 0) {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        const patterns = await this.listPatterns();
+        const results = [];
+        for (const patternId of patterns) {
+            const history = await this.getHistory(patternId);
+            if (history && history.averageScore >= minScore) {
+                results.push({
+                    patternId,
+                    averageScore: history.averageScore,
+                    runCount: history.rewards.length,
+                });
+            }
+        }
+        return results
+            .sort((a, b) => b.averageScore - a.averageScore)
+            .slice(0, limit);
+    }
+    async findSimilarPatterns(description, limit = 5) {
+        if (!this.useQdrant || !this.openai) {
+            return [];
+        }
+        try {
+            const embedding = await this.getEmbedding(description);
+            const client = await getQdrantClient();
+            if (!client)
+                return [];
+            const result = await client.search(COLLECTION_NAME, {
+                vector: embedding,
+                limit: limit * 2,
+                with_payload: ["patternId"],
+            });
+            const patternScores = new Map();
+            for (const hit of result) {
+                const payload = hit.payload;
+                if (payload?.patternId) {
+                    const existing = patternScores.get(payload.patternId) || 0;
+                    if (hit.score > existing) {
+                        patternScores.set(payload.patternId, hit.score);
+                    }
+                }
+            }
+            return Array.from(patternScores.entries())
+                .map(([patternId, score]) => ({ patternId, score }))
+                .sort((a, b) => b.score - a.score)
+                .slice(0, limit);
+        }
+        catch (error) {
+            console.error(`[RewardHistoryStore] findSimilarPatterns error: ${error}`);
+            return [];
+        }
+    }
+    async clearPattern(patternId) {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        inMemoryStore.delete(patternId);
+        if (this.useQdrant) {
+            try {
+                const client = await getQdrantClient();
+                if (client) {
+                    await client.delete(COLLECTION_NAME, {
+                        filter: {
+                            must: [{ key: "patternId", match: { value: patternId } }],
+                        },
+                    });
+                }
+            }
+            catch (error) {
+                console.error(`[RewardHistoryStore] clearPattern error: ${error}`);
+            }
+        }
+    }
+    async clearAll() {
+        inMemoryStore.clear();
+        if (this.useQdrant) {
+            try {
+                const client = await getQdrantClient();
+                if (client) {
+                    await client.deleteCollection(COLLECTION_NAME);
+                    this.initialized = false;
+                    await this.initialize();
+                }
+            }
+            catch (error) {
+                console.error(`[RewardHistoryStore] clearAll error: ${error}`);
+            }
+        }
+    }
+    async getStats() {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        const patterns = await this.listPatterns();
+        let totalRewards = 0;
+        for (const patternId of patterns) {
+            const history = await this.getHistory(patternId);
+            if (history) {
+                totalRewards += history.rewards.length;
+            }
+        }
+        const stats = {
+            mode: this.useQdrant ? "qdrant" : "memory",
+            patternCount: patterns.length,
+            totalRewards,
+        };
+        if (this.useQdrant) {
+            try {
+                const client = await getQdrantClient();
+                if (client) {
+                    const info = await client.getCollection(COLLECTION_NAME);
+                    stats.collectionInfo = {
+                        pointsCount: info.points_count || 0,
+                        vectorsCount: info.vectors_count || 0,
+                    };
+                }
+            }
+            catch {
+            }
+        }
+        return stats;
+    }
+    async saveToQdrant(signal) {
+        try {
+            const client = await getQdrantClient();
+            if (!client || !this.openai)
+                return;
+            const embeddingText = `Pattern: ${signal.patternId}\n${signal.breakdown.map((b) => `${b.category}: ${b.details}`).join("\n")}`;
+            const embedding = await this.getEmbedding(embeddingText);
+            const pointId = this.generatePointId();
+            await client.upsert(COLLECTION_NAME, {
+                points: [
+                    {
+                        id: pointId,
+                        vector: embedding,
+                        payload: {
+                            patternId: signal.patternId,
+                            timestamp: signal.timestamp,
+                            passed: signal.passed,
+                            correctnessScore: signal.correctnessScore,
+                            performanceScore: signal.performanceScore,
+                            reliabilityScore: signal.reliabilityScore,
+                            overallScore: signal.overallScore,
+                            confidence: signal.confidence,
+                            sampleSize: signal.sampleSize,
+                            trend: signal.trend,
+                            breakdown: signal.breakdown,
+                        },
+                    },
+                ],
+            });
+        }
+        catch (error) {
+            console.error(`[RewardHistoryStore] saveToQdrant error: ${error}`);
+        }
+    }
+    async loadFromQdrant(patternId) {
+        try {
+            const client = await getQdrantClient();
+            if (!client)
+                return undefined;
+            const result = await client.scroll(COLLECTION_NAME, {
+                filter: {
+                    must: [{ key: "patternId", match: { value: patternId } }],
+                },
+                limit: MAX_REWARDS_PER_PATTERN,
+                with_payload: true,
+                order_by: { key: "timestamp", direction: "desc" },
+            });
+            if (result.points.length === 0) {
+                return undefined;
+            }
+            const rewards = result.points.map((point) => {
+                const p = point.payload;
+                return {
+                    patternId: p.patternId,
+                    timestamp: p.timestamp,
+                    passed: p.passed,
+                    correctnessScore: p.correctnessScore,
+                    performanceScore: p.performanceScore,
+                    reliabilityScore: p.reliabilityScore,
+                    overallScore: p.overallScore,
+                    confidence: p.confidence,
+                    sampleSize: p.sampleSize,
+                    trend: p.trend,
+                    breakdown: p.breakdown,
+                };
+            });
+            rewards.sort((a, b) => new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime());
+            const history = this.buildHistory(patternId, rewards);
+            inMemoryStore.set(patternId, history);
+            return history;
+        }
+        catch (error) {
+            console.error(`[RewardHistoryStore] loadFromQdrant error: ${error}`);
+            return undefined;
+        }
+    }
+    updateInMemory(signal) {
+        let history = inMemoryStore.get(signal.patternId);
+        if (!history) {
+            history = {
+                patternId: signal.patternId,
+                rewards: [],
+                averageScore: 0,
+                variability: 0,
+                recentTrend: "stable",
+                firstPassed: null,
+                lastFailed: null,
+                consecutivePasses: 0,
+            };
+        }
+        history.rewards.push(signal);
+        if (history.rewards.length > MAX_REWARDS_PER_PATTERN) {
+            history.rewards = history.rewards.slice(-MAX_REWARDS_PER_PATTERN);
+        }
+        const scores = history.rewards.map((r) => r.overallScore);
+        history.averageScore = scores.reduce((a, b) => a + b, 0) / scores.length;
+        const variance = scores.reduce((sum, score) => sum + (score - history.averageScore) ** 2, 0) / scores.length;
+        history.variability = Math.sqrt(variance);
+        history.recentTrend = signal.trend;
+        if (signal.passed) {
+            if (!history.firstPassed) {
+                history.firstPassed = signal.timestamp;
+            }
+            history.consecutivePasses++;
+        }
+        else {
+            history.lastFailed = signal.timestamp;
+            history.consecutivePasses = 0;
+        }
+        inMemoryStore.set(signal.patternId, history);
+    }
+    buildHistory(patternId, rewards) {
+        const scores = rewards.map((r) => r.overallScore);
+        const averageScore = scores.length > 0 ? scores.reduce((a, b) => a + b, 0) / scores.length : 0;
+        const variance = scores.length > 0
+            ? scores.reduce((sum, score) => sum + (score - averageScore) ** 2, 0) /
+                scores.length
+            : 0;
+        const variability = Math.sqrt(variance);
+        let firstPassed = null;
+        let lastFailed = null;
+        let consecutivePasses = 0;
+        for (const reward of rewards) {
+            if (reward.passed) {
+                if (!firstPassed)
+                    firstPassed = reward.timestamp;
+                consecutivePasses++;
+            }
+            else {
+                lastFailed = reward.timestamp;
+                consecutivePasses = 0;
+            }
+        }
+        const recentTrend = rewards.length > 0 ? rewards[rewards.length - 1].trend : "stable";
+        return {
+            patternId,
+            rewards,
+            averageScore,
+            variability,
+            recentTrend,
+            firstPassed,
+            lastFailed,
+            consecutivePasses,
+        };
+    }
+    async getEmbedding(text) {
+        if (!this.openai) {
+            throw new Error("OpenAI client not initialized");
+        }
+        const response = await this.openai.embeddings.create({
+            model: "text-embedding-3-large",
+            input: text,
+        });
+        return response.data[0].embedding;
+    }
+    generatePointId() {
+        return randomUUID();
+    }
+}
+let defaultStore = null;
+export function getRewardHistoryStore(config) {
+    if (!defaultStore) {
+        defaultStore = new RewardHistoryStore(config);
+    }
+    return defaultStore;
+}
+export function resetRewardHistoryStore() {
+    defaultStore = null;
+    inMemoryStore.clear();
+}
+//# sourceMappingURL=RewardHistoryStore.js.map

package/dist/utils/rlvr/RewardHistoryStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RewardHistoryStore.js","sourceRoot":"","sources":["../../../src/utils/rlvr/RewardHistoryStore.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAOhE,MAAM,eAAe,GAAG,qBAAqB,CAAC;AAC9C,MAAM,WAAW,GAAG,IAAI,CAAC;AACzB,MAAM,uBAAuB,GAAG,GAAG,CAAC;AAapC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAyB,CAAC;AAMvD,MAAM,OAAO,kBAAkB;IACrB,MAAM,GAAkB,IAAI,CAAC;IAC7B,MAAM,CAA2B;IACjC,WAAW,GAAG,KAAK,CAAC;IACpB,SAAS,GAAG,KAAK,CAAC;IAE1B,YAAY,SAAmC,EAAE;QAC/C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YACpE,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;gBAC1C,OAAO,EAAE,qBAAqB;aAC/B,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,oCAAoC,EAAE,CAAC;QAC3E,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBACxB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,OAAO;oBACL,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,8CAA8C;iBACxD,CAAC;YACJ,CAAC;YAGD,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;YAClD,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,CAAC,IAAI,CACzC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAClC,CAAC;YAEF,IAAI,CAAC,MAAM,EAAE,CAAC;gBAEZ,MAAM,MAAM,CAAC,gBAAgB,CAAC,eAAe,EAAE;oBAC7C,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW;wBACjB,QAAQ,EAAE,QAAQ;qBACnB;iBACF,CAAC,CAAC;gBAGH,MAAM,MAAM,CAAC,kBAAkB,CAAC,eAAe,EAAE;oBAC/C,UAAU,EAAE,WAAW;oBACvB,YAAY,EAAE,SAAS;iBACxB,CAAC,CAAC;gBAEH,MAAM,MAAM,CAAC,kBAAkB,CAAC,eAAe,EAAE;oBAC/C,UAAU,EAAE,WAAW;oBACvB,YAAY,EAAE,UAAU;iBACzB,CAAC,CAAC;gBAEH,MAAM,MAAM,CAAC,kBAAkB,CAAC,eAAe,EAAE;oBAC/C,UAAU,EAAE,QAAQ;oBACpB,YAAY,EAAE,MAAM;iBACrB,CAAC,CAAC;gBAEH,OAAO,CAAC,KAAK,CACX,4CAA4C,eAAe,EAAE,CAC9D,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,4BAA4B,eAAe,EAAE;aACvD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,6DAA6D,KAAK,EAAE,CACrE,CAAC;YACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,iBAAiB,KAAK,4BAA4B;aAC5D,CAAC;QACJ,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,UAAU,CAAC,MAAoB;QACnC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAGD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAKD,KAAK,CAAC,UAAU,CAAC,SAAiB;QAChC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAGD,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAChD,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,UAAU,CAAC;QACpB,CAAC;QAGD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAKD,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;gBACvC,IAAI,MAAM,EAAE,CAAC;oBAEX,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE;wBAClD,KAAK,EAAE,IAAI;wBACX,YAAY,EAAE,CAAC,WAAW,CAAC;qBAC5B,CAAC,CAAC;oBAEH,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;oBACrC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;wBAClC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAwC,CAAC;wBAC/D,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;4BACvB,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;wBACpC,CAAC;oBACH,CAAC;oBACD,OAAO,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,KAAK,EAAE,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;IAKD,KAAK,CAAC,cAAc,CAClB,QAAgB,EAAE,EAClB,WAAmB,CAAC;QAIpB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3C,MAAM,OAAO,GAIR,EAAE,CAAC;QAER,KAAK,MAAM,SAAS,IAAI,QAAQ,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YACjD,IAAI,OAAO,IAAI,OAAO,CAAC,YAAY,IAAI,QAAQ,EAAE,CAAC;gBAChD,OAAO,CAAC,IAAI,CAAC;oBACX,SAAS;oBACT,YAAY,EAAE,OAAO,CAAC,YAAY;oBAClC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM;iBACjC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,OAAO;aACX,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC;aAC/C,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACrB,CAAC;IAKD,KAAK,CAAC,mBAAmB,CACvB,WAAmB,EACnB,QAAgB,CAAC;QAEjB,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACpC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;YAEvC,IAAI,CAAC,MAAM;gBAAE,OAAO,EAAE,CAAC;YAEvB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE;gBAClD,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,KAAK,GAAG,CAAC;gBAChB,YAAY,EAAE,CAAC,WAAW,CAAC;aAC5B,CAAC,CAAC;YAGH,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC;YAChD,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,GAAG,CAAC,OAAwC,CAAC;gBAC7D,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;oBACvB,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC3D,IAAI,GAAG,CAAC,KAAK,GAAG,QAAQ,EAAE,CAAC;wBACzB,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;iBACvC,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;iBACnD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;iBACjC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mDAAmD,KAAK,EAAE,CAAC,CAAC;YAC1E,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,YAAY,CAAC,SAAiB;QAClC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEhC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;gBACvC,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE;wBACnC,MAAM,EAAE;4BACN,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC;yBAC1D;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,KAAK,EAAE,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,QAAQ;QACZ,aAAa,CAAC,KAAK,EAAE,CAAC;QAEtB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;gBACvC,IAAI,MAAM,EAAE,CAAC;oBAEX,MAAM,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;oBAC/C,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;oBACzB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC1B,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,wCAAwC,KAAK,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,QAAQ;QASZ,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3C,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,KAAK,MAAM,SAAS,IAAI,QAAQ,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YACjD,IAAI,OAAO,EAAE,CAAC;gBACZ,YAAY,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACzC,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAKP;YACF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;YAC1C,YAAY,EAAE,QAAQ,CAAC,MAAM;YAC7B,YAAY;SACb,CAAC;QAEF,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;gBACvC,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;oBACzD,KAAK,CAAC,cAAc,GAAG;wBACrB,WAAW,EAAE,IAAI,CAAC,YAAY,IAAI,CAAC;wBACnC,YAAY,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;qBACtC,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAMO,KAAK,CAAC,YAAY,CAAC,MAAoB;QAC7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,OAAO;YAGpC,MAAM,aAAa,GAAG,YAAY,MAAM,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/H,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAGzD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;YAEvC,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE;gBACnC,MAAM,EAAE;oBACN;wBACE,EAAE,EAAE,OAAO;wBACX,MAAM,EAAE,SAAS;wBACjB,OAAO,EAAE;4BACP,SAAS,EAAE,MAAM,CAAC,SAAS;4BAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;4BAC3B,MAAM,EAAE,MAAM,CAAC,MAAM;4BACrB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;4BACzC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;4BACzC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;4BACzC,YAAY,EAAE,MAAM,CAAC,YAAY;4BACjC,UAAU,EAAE,MAAM,CAAC,UAAU;4BAC7B,UAAU,EAAE,MAAM,CAAC,UAAU;4BAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;4BACnB,SAAS,EAAE,MAAM,CAAC,SAAS;yBAC5B;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,KAAK,EAAE,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,SAAiB;QAEjB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM;gBAAE,OAAO,SAAS,CAAC;YAE9B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE;gBAClD,MAAM,EAAE;oBACN,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC;iBAC1D;gBACD,KAAK,EAAE,uBAAuB;gBAC9B,YAAY,EAAE,IAAI;gBAClB,QAAQ,EAAE,EAAE,GAAG,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE;aAClD,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/B,OAAO,SAAS,CAAC;YACnB,CAAC;YAGD,MAAM,OAAO,GAAmB,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,GAAG,KAAK,CAAC,OAAkC,CAAC;gBACnD,OAAO;oBACL,SAAS,EAAE,CAAC,CAAC,SAAmB;oBAChC,SAAS,EAAE,CAAC,CAAC,SAAmB;oBAChC,MAAM,EAAE,CAAC,CAAC,MAAiB;oBAC3B,gBAAgB,EAAE,CAAC,CAAC,gBAA0B;oBAC9C,gBAAgB,EAAE,CAAC,CAAC,gBAA0B;oBAC9C,gBAAgB,EAAE,CAAC,CAAC,gBAA0B;oBAC9C,YAAY,EAAE,CAAC,CAAC,YAAsB;oBACtC,UAAU,EAAE,CAAC,CAAC,UAAoB;oBAClC,UAAU,EAAE,CAAC,CAAC,UAAoB;oBAClC,KAAK,EAAE,CAAC,CAAC,KAAoB;oBAC7B,SAAS,EAAE,CAAC,CAAC,SAAsC;iBACpD,CAAC;YACJ,CAAC,CAAC,CAAC;YAGH,OAAO,CAAC,IAAI,CACV,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACP,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CACpE,CAAC;YAGF,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAGtD,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAEtC,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8CAA8C,KAAK,EAAE,CAAC,CAAC;YACrE,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,MAAoB;QACzC,IAAI,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAElD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG;gBACR,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,EAAE;gBACX,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,CAAC;gBACd,WAAW,EAAE,QAAQ;gBACrB,WAAW,EAAE,IAAI;gBACjB,UAAU,EAAE,IAAI;gBAChB,iBAAiB,EAAE,CAAC;aACrB,CAAC;QACJ,CAAC;QAGD,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAG7B,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,uBAAuB,EAAE,CAAC;YACrD,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,uBAAuB,CAAC,CAAC;QACpE,CAAC;QAGD,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;QAEzE,MAAM,QAAQ,GACZ,MAAM,CAAC,MAAM,CACX,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,KAAK,GAAG,OAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,EAC1D,CAAC,CACF,GAAG,MAAM,CAAC,MAAM,CAAC;QACpB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC;QAEnC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBACzB,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC;YACzC,CAAC;YACD,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC;YACtC,OAAO,CAAC,iBAAiB,GAAG,CAAC,CAAC;QAChC,CAAC;QAED,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAEO,YAAY,CAClB,SAAiB,EACjB,OAAuB;QAEvB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,YAAY,GAChB,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5E,MAAM,QAAQ,GACZ,MAAM,CAAC,MAAM,GAAG,CAAC;YACf,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM;YACf,CAAC,CAAC,CAAC,CAAC;QACR,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExC,IAAI,WAAW,GAAkB,IAAI,CAAC;QACtC,IAAI,UAAU,GAAkB,IAAI,CAAC;QACrC,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAE1B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,IAAI,CAAC,WAAW;oBAAE,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC;gBACjD,iBAAiB,EAAE,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC;gBAC9B,iBAAiB,GAAG,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GACf,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;QAErE,OAAO;YACL,SAAS;YACT,OAAO;YACP,YAAY;YACZ,WAAW;YACX,WAAW;YACX,WAAW;YACX,UAAU;YACV,iBAAiB;SAClB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,IAAY;QACrC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YACnD,KAAK,EAAE,wBAAwB;YAC/B,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,SAAS,CAAC;IACrC,CAAC;IAEO,eAAe;QAErB,OAAO,UAAU,EAAE,CAAC;IACtB,CAAC;CACF;AAMD,IAAI,YAAY,GAA8B,IAAI,CAAC;AAKnD,MAAM,UAAU,qBAAqB,CACnC,MAAiC;IAEjC,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAKD,MAAM,UAAU,uBAAuB;IACrC,YAAY,GAAG,IAAI,CAAC;IACpB,aAAa,CAAC,KAAK,EAAE,CAAC;AACxB,CAAC"}

package/dist/utils/rlvr/SQLSandbox.d.ts

@@ -0,0 +1,21 @@
+import { ExecutionSandbox, ExecutionResult, ResourceLimits } from "./types.js";
+export interface SQLInput {
+    setup?: string;
+    params?: unknown[];
+    readOnly?: boolean;
+}
+export declare class SQLSandbox implements ExecutionSandbox {
+    readonly language: "sql";
+    private limits;
+    private initialized;
+    private pythonPath;
+    private executorPath;
+    constructor();
+    initialize(): Promise<void>;
+    setLimits(limits: Partial<ResourceLimits>): void;
+    execute(code: string, input: unknown): Promise<ExecutionResult>;
+    cleanup(): Promise<void>;
+    private isSQLInput;
+    private checkPythonAvailable;
+}
+//# sourceMappingURL=SQLSandbox.d.ts.map

package/dist/utils/rlvr/SQLSandbox.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SQLSandbox.d.ts","sourceRoot":"","sources":["../../../src/utils/rlvr/SQLSandbox.ts"],"names":[],"mappings":"AAgBA,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,cAAc,EAEf,MAAM,YAAY,CAAC;AAUpB,MAAM,WAAW,QAAQ;IAEvB,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC;IAEnB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,qBAAa,UAAW,YAAW,gBAAgB;IACjD,QAAQ,CAAC,QAAQ,EAAG,KAAK,CAAU;IACnC,OAAO,CAAC,MAAM,CAAyC;IACvD,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAqB;IACvC,OAAO,CAAC,YAAY,CAAS;;IAMvB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAWjC,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,IAAI;IAU1C,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC,eAAe,CAAC;IAmL/D,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAO9B,OAAO,CAAC,UAAU;YAWJ,oBAAoB;CAmBnC"}

package/dist/utils/rlvr/SQLSandbox.js

@@ -0,0 +1,199 @@
+import { spawn } from "child_process";
+import * as path from "path";
+import { fileURLToPath } from "url";
+import { DEFAULT_LIMITS, } from "./types.js";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+export class SQLSandbox {
+    language = "sql";
+    limits = { ...DEFAULT_LIMITS };
+    initialized = false;
+    pythonPath = "python3";
+    executorPath;
+    constructor() {
+        this.executorPath = path.join(__dirname, "sql_executor.py");
+    }
+    async initialize() {
+        try {
+            await this.checkPythonAvailable();
+            this.initialized = true;
+        }
+        catch (error) {
+            throw new Error(`SQL sandbox initialization failed: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    setLimits(limits) {
+        this.limits = { ...this.limits, ...limits };
+    }
+    async execute(code, input) {
+        if (!this.initialized) {
+            throw new Error("Sandbox not initialized. Call initialize() first.");
+        }
+        const startTime = Date.now();
+        return new Promise((resolve) => {
+            let stdout = "";
+            let stderr = "";
+            let killed = false;
+            let processExited = false;
+            let executorInput;
+            if (this.isSQLInput(input)) {
+                executorInput = {
+                    query: code,
+                    setup: input.setup,
+                    params: input.params,
+                    readOnly: input.readOnly,
+                };
+            }
+            else if (Array.isArray(input)) {
+                executorInput = {
+                    query: code,
+                    params: input,
+                };
+            }
+            else if (input !== null && input !== undefined) {
+                executorInput = {
+                    query: code,
+                    params: [input],
+                };
+            }
+            else {
+                executorInput = {
+                    query: code,
+                };
+            }
+            const inputJson = JSON.stringify(executorInput);
+            const pythonProcess = spawn(this.pythonPath, [this.executorPath], {
+                stdio: ["pipe", "pipe", "pipe"],
+                timeout: this.limits.executionTimeout,
+            });
+            const timeoutId = setTimeout(() => {
+                if (!processExited) {
+                    killed = true;
+                    pythonProcess.kill("SIGKILL");
+                }
+            }, this.limits.executionTimeout);
+            pythonProcess.stdout?.on("data", (data) => {
+                stdout += data.toString();
+            });
+            pythonProcess.stderr?.on("data", (data) => {
+                stderr += data.toString();
+            });
+            pythonProcess.on("close", (exitCode) => {
+                processExited = true;
+                clearTimeout(timeoutId);
+                const executionTime = Date.now() - startTime;
+                if (killed) {
+                    resolve({
+                        status: "timeout",
+                        stdout: stdout.trim(),
+                        stderr: `Execution exceeded ${this.limits.executionTimeout}ms limit`,
+                        returnValue: undefined,
+                        executionTime,
+                        memoryUsed: 0,
+                        error: {
+                            type: "TimeoutError",
+                            message: `Execution exceeded ${this.limits.executionTimeout}ms limit`,
+                        },
+                    });
+                    return;
+                }
+                try {
+                    const result = JSON.parse(stdout);
+                    if (result.status === "success") {
+                        const returnValue = result.rows.length > 0
+                            ? result.rows
+                            : result.changes > 0
+                                ? { changes: result.changes, lastRowId: result.lastRowId }
+                                : result.rowCount;
+                        resolve({
+                            status: "success",
+                            stdout: "",
+                            stderr: stderr.trim(),
+                            returnValue,
+                            executionTime,
+                            memoryUsed: 0,
+                        });
+                    }
+                    else {
+                        resolve({
+                            status: "error",
+                            stdout: "",
+                            stderr: stderr.trim(),
+                            returnValue: undefined,
+                            executionTime,
+                            memoryUsed: 0,
+                            error: result.error || {
+                                type: "ExecutionError",
+                                message: "Unknown SQL execution error",
+                            },
+                        });
+                    }
+                }
+                catch (parseError) {
+                    resolve({
+                        status: "error",
+                        stdout: stdout.trim(),
+                        stderr: stderr.trim() || `Exit code: ${exitCode}`,
+                        returnValue: undefined,
+                        executionTime,
+                        memoryUsed: 0,
+                        error: {
+                            type: "ParseError",
+                            message: `Failed to parse executor output: ${parseError instanceof Error ? parseError.message : String(parseError)}`,
+                        },
+                    });
+                }
+            });
+            pythonProcess.on("error", (error) => {
+                processExited = true;
+                clearTimeout(timeoutId);
+                const executionTime = Date.now() - startTime;
+                resolve({
+                    status: "error",
+                    stdout: stdout.trim(),
+                    stderr: error.message,
+                    returnValue: undefined,
+                    executionTime,
+                    memoryUsed: 0,
+                    error: {
+                        type: "ProcessError",
+                        message: error.message,
+                    },
+                });
+            });
+            if (pythonProcess.stdin) {
+                pythonProcess.stdin.write(inputJson);
+                pythonProcess.stdin.end();
+            }
+        });
+    }
+    async cleanup() {
+        this.initialized = false;
+    }
+    isSQLInput(input) {
+        if (typeof input !== "object" || input === null) {
+            return false;
+        }
+        const obj = input;
+        return "setup" in obj || "params" in obj || "readOnly" in obj;
+    }
+    async checkPythonAvailable() {
+        return new Promise((resolve, reject) => {
+            const pythonProcess = spawn(this.pythonPath, ["--version"], {
+                stdio: ["pipe", "pipe", "pipe"],
+            });
+            pythonProcess.on("close", (exitCode) => {
+                if (exitCode === 0) {
+                    resolve();
+                }
+                else {
+                    reject(new Error("Python not found or not executable"));
+                }
+            });
+            pythonProcess.on("error", (error) => {
+                reject(new Error(`Python not available: ${error.message}`));
+            });
+        });
+    }
+}
+//# sourceMappingURL=SQLSandbox.js.map