genai-lite 0.1.1 → 0.1.3

package/dist/utils/prompt.test.js

@@ -0,0 +1,115 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const prompt_1 = require("./prompt");
+describe('Prompt Utilities', () => {
+    describe('countTokens', () => {
+        it('should return 0 for empty string', () => {
+            expect((0, prompt_1.countTokens)('')).toBe(0);
+        });
+        it('should count tokens for simple text', () => {
+            const text = 'Hello, world!';
+            const count = (0, prompt_1.countTokens)(text);
+            expect(count).toBeGreaterThan(0);
+            expect(count).toBeLessThan(text.length); // Tokens are typically fewer than characters
+        });
+        it('should count tokens with default gpt-4 model', () => {
+            const text = 'The quick brown fox jumps over the lazy dog';
+            const count = (0, prompt_1.countTokens)(text);
+            expect(count).toBeGreaterThan(0);
+        });
+        it('should count tokens with different models', () => {
+            const text = 'Testing different models';
+            const gpt4Count = (0, prompt_1.countTokens)(text, 'gpt-4');
+            const gpt35Count = (0, prompt_1.countTokens)(text, 'gpt-3.5-turbo');
+            expect(gpt4Count).toBeGreaterThan(0);
+            expect(gpt35Count).toBeGreaterThan(0);
+        });
+        it('should handle special characters and emojis', () => {
+            const text = '🚀 Special chars: @#$% and \n\t newlines';
+            const count = (0, prompt_1.countTokens)(text);
+            expect(count).toBeGreaterThan(0);
+        });
+        it('should fallback to estimate for invalid model', () => {
+            const text = 'Test fallback behavior';
+            const count = (0, prompt_1.countTokens)(text, 'invalid-model');
+            // Should fallback to length/4 estimate
+            expect(count).toBe(Math.ceil(text.length / 4));
+        });
+        it('should handle very long text', () => {
+            const longText = 'a'.repeat(10000);
+            const count = (0, prompt_1.countTokens)(longText);
+            expect(count).toBeGreaterThan(0);
+            expect(count).toBeLessThan(longText.length);
+        });
+    });
+    describe('getSmartPreview', () => {
+        const config = { minLines: 5, maxLines: 10 };
+        it('should return full content if shorter than maxLines', () => {
+            const content = 'Line 1\nLine 2\nLine 3';
+            const preview = (0, prompt_1.getSmartPreview)(content, config);
+            expect(preview).toBe(content);
+        });
+        it('should truncate at maxLines if no empty lines found', () => {
+            const lines = Array.from({ length: 20 }, (_, i) => `Line ${i + 1}`);
+            const content = lines.join('\n');
+            const preview = (0, prompt_1.getSmartPreview)(content, config);
+            const previewLines = preview.split('\n');
+            // Should extend up to maxLines when no empty lines are found
+            expect(previewLines.length).toBe(config.maxLines + 1); // +1 for truncation message
+            expect(preview).toContain('... (content truncated)');
+        });
+        it('should extend to next empty line within maxLines', () => {
+            const content = `Line 1
+Line 2
+Line 3
+Line 4
+Line 5
+Line 6
+
+Line 8
+Line 9
+Line 10
+Line 11`;
+            const preview = (0, prompt_1.getSmartPreview)(content, config);
+            const previewLines = preview.split('\n');
+            // Should include up to line 7 (the empty line)
+            expect(previewLines[6]).toBe('');
+            expect(preview).toContain('... (content truncated)');
+        });
+        it('should handle content exactly at maxLines', () => {
+            const lines = Array.from({ length: config.maxLines }, (_, i) => `Line ${i + 1}`);
+            const content = lines.join('\n');
+            const preview = (0, prompt_1.getSmartPreview)(content, config);
+            expect(preview).toBe(content);
+        });
+        it('should handle empty content', () => {
+            const preview = (0, prompt_1.getSmartPreview)('', config);
+            expect(preview).toBe('');
+        });
+        it('should handle content with multiple consecutive empty lines', () => {
+            const content = `Line 1
+Line 2
+
+
+Line 5
+Line 6
+
+Line 8
+Line 9
+Line 10
+Line 11`;
+            const preview = (0, prompt_1.getSmartPreview)(content, config);
+            const previewLines = preview.split('\n');
+            // Should stop at first empty line after minLines
+            expect(previewLines.length).toBeLessThanOrEqual(config.maxLines + 1);
+            expect(preview).toContain('... (content truncated)');
+        });
+        it('should respect maxLines limit even with empty lines', () => {
+            const lines = Array.from({ length: 15 }, (_, i) => i % 3 === 0 ? '' : `Line ${i + 1}`);
+            const content = lines.join('\n');
+            const preview = (0, prompt_1.getSmartPreview)(content, config);
+            const previewLines = preview.split('\n');
+            expect(previewLines.length).toBeLessThanOrEqual(config.maxLines + 1);
+        });
+    });
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "genai-lite",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "A lightweight, portable toolkit for interacting with various Generative AI APIs.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -30,16 +30,21 @@
     "url": "https://github.com/lacerbi/genai-lite/issues"
   },
   "homepage": "https://github.com/lacerbi/genai-lite#readme",
+  "files": [
+    "dist",
+    "src/config/presets.json"
+  ],
   "scripts": {
     "build": "tsc",
     "test": "jest --coverage",
-    "test:watch": "jest --watch"
+    "test:watch": "jest --watch",
+    "test:e2e": "npm run build && jest --config jest.e2e.config.js"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.52.0",
+    "@anthropic-ai/sdk": "^0.56.0",
     "@google/genai": "^1.0.1",
     "js-tiktoken": "^1.0.20",
-    "openai": "^4.103.0"
+    "openai": "^5.8.2"
   },
   "devDependencies": {
     "@types/jest": ">=30.0.0",

package/src/config/presets.json

@@ -0,0 +1,222 @@
+[
+  {
+    "id": "anthropic-claude-sonnet-4-20250514-default",
+    "displayName": "Anthropic - Claude Sonnet 4",
+    "description": "Default preset for Claude Sonnet 4.",
+    "providerId": "anthropic",
+    "modelId": "claude-sonnet-4-20250514",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "anthropic-claude-opus-4-20250514-default",
+    "displayName": "Anthropic - Claude Opus 4",
+    "description": "Default preset for Claude Opus 4.",
+    "providerId": "anthropic",
+    "modelId": "claude-opus-4-20250514",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "anthropic-claude-3-7-sonnet-20250219-default",
+    "displayName": "Anthropic - Claude 3.7 Sonnet",
+    "description": "Default preset for Claude 3.7 Sonnet.",
+    "providerId": "anthropic",
+    "modelId": "claude-3-7-sonnet-20250219",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "anthropic-claude-3-5-sonnet-20241022-default",
+    "displayName": "Anthropic - Claude 3.5 Sonnet",
+    "description": "Default preset for Claude 3.5 Sonnet.",
+    "providerId": "anthropic",
+    "modelId": "claude-3-5-sonnet-20241022",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "anthropic-claude-3-5-haiku-20241022-default",
+    "displayName": "Anthropic - Claude 3.5 Haiku",
+    "description": "Default preset for Claude 3.5 Haiku.",
+    "providerId": "anthropic",
+    "modelId": "claude-3-5-haiku-20241022",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "google-gemini-2.5-pro",
+    "displayName": "Google - Gemini 2.5 Pro",
+    "description": "Default preset for Gemini 2.5 Pro.",
+    "providerId": "gemini",
+    "modelId": "gemini-2.5-pro",
+    "settings": {
+      "temperature": 0.3,
+      "geminiSafetySettings": [
+        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE" },
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "threshold": "BLOCK_NONE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "threshold": "BLOCK_NONE"
+        },
+        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE" }
+      ]
+    }
+  },
+  {
+    "id": "google-gemini-2.5-flash",
+    "displayName": "Google - Gemini 2.5 Flash",
+    "description": "Default preset for Gemini 2.5 Flash.",
+    "providerId": "gemini",
+    "modelId": "gemini-2.5-flash",
+    "settings": {
+      "temperature": 0.3,
+      "geminiSafetySettings": [
+        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE" },
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "threshold": "BLOCK_NONE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "threshold": "BLOCK_NONE"
+        },
+        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE" }
+      ]
+    }
+  },
+  {
+    "id": "google-gemini-2.5-flash-lite-preview",
+    "displayName": "Google - Gemini 2.5 Flash-Lite Preview",
+    "description": "Default preset for Gemini 2.5 Flash-Lite.",
+    "providerId": "gemini",
+    "modelId": "gemini-2.5-flash-lite-preview-06-17",
+    "settings": {
+      "temperature": 0.3,
+      "geminiSafetySettings": [
+        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE" },
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "threshold": "BLOCK_NONE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "threshold": "BLOCK_NONE"
+        },
+        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE" }
+      ]
+    }
+  },
+  {
+    "id": "google-gemini-2.0-flash-default",
+    "displayName": "Google - Gemini 2.0 Flash",
+    "description": "Default preset for Gemini 2.0 Flash.",
+    "providerId": "gemini",
+    "modelId": "gemini-2.0-flash",
+    "settings": {
+      "temperature": 0.3,
+      "geminiSafetySettings": [
+        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE" },
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "threshold": "BLOCK_NONE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "threshold": "BLOCK_NONE"
+        },
+        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE" }
+      ]
+    }
+  },
+  {
+    "id": "google-gemini-2.0-flash-lite-default",
+    "displayName": "Google - Gemini 2.0 Flash Lite",
+    "description": "Default preset for Gemini 2.0 Flash Lite.",
+    "providerId": "gemini",
+    "modelId": "gemini-2.0-flash-lite",
+    "settings": {
+      "temperature": 0.3,
+      "geminiSafetySettings": [
+        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE" },
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "threshold": "BLOCK_NONE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "threshold": "BLOCK_NONE"
+        },
+        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE" }
+      ]
+    }
+  },
+  {
+    "id": "openai-o4-mini-default",
+    "displayName": "OpenAI - o4-mini",
+    "description": "Default preset for o4-mini.",
+    "providerId": "openai",
+    "modelId": "o4-mini",
+    "settings": {
+      "temperature": 1.0
+    }
+  },
+  {
+    "id": "openai-gpt-4.1-default",
+    "displayName": "OpenAI - GPT-4.1",
+    "description": "Default preset for GPT-4.1.",
+    "providerId": "openai",
+    "modelId": "gpt-4.1",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "openai-gpt-4.1-mini-default",
+    "displayName": "OpenAI - GPT-4.1 Mini",
+    "description": "Default preset for GPT-4.1 Mini.",
+    "providerId": "openai",
+    "modelId": "gpt-4.1-mini",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "openai-gpt-4.1-nano-default",
+    "displayName": "OpenAI - GPT-4.1 Nano",
+    "description": "Default preset for GPT-4.1 Nano.",
+    "providerId": "openai",
+    "modelId": "gpt-4.1-nano",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "mistral-codestral-2501-default",
+    "displayName": "Mistral AI - Codestral",
+    "description": "Default preset for Codestral.",
+    "providerId": "mistral",
+    "modelId": "codestral-2501",
+    "settings": {
+      "temperature": 0.3
+    }
+  },
+  {
+    "id": "mistral-devstral-small-2505-default",
+    "displayName": "Mistral AI - Devstral Small",
+    "description": "Default preset for Devstral Small.",
+    "providerId": "mistral",
+    "modelId": "devstral-small-2505",
+    "settings": {
+      "temperature": 0.3
+    }
+  }
+]