tuneprompt 1.0.6 → 1.0.7

package/dist/commands/init.js

@@ -91,9 +91,6 @@ async function initCommand() {
         const envContent = `OPENAI_API_KEY=your_key_here
 ANTHROPIC_API_KEY=your_key_here
 OPENROUTER_API_KEY=your_key_here
-
-# For self-hosted or local testing:
-# TUNEPROMPT_API_URL=http://localhost:3000
 `;
         fs.writeFileSync(envPath, envContent);
         console.log(chalk_1.default.green('✓ Created .env'));

package/dist/engine/optimizer.d.ts

@@ -2,6 +2,7 @@ import { FailedTest, OptimizationResult } from '../types/fix';
 export declare class PromptOptimizer {
     private anthropic?;
     private openai?;
+    private openrouter?;
     constructor();
     /**
      * Main optimization method

package/dist/engine/optimizer.js

@@ -44,19 +44,34 @@ const constraintExtractor_1 = require("./constraintExtractor");
 class PromptOptimizer {
     anthropic;
     openai;
+    openrouter;
     constructor() {
         const anthropicKey = process.env.ANTHROPIC_API_KEY;
-        if (anthropicKey && !anthropicKey.startsWith('api_key') && anthropicKey !== 'phc_xxxxx') {
+        if (anthropicKey &&
+            !anthropicKey.includes('your_key') &&
+            !anthropicKey.startsWith('api_key') &&
+            anthropicKey !== 'phc_xxxxx') {
             this.anthropic = new sdk_1.default({
                 apiKey: anthropicKey
             });
         }
         const openaiKey = process.env.OPENAI_API_KEY;
-        if (openaiKey && !openaiKey.startsWith('api_key')) {
+        if (openaiKey && !openaiKey.includes('your_key')) {
             this.openai = new openai_1.default({
                 apiKey: openaiKey
             });
         }
+        const openrouterKey = process.env.OPENROUTER_API_KEY;
+        if (openrouterKey && !openrouterKey.includes('your_key')) {
+            this.openrouter = new openai_1.default({
+                baseURL: 'https://openrouter.ai/api/v1',
+                apiKey: openrouterKey,
+                defaultHeaders: {
+                    'HTTP-Referer': 'https://tuneprompt.xyz',
+                    'X-Title': 'TunePrompt CLI',
+                },
+            });
+        }
     }
     /**
      * Main optimization method
@@ -118,7 +133,7 @@ class PromptOptimizer {
                 if (provider === 'anthropic' && this.anthropic) {
                     console.log(`⚡ Using Anthropic for candidate generation...`);
                     const response = await this.anthropic.messages.create({
-                        model: 'claude-sonnet-4-20250514',
+                        model: 'claude-3-5-sonnet-20240620',
                         max_tokens: 4000,
                         temperature: 0.7, // Some creativity for prompt rewriting
                         messages: [{
@@ -174,16 +189,34 @@ class PromptOptimizer {
                         }
                     ];
                 }
-                else if (provider === 'openrouter') {
-                    // For OpenRouter, we'll use the shadowTester to get a response
+                else if (provider === 'openrouter' && this.openrouter) {
                     console.log(`⚡ Using OpenRouter for candidate generation...`);
-                    // Since OpenRouter is used in shadow testing, we'll use a different approach
-                    // For now, we'll return a basic fallback since OpenRouter doesn't support structured outputs as well
-                    return [{
-                            prompt: this.createFallbackPrompt(failedTest),
-                            reasoning: 'Generated using fallback method',
+                    const response = await this.openrouter.chat.completions.create({
+                        model: 'anthropic/claude-3-sonnet', // Default robust model on OpenRouter
+                        messages: [{
+                                role: 'user',
+                                content: metaPrompt
+                            }],
+                        response_format: { type: 'json_object' }
+                    });
+                    const content = response.choices[0]?.message?.content;
+                    if (!content) {
+                        // Fallback if model doesn't support JSON mode or returns empty
+                        throw new Error('No content returned from OpenRouter');
+                    }
+                    const parsed = JSON.parse(content);
+                    return [
+                        {
+                            prompt: parsed.candidateA.prompt,
+                            reasoning: parsed.candidateA.reasoning,
                             score: 0
-                        }];
+                        },
+                        {
+                            prompt: parsed.candidateB.prompt,
+                            reasoning: parsed.candidateB.reasoning,
+                            score: 0
+                        }
+                    ];
                 }
             }
             catch (error) {
@@ -195,7 +228,7 @@ class PromptOptimizer {
         console.error('All providers failed for candidate generation');
         return [{
                 prompt: this.createFallbackPrompt(failedTest),
-                reasoning: 'Fallback prompt with basic improvements',
+                reasoning: 'Generated using fallback method',
                 score: 0
             }];
     }

package/dist/services/cloud.service.js

@@ -10,7 +10,7 @@ class CloudService {
     backendUrl;
     subscriptionId;
     constructor() {
-        this.backendUrl = process.env.TUNEPROMPT_API_URL || process.env.BACKEND_URL || 'https://api.tuneprompt.com';
+        this.backendUrl = process.env.TUNEPROMPT_API_URL || process.env.BACKEND_URL || 'https://i8e3mu8jlk.execute-api.ap-south-1.amazonaws.com/dev';
     }
     async init() {
         // Load subscription ID from local storage (Phase 2 activation)
@@ -21,9 +21,16 @@ class CloudService {
         return license?.subscriptionId;
     }
     async uploadRun(data) {
-        if (!this.subscriptionId) {
+        // Enforce Pro plan check
+        const license = (0, license_1.loadLicense)();
+        if (!this.subscriptionId || !license) {
             return { success: false, error: 'Not activated. Run `tuneprompt activate` first.' };
         }
+        // Check for specific pro plans (if we ever add free tiers)
+        const proPlans = ['pro-monthly', 'pro-yearly', 'lifetime'];
+        if (!proPlans.includes(license.plan)) {
+            return { success: false, error: 'Cloud features are restricted to Pro users. Please upgrade your plan.' };
+        }
         try {
             const response = await axios_1.default.post(`${this.backendUrl}/api/cloud/ingest-run`, data, {
                 headers: {

package/dist/storage/database.js

@@ -93,9 +93,13 @@ class TestDatabase {
         }
         catch (e) {
             // Column might already exist
-            if (!e.message.includes('duplicate column name')) {
-                // Ignore safe errors
-            }
+        }
+        // Migration for provider column
+        try {
+            this.db.exec(`ALTER TABLE test_results ADD COLUMN provider TEXT`);
+        }
+        catch (e) {
+            // Column might already exist
         }
         // Run external migrations (Phase 2)
         (0, migrate_1.runMigrations)(this.db).catch((err) => {
@@ -110,11 +114,11 @@ class TestDatabase {
         insertRun.run(run.id, run.timestamp.getTime(), run.totalTests, run.passed, run.failed, run.duration);
         const insertResult = this.db.prepare(`
       INSERT INTO test_results 
-      (id, run_id, description, prompt, variables, expect, config, file_path, status, score, actual_output, expected_output, error, duration, tokens, cost)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      (id, run_id, description, prompt, variables, expect, config, file_path, status, score, actual_output, expected_output, error, duration, tokens, cost, provider)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
     `);
         for (const result of run.results) {
-            insertResult.run(result.id, run.id, result.testCase.description, typeof result.testCase.prompt === 'string' ? result.testCase.prompt : JSON.stringify(result.testCase.prompt), result.testCase.variables ? JSON.stringify(result.testCase.variables) : null, typeof result.testCase.expect === 'string' ? result.testCase.expect : JSON.stringify(result.testCase.expect), result.testCase.config ? JSON.stringify(result.testCase.config) : null, result.testCase.filePath || null, result.status, result.score, result.actualOutput, result.expectedOutput, result.error || null, result.metadata.duration, result.metadata.tokens || null, result.metadata.cost || null);
+            insertResult.run(result.id, run.id, result.testCase.description, typeof result.testCase.prompt === 'string' ? result.testCase.prompt : JSON.stringify(result.testCase.prompt), result.testCase.variables ? JSON.stringify(result.testCase.variables) : null, typeof result.testCase.expect === 'string' ? result.testCase.expect : JSON.stringify(result.testCase.expect), result.testCase.config ? JSON.stringify(result.testCase.config) : null, result.testCase.filePath || null, result.status, result.score, result.actualOutput, result.expectedOutput, result.error || null, result.metadata.duration, result.metadata.tokens || null, result.metadata.cost || null, result.metadata.provider || null);
         }
     }
     getRecentRuns(limit = 10) {
@@ -194,7 +198,8 @@ class TestDatabase {
                     duration: r.duration,
                     timestamp: new Date(),
                     tokens: r.tokens,
-                    cost: r.cost
+                    cost: r.cost,
+                    provider: r.provider
                 }
             };
         });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tuneprompt",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "Industrial-grade testing framework for LLM prompts",
   "repository": {
     "type": "git",