lazlo-ai 1.0.0

package/test-full.mjs

@@ -0,0 +1,552 @@
+/**
+ * COMPREHENSIVE FEATURE TEST - ALL LAZLO JS FEATURES
+ */
+
+import { 
+  // Providers
+  OpenAI,
+  Ollama,
+  
+  // Chains
+  LLMChain,
+  SequentialChain,
+  PromptTemplate,
+  RetrievalQAChain,
+  createRetrievalChain,
+  
+  // Router
+  SmartRouter,
+  
+  // Cache
+  SemanticCache,
+  
+  // Memory
+  BufferMemory,
+  TokenBudgetMemory,
+  
+  // Tools
+  Tool,
+  tool,
+  defineTool,
+  calculator,
+  search,
+  weather,
+  KeiroSearch,
+  
+  // Tracing
+  tracer,
+  Tracer,
+  
+  // Evals
+  LLMJudge,
+  EvalCriteria,
+  
+  // Text Splitters
+  CharacterTextSplitter,
+  RecursiveCharacterTextSplitter,
+  createCharacterSplitter,
+  createRecursiveSplitter,
+  
+  // Vector Stores
+  InMemoryVectorStore,
+  createVectorStore,
+  
+  // Utils
+  logger,
+  getPricing,
+  setPricing,
+  calculateCost,
+  estimateCost,
+  estimateTokens,
+} from './dist/index.js';
+
+console.log('\n🧪 LAZLO JS COMPREHENSIVE FEATURE TEST\n');
+console.log('=' .repeat(60));
+
+let passed = 0;
+let failed = 0;
+
+function test(name, fn) {
+  try {
+    const result = fn();
+    if (result && typeof result.then === 'function') {
+      result.then(() => {
+        console.log(`✅ ${name}`);
+        passed++;
+      }).catch(e => {
+        console.log(`❌ ${name}: ${e.message}`);
+        failed++;
+      });
+    } else {
+      console.log(`✅ ${name}`);
+      passed++;
+    }
+  } catch (e) {
+    console.log(`❌ ${name}: ${e.message}`);
+    failed++;
+  }
+}
+
+// ============================================================================
+// 1. CORE TYPES
+// ============================================================================
+
+console.log('\n📦 1. CORE TYPES');
+console.log('-'.repeat(60));
+
+test('Message type works (runtime check)', () => {
+  const msg = { role: 'user', content: 'test' };
+  return msg.role === 'user' && msg.content === 'test';
+});
+
+test('ChatResponse type works (runtime check)', () => {
+  const resp = { content: 'hello' };
+  return resp.content === 'hello';
+});
+
+// ============================================================================
+// 2. PROVIDERS
+// ============================================================================
+
+console.log('\n🔌 2. PROVIDERS');
+console.log('-'.repeat(60));
+
+test('OpenAI provider instantiates', () => {
+  const openai = new OpenAI('test-key', 'gpt-4o');
+  return openai.defaultModel === 'gpt-4o';
+});
+
+test('OpenAI supports tools', () => {
+  const openai = new OpenAI('test-key');
+  return openai.supportsTools === true;
+});
+
+test('OpenAI supports structured output', () => {
+  const openai = new OpenAI('test-key');
+  return openai.supportsStructuredOutput === true;
+});
+
+test('Ollama provider instantiates', () => {
+  const ollama = new Ollama({ model: 'llama3.2' });
+  return ollama.model === 'llama3.2';
+});
+
+test('Ollama supports tools', () => {
+  const ollama = new Ollama();
+  return ollama.supportsTools === true;
+});
+
+// ============================================================================
+// 3. PROMPT TEMPLATE
+// ============================================================================
+
+console.log('\n📝 3. PROMPT TEMPLATE');
+console.log('-'.repeat(60));
+
+test('PromptTemplate basic format', () => {
+  const p = new PromptTemplate('Hello {name}!');
+  return p.format({ name: 'World' }) === 'Hello World!';
+});
+
+test('PromptTemplate inputVars extracted', () => {
+  const p = new PromptTemplate('Hello {name}, you are {role}');
+  return p.inputVars.length === 2;
+});
+
+// ============================================================================
+// 4. LLM CHAIN
+// ============================================================================
+
+console.log('\n⛓️ 4. LLM CHAIN');
+console.log('-'.repeat(60));
+
+test('LLMChain instantiates with mock LLM', () => {
+  const mockLlm = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'response', usage: {} }; },
+  };
+  
+  const chain = new LLMChain({
+    llm: mockLlm,
+    prompt: new PromptTemplate('Hello {name}'),
+  });
+  
+  return chain !== undefined;
+});
+
+test('LLMChain estimateCost works', () => {
+  const mockLlm = {
+    defaultModel: 'gpt-4o',
+    supportsTools: false,
+    supportsStructuredOutput: false,
+  };
+  
+  const chain = new LLMChain({
+    llm: mockLlm,
+    prompt: new PromptTemplate('Hello {name}'),
+  });
+  
+  const estimate = chain.estimateCost({ name: 'World' });
+  return estimate.model === 'gpt-4o' && estimate.estimatedCostUsd >= 0;
+});
+
+// ============================================================================
+// 5. SEQUENTIAL CHAIN
+// ============================================================================
+
+console.log('\n🔄 5. SEQUENTIAL CHAIN');
+console.log('-'.repeat(60));
+
+test('SequentialChain instantiates', () => {
+  const mockLlm = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'response' }; },
+  };
+  
+  const chain1 = new LLMChain({
+    llm: mockLlm,
+    prompt: new PromptTemplate('Step 1: {input}'),
+  });
+  
+  const chain2 = new LLMChain({
+    llm: mockLlm,
+    prompt: new PromptTemplate('Step 2: {output}'),
+  });
+  
+  const seqChain = new SequentialChain({
+    chains: [chain1, chain2],
+    inputVariables: ['input'],
+    outputVariables: ['text'],
+  });
+  
+  return seqChain !== undefined;
+});
+
+// ============================================================================
+// 6. RAG CHAIN
+// ============================================================================
+
+console.log('\n📚 6. RAG CHAIN');
+console.log('-'.repeat(60));
+
+test('RetrievalQAChain factory works', () => {
+  const mockLlm = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'answer' }; },
+  };
+  
+  const mockRetriever = {
+    async getRelevantDocuments() { 
+      return [{ pageContent: 'doc1', metadata: {} }]; 
+    },
+  };
+  
+  const rag = createRetrievalChain(mockLlm, mockRetriever);
+  return rag !== undefined;
+});
+
+// ============================================================================
+// 7. SMART ROUTER
+// ============================================================================
+
+console.log('\n🧭 7. SMART ROUTER');
+console.log('-'.repeat(60));
+
+test('SmartRouter instantiates', () => {
+  const fast = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'fast response' }; },
+  };
+  
+  const smart = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'smart response' }; },
+  };
+  
+  const router = new SmartRouter({ fast, smart, threshold: 0.5 });
+  return router !== undefined;
+});
+
+test('SmartRouter printStats works', () => {
+  const fast = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'fast' }; },
+  };
+  
+  const smart = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { return { content: 'smart' }; },
+  };
+  
+  const router = new SmartRouter({ fast, smart });
+  router.printStats();
+  return true;
+});
+
+// ============================================================================
+// 8. SEMANTIC CACHE
+// ============================================================================
+
+console.log('\n💾 8. SEMANTIC CACHE');
+console.log('-'.repeat(60));
+
+test('SemanticCache instantiates', () => {
+  const cache = new SemanticCache({ threshold: 0.9 });
+  return cache !== undefined;
+});
+
+test('SemanticCache stores and looks up', () => {
+  const cache = new SemanticCache({ threshold: 0.5 });
+  cache.put('What is AI?', 'AI is artificial intelligence.');
+  const result = cache.lookup('What is AI?');
+  return result === 'AI is artificial intelligence.';
+});
+
+test('SemanticCache printStats works', () => {
+  const cache = new SemanticCache();
+  cache.put('test', 'result');
+  cache.lookup('test');
+  cache.printStats();
+  return true;
+});
+
+// ============================================================================
+// 9. MEMORY
+// ============================================================================
+
+console.log('\n🧠 9. MEMORY');
+console.log('-'.repeat(60));
+
+test('BufferMemory save and load', async () => {
+  const memory = new BufferMemory();
+  await memory.saveContext({ input: 'Hello' }, { text: 'Hi there!' });
+  const vars = await memory.loadMemoryVariables({});
+  return vars.history !== undefined;
+});
+
+test('BufferMemory with max tokens', () => {
+  const memory = new BufferMemory({ maxTokenLimit: 100 });
+  return memory !== undefined;
+});
+
+test('TokenBudgetMemory instantiates', () => {
+  const memory = new TokenBudgetMemory({ maxTokens: 4000 });
+  return memory !== undefined;
+});
+
+test('TokenBudgetMemory tokenUsage', async () => {
+  const memory = new TokenBudgetMemory({ maxTokens: 4000 });
+  await memory.saveContext({ input: 'Hello' }, { text: 'Hi!' });
+  const usage = memory.tokenUsage;
+  return usage.budget === 4000;
+});
+
+// ============================================================================
+// 10. TOOLS
+// ============================================================================
+
+console.log('\n🔧 10. TOOLS');
+console.log('-'.repeat(60));
+
+test('Tool class works', async () => {
+  const t = new Tool(
+    (args) => args.x + args.y,
+    { name: 'add', description: 'Add numbers' }
+  );
+  const result = await t.invoke({ x: 5, y: 3 });
+  return result === '8';
+});
+
+test('defineTool factory works', async () => {
+  const multiply = defineTool(
+    (args) => args.a * args.b,
+    'multiply',
+    'Multiply two numbers'
+  );
+  const result = await multiply.invoke({ a: 4, b: 5 });
+  return result === '20';
+});
+
+test('calculator tool works', async () => {
+  const result = await calculator.invoke({ expression: '2 + 2' });
+  return result === '4';
+});
+
+test('search tool exists', () => {
+  return search.name === 'search';
+});
+
+test('weather tool exists', () => {
+  return weather.name === 'weather';
+});
+
+test('KeiroSearch instantiates', () => {
+  const keiro = new KeiroSearch();
+  return keiro.name === 'keiro_search';
+});
+
+// ============================================================================
+// 11. TRACING
+// ============================================================================
+
+console.log('\n📊 11. TRACING');
+console.log('-'.repeat(60));
+
+test('Tracer trace context works', () => {
+  const { tracer: t, span, end } = tracer.trace('test');
+  span('step1').finish();
+  end();
+  return true;
+});
+
+test('Tracer event works', () => {
+  tracer.event('test_event', 'llm', 100, 50);
+  return true;
+});
+
+test('Tracer printSummary works', () => {
+  tracer.printSummary();
+  return true;
+});
+
+test('Tracer clear works', () => {
+  tracer.clear();
+  return true;
+});
+
+// ============================================================================
+// 12. EVALS
+// ============================================================================
+
+console.log('\n⚖️ 12. EVALS');
+console.log('-'.repeat(60));
+
+test('EvalCriteria enum exists', () => {
+  return EvalCriteria.ACCURACY === 'accuracy';
+});
+
+test('LLMJudge instantiates', () => {
+  const mockLlm = {
+    supportsTools: false,
+    supportsStructuredOutput: false,
+    async invoke() { 
+      return { content: '{"score": 8, "explanation": "good", "passed": true}' }; 
+    },
+  };
+  
+  const judge = new LLMJudge(mockLlm);
+  return judge !== undefined;
+});
+
+// ============================================================================
+// 13. TEXT SPLITTERS
+// ============================================================================
+
+console.log('\n✂️ 13. TEXT SPLITTERS');
+console.log('-'.repeat(60));
+
+test('CharacterTextSplitter works', () => {
+  const splitter = new CharacterTextSplitter({ chunkSize: 10 });
+  const chunks = splitter.splitText('Hello World Test');
+  return chunks.length > 1;
+});
+
+test('RecursiveCharacterTextSplitter works', () => {
+  const splitter = new RecursiveCharacterTextSplitter({ chunkSize: 20 });
+  const chunks = splitter.splitText('This is a long text that needs to be split.');
+  return chunks.length >= 1;
+});
+
+test('createCharacterSplitter factory', () => {
+  const splitter = createCharacterSplitter({ chunkSize: 5 });
+  return splitter !== undefined;
+});
+
+test('createRecursiveSplitter factory', () => {
+  const splitter = createRecursiveSplitter({ chunkSize: 10 });
+  return splitter !== undefined;
+});
+
+// ============================================================================
+// 14. VECTOR STORES
+// ============================================================================
+
+console.log('\n🗄️ 14. VECTOR STORES');
+console.log('-'.repeat(60));
+
+test('InMemoryVectorStore instantiates', () => {
+  const mockEmbeddings = {
+    async embedQuery() { return new Array(384).fill(0); },
+    async embedDocuments() { return [new Array(384).fill(0)]; },
+  };
+  
+  const store = new InMemoryVectorStore(mockEmbeddings);
+  return store !== undefined;
+});
+
+test('createVectorStore factory', () => {
+  const store = createVectorStore();
+  return store !== undefined;
+});
+
+// ============================================================================
+// 15. UTILS
+// ============================================================================
+
+console.log('\n🔧 15. UTILS');
+console.log('-'.repeat(60));
+
+test('getPricing works', () => {
+  const pricing = getPricing('gpt-4o');
+  return pricing.input === 2.5 && pricing.output === 10;
+});
+
+test('setPricing works', () => {
+  setPricing('test-model', 1.0, 2.0);
+  const pricing = getPricing('test-model');
+  return pricing.input === 1.0;
+});
+
+test('calculateCost works', () => {
+  const cost = calculateCost('gpt-4o', 1000, 500);
+  return cost > 0;
+});
+
+test('estimateCost works', () => {
+  const cost = estimateCost('gpt-4o', 'Hello world', 'Hi there');
+  return cost >= 0;
+});
+
+test('estimateTokens works', () => {
+  const tokens = estimateTokens('Hello world');
+  return tokens > 0;
+});
+
+test('logger works', () => {
+  logger.info('Test log');
+  logger.debug('Debug log');
+  logger.warn('Warning log');
+  logger.error('Error log');
+  return true;
+});
+
+// ============================================================================
+// SUMMARY
+// ============================================================================
+
+console.log('\n' + '='.repeat(60));
+console.log(`📊 TEST SUMMARY: ${passed} passed, ${failed} failed`);
+console.log('='.repeat(60));
+
+if (failed === 0) {
+  console.log('\n🎉 ALL FEATURES VERIFIED AND WORKING!\n');
+} else {
+  console.log(`\n⚠️ ${failed} tests failed\n`);
+}

package/test.mjs

@@ -0,0 +1,74 @@
+/**
+ * Test script for lazlo-js
+ */
+
+import { 
+  OpenAI, 
+  LLMChain, 
+  PromptTemplate, 
+  SmartRouter, 
+  SemanticCache,
+  BufferMemory,
+  TokenBudgetMemory,
+  Tool,
+  KeiroSearch,
+  tracer,
+  LLMJudge,
+  EvalCriteria,
+  CharacterTextSplitter,
+  RecursiveCharacterTextSplitter,
+  InMemoryVectorStore,
+  logger,
+  getPricing,
+  setPricing,
+} from './dist/index.js';
+
+console.log('✅ All imports successful!\n');
+
+// Test 1: Pricing
+console.log('=== Test 1: Pricing ===');
+const gpt4oPricing = getPricing('gpt-4o');
+console.log('gpt-4o pricing:', gpt4oPricing);
+setPricing('test-model', 1.0, 2.0);
+console.log('Custom pricing set:', getPricing('test-model'));
+
+// Test 2: PromptTemplate
+console.log('\n=== Test 2: PromptTemplate ===');
+const prompt = new PromptTemplate('Hello {name}, you are {role}');
+const formatted = prompt.format({ name: 'World', role: 'AI' });
+console.log('Formatted prompt:', formatted);
+
+// Test 3: BufferMemory
+console.log('\n=== Test 3: BufferMemory ===');
+const memory = new BufferMemory();
+await memory.saveContext({ input: 'Hi' }, { text: 'Hello!' });
+const vars = await memory.loadMemoryVariables({});
+console.log('Memory variables:', vars);
+
+// Test 4: Text Splitter
+console.log('\n=== Test 4: TextSplitter ===');
+const splitter = new RecursiveCharacterTextSplitter({ chunkSize: 50 });
+const chunks = splitter.splitText('This is a long text that needs to be split into smaller chunks for processing.');
+console.log('Chunks:', chunks);
+
+// Test 5: Tracer
+console.log('\n=== Test 5: Tracer ===');
+const { tracer: t, span, end } = tracer.trace('test_run');
+span('test_span').finish();
+end();
+tracer.printSummary();
+
+// Test 6: Logger
+console.log('\n=== Test 6: Logger ===');
+logger.info('Test log message');
+
+// Test 7: Tool
+console.log('\n=== Test 7: Tool ===');
+const addTool = new Tool(
+  (args) => String(args.a + args.b),
+  { name: 'add', description: 'Add two numbers' }
+);
+const result = await addTool.invoke({ a: 5, b: 3 });
+console.log('Tool result:', result);
+
+console.log('\n✅ All tests passed!');

package/tsconfig.json

@@ -0,0 +1,30 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "lib": ["ES2022", "DOM"],
+    "moduleResolution": "node",
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "allowSyntheticDefaultImports": true,
+    "noImplicitAny": true,
+    "strictNullChecks": true,
+    "strictFunctionTypes": true,
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noImplicitReturns": true,
+    "noFallthroughCasesInSwitch": true,
+    "allowImportingTsExtensions": false,
+    "isolatedModules": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist"]
+}