llmjs2 1.3.9 → 1.7.1

package/test.js

@@ -1,246 +0,0 @@
-// Import all modules
-const { completion: llmCompletion, LLMJS2: LLMJS2Class, router, app } = require('./index');
-
-// Basic validation tests
-async function runTests() {
-  console.log('Running LLMJS2 comprehensive tests...\n');
-
-  let testCount = 0;
-  let passedCount = 0;
-
-  // Test 1: Input validation
-  testCount++;
-  try {
-    await llmCompletion('');
-    console.log('❌ Test 1 failed: Should reject empty prompt');
-  } catch (error) {
-    console.log('✅ Test 1 passed: Empty prompt rejected');
-    passedCount++;
-  }
-
-  // Test 2: Invalid input type
-  testCount++;
-  try {
-    await llmCompletion(123);
-    console.log('❌ Test 2 failed: Should reject non-string/non-object input');
-  } catch (error) {
-    console.log('✅ Test 2 passed: Invalid input type rejected');
-    passedCount++;
-  }
-
-  // Test 3: Invalid messages array
-  testCount++;
-  try {
-    await llmCompletion({ messages: [] });
-    console.log('❌ Test 3 failed: Should reject empty messages array');
-  } catch (error) {
-    console.log('✅ Test 3 passed: Empty messages array rejected');
-    passedCount++;
-  }
-
-  // Test 4: Invalid message format
-  testCount++;
-  try {
-    await llmCompletion({ messages: [{ content: 'test' }] });
-    console.log('❌ Test 4 failed: Should reject message without role');
-  } catch (error) {
-    console.log('✅ Test 4 passed: Message without role rejected');
-    passedCount++;
-  }
-
-  // Test 5: Invalid provider
-  testCount++;
-  try {
-    const llm = new LLMJS2();
-    llm.getProvider('unknown/test');
-    console.log('❌ Test 5 failed: Should reject unknown provider');
-  } catch (error) {
-    console.log('✅ Test 5 passed: Unknown provider rejected -', error.message);
-    passedCount++;
-  }
-
-  // Test 6: LLMJS2 class instantiation
-  testCount++;
-  try {
-    const llm = new LLMJS2Class();
-    console.log('✅ Test 6 passed: LLMJS2 class instantiated successfully');
-    passedCount++;
-  } catch (error) {
-    console.log('❌ Test 6 failed:', error.message);
-  }
-
-  // Test 7: Provider availability check (without API keys)
-  testCount++;
-  try {
-    // Temporarily clear API keys for this test
-    const originalKeys = {
-      openai: process.env.OPENAI_API_KEY,
-      ollama: process.env.OLLAMA_API_KEY,
-      openrouter: process.env.OPEN_ROUTER_API_KEY
-    };
-
-    delete process.env.OPENAI_API_KEY;
-    delete process.env.OLLAMA_API_KEY;
-    delete process.env.OPEN_ROUTER_API_KEY;
-
-    const llm = new LLMJS2Class();
-    const available = llm.getAvailableProviders();
-
-    // Restore original keys
-    process.env.OPENAI_API_KEY = originalKeys.openai;
-    process.env.OLLAMA_API_KEY = originalKeys.ollama;
-    process.env.OPEN_ROUTER_API_KEY = originalKeys.openrouter;
-
-    if (available.length === 0) {
-      console.log('✅ Test 7 passed: No providers available without API keys');
-      passedCount++;
-    } else {
-      console.log('❌ Test 7 failed: Should have no providers without API keys');
-    }
-  } catch (error) {
-    console.log('❌ Test 7 failed:', error.message);
-  }
-
-  // Test 8: Model parsing
-  testCount++;
-  try {
-    const llm = new LLMJS2Class();
-    const { provider, model } = llm.parseModel('openai/gpt-3.5-turbo');
-    if (provider === 'openai' && model === 'gpt-3.5-turbo') {
-      console.log('✅ Test 8 passed: Model parsing works correctly');
-      passedCount++;
-    } else {
-      console.log('❌ Test 8 failed: Model parsing incorrect');
-    }
-  } catch (error) {
-    console.log('❌ Test 8 failed:', error.message);
-  }
-
-  // Test 9: Router creation
-  testCount++;
-  try {
-    const modelList = [
-      {
-        model_name: 'test',
-        llm_params: {
-          model: 'ollama/test',
-          api_key: 'test-key'
-        }
-      }
-    ];
-    const route = router(modelList, 'random');
-    console.log('✅ Test 9 passed: Router created successfully');
-    passedCount++;
-  } catch (error) {
-    console.log('❌ Test 9 failed:', error.message);
-  }
-
-  // Test 10: Router model selection
-  testCount++;
-  try {
-    const modelList = [
-      {
-        model_name: 'test',
-        llm_params: {
-          model: 'ollama/test',
-          api_key: 'test-key'
-        }
-      }
-    ];
-    const route = router(modelList);
-    const selectedModel = route.selectModel('test');
-    if (selectedModel.llm_params.model === 'ollama/test') {
-      console.log('✅ Test 10 passed: Router model selection works');
-      passedCount++;
-    } else {
-      console.log('❌ Test 10 failed: Incorrect model selected');
-    }
-  } catch (error) {
-    console.log('❌ Test 10 failed:', error.message);
-  }
-
-  // Test 11: Router auto-selection
-  testCount++;
-  try {
-    const modelList = [
-      {
-        model_name: 'test1',
-        llm_params: {
-          model: 'ollama/test1',
-          api_key: 'test-key'
-        }
-      },
-      {
-        model_name: 'test2',
-        llm_params: {
-          model: 'ollama/test2',
-          api_key: 'test-key'
-        }
-      }
-    ];
-    const route = router(modelList, 'random');
-    const selectedModel = route.autoSelectModel();
-    if (selectedModel && selectedModel.llm_params.model.startsWith('ollama/')) {
-      console.log('✅ Test 11 passed: Router auto-selection works');
-      passedCount++;
-    } else {
-      console.log('❌ Test 11 failed: Auto-selection failed');
-    }
-  } catch (error) {
-    console.log('❌ Test 11 failed:', error.message);
-  }
-
-  // Test 12: Guardrails execution
-  testCount++;
-  try {
-    const modelList = [
-      {
-        model_name: 'test',
-        llm_params: {
-          model: 'ollama/test',
-          api_key: 'test-key'
-        }
-      }
-    ];
-    const route = router(modelList);
-
-    route.setGuardrails([
-      {
-        name: 'test_guardrail',
-        mode: 'pre_call',
-        code: (processId, input) => {
-          return { ...input, messages: [...input.messages, { role: 'system', content: 'Test' }] };
-        }
-      }
-    ]);
-
-    console.log('✅ Test 12 passed: Guardrails set successfully');
-    passedCount++;
-  } catch (error) {
-    console.log('❌ Test 12 failed:', error.message);
-  }
-
-  // Test 13: Server app creation
-  testCount++;
-  try {
-    const testApp = app; // Already created
-    if (testApp && typeof testApp.use === 'function') {
-      console.log('✅ Test 13 passed: Server app created successfully');
-      passedCount++;
-    } else {
-      console.log('❌ Test 13 failed: Server app creation failed');
-    }
-  } catch (error) {
-    console.log('❌ Test 13 failed:', error.message);
-  }
-
-  console.log(`\nTest Results: ${passedCount}/${testCount} tests passed`);
-  if (passedCount === testCount) {
-    console.log('🎉 All tests passed!');
-  } else {
-    console.log('⚠️  Some tests failed. Please review the implementation.');
-  }
-}
-
-// Run tests
-runTests().catch(console.error);