llmjs2 1.0.8 → 1.1.0

package/example.js

@@ -0,0 +1,298 @@
+import { completion } from './index.js';
+
+/**
+ * Real usage test program for vs code
+ * This demonstrates actual API calls to Ollama and OpenRouter
+ * Shows all three calling conventions including simple API with model
+ */
+
+async function testSimpleAPI() {
+  console.log('=== Testing Simple API (Auto-Detection) ===\n');
+
+  try {
+    const prompt = 'Explain the use of vs code in 2-3 sentences.';
+    console.log(`Prompt: ${prompt}\n`);
+
+    const result = await completion(prompt);
+
+    console.log('Response:');
+    console.log(result);
+    console.log('\n✓ Simple API test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ Simple API test failed:', error.message);
+    return false;
+  }
+}
+
+async function testSimpleAPIWithModel() {
+  console.log('=== Testing Simple API with Model ===\n');
+
+  try {
+    const model = 'ollama/qwen3.5:397b-cloud';
+    const prompt = 'Explain the use of vs code in 2-3 sentences.';
+    console.log(`Model: ${model}`);
+    console.log(`Prompt: ${prompt}\n`);
+
+    const result = await completion(model, prompt);
+
+    console.log('Response:');
+    console.log(result);
+    console.log('\n✓ Simple API with model test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ Simple API with model test failed:', error.message);
+    return false;
+  }
+}
+
+async function testOllamaFunctionAPI() {
+  console.log('=== Testing Ollama with Function-Based API ===\n');
+
+  try {
+    const prompt = 'Explain the use of vs code in 2-3 sentences.';
+    console.log(`Prompt: ${prompt}\n`);
+
+    const result = await completion('ollama/minimax-m2.5:cloud', prompt);
+
+    console.log('Response:');
+    console.log(result);
+    console.log('\n✓ Ollama function-based API test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ Ollama function-based API test failed:', error.message);
+    return false;
+  }
+}
+
+async function testOllamaObjectAPI() {
+  console.log('=== Testing Ollama with Object-Based API ===\n');
+
+  try {
+    const result = await completion({
+      model: 'ollama/minimax-m2.5:cloud',
+      messages: [
+        { role: 'system', content: 'You are a helpful AI assistant.' },
+        { role: 'user', content: 'Explain the use of vs code in 2-3 sentences.' }
+      ]
+    });
+
+    console.log('Response:');
+    console.log(result);
+    console.log('\n✓ Ollama object-based API test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ Ollama object-based API test failed:', error.message);
+    return false;
+  }
+}
+
+async function testOllamaWithTools() {
+  console.log('=== Testing Ollama with Tools ===\n');
+
+  try {
+    const result = await completion({
+      model: 'ollama/minimax-m2.5:cloud',
+      messages: [
+        { role: 'user', content: 'What is the weather like in Paris?' }
+      ],
+      tools: [
+        {
+          type: 'function',
+          function: {
+            name: 'get_weather',
+            description: 'Get the current weather in a given location',
+            parameters: {
+              type: 'object',
+              properties: {
+                location: {
+                  type: 'string',
+                  description: 'The city and state, e.g. San Francisco, CA'
+                },
+                unit: {
+                  type: 'string',
+                  enum: ['celsius', 'fahrenheit'],
+                  description: 'The temperature unit to use'
+                }
+              },
+              required: ['location']
+            }
+          }
+        }
+      ]
+    });
+
+    console.log('Response:');
+    console.log(JSON.stringify(result, null, 2));
+    console.log('\n✓ Ollama tools test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ Ollama tools test failed:', error.message);
+    return false;
+  }
+}
+
+async function testOpenRouterFunctionAPI() {
+  console.log('=== Testing OpenRouter with Function-Based API ===\n');
+
+  try {
+    const prompt = 'What is the capital of France? Answer in one sentence.';
+    console.log(`Prompt: ${prompt}\n`);
+
+    const result = await completion('openrouter/openrouter/free', prompt);
+
+    console.log('Response:');
+    console.log(result);
+    console.log('\n✓ OpenRouter function-based API test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ OpenRouter function-based API test failed:', error.message);
+    return false;
+  }
+}
+
+async function testOpenRouterObjectAPI() {
+  console.log('=== Testing OpenRouter with Object-Based API ===\n');
+
+  try {
+    const result = await completion({
+      model: 'openrouter/openrouter/free',
+      messages: [
+        { role: 'system', content: 'You are a helpful AI assistant.' },
+        { role: 'user', content: 'What is the capital of France? Answer in one sentence.' }
+      ]
+    });
+
+    console.log('Response:');
+    console.log(result);
+    console.log('\n✓ OpenRouter object-based API test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ OpenRouter object-based API test failed:', error.message);
+    return false;
+  }
+}
+
+async function testOpenRouterWithTools() {
+  console.log('=== Testing OpenRouter with Tools ===\n');
+
+  try {
+    const result = await completion({
+      model: 'openrouter/openrouter/free',
+      messages: [
+        { role: 'user', content: 'What is the weather like in Paris?' }
+      ],
+      tools: [
+        {
+          type: 'function',
+          function: {
+            name: 'get_weather',
+            description: 'Get the current weather in a given location',
+            parameters: {
+              type: 'object',
+              properties: {
+                location: {
+                  type: 'string',
+                  description: 'The city and state, e.g. San Francisco, CA'
+                },
+                unit: {
+                  type: 'string',
+                  enum: ['celsius', 'fahrenheit'],
+                  description: 'The temperature unit to use'
+                }
+              },
+              required: ['location']
+            }
+          }
+        }
+      ]
+    });
+
+    console.log('Response:');
+    console.log(JSON.stringify(result, null, 2));
+    console.log('\n✓ OpenRouter tools test completed successfully\n');
+    return true;
+  } catch (error) {
+    console.error('✗ OpenRouter tools test failed:', error.message);
+    return false;
+  }
+}
+
+async function main() {
+  console.log('vs code Real Usage Test Program');
+  console.log('==============================\n');
+
+  // Check for API keys
+  const hasOllamaKey = !!process.env.OLLAMA_API_KEY;
+  const hasOpenRouterKey = !!process.env.OPEN_ROUTER_API_KEY;
+
+  console.log('API Key Status:');
+  console.log(`  Ollama: ${hasOllamaKey ? '✓ Set' : '✗ Not set'}`);
+  console.log(`  OpenRouter: ${hasOpenRouterKey ? '✓ Set' : '✗ Not set'}\n`);
+
+  if (!hasOllamaKey && !hasOpenRouterKey) {
+    console.log('⚠ No API keys found. Please set environment variables:');
+    console.log('  export OLLAMA_API_KEY=your-ollama-api-key');
+    console.log('  export OPEN_ROUTER_API_KEY=your-openrouter-api-key\n');
+    console.log('Or modify this script to pass API keys directly.\n');
+  }
+
+  const results = {
+    simpleAPI: false,
+    simpleAPIWithModel: false,
+    ollamaFunction: false,
+    ollamaObject: false,
+    ollamaTools: false,
+    openrouterFunction: false,
+    openrouterObject: false,
+    openrouterTools: false
+  };
+
+  // Test simple API if any key is available
+  if (hasOllamaKey || hasOpenRouterKey) {
+    results.simpleAPI = await testSimpleAPI();
+    results.simpleAPIWithModel = await testSimpleAPIWithModel();
+  } else {
+    console.log('=== Skipping Simple API tests (no API key) ===\n');
+  }
+
+  // Test Ollama if key is available
+  if (hasOllamaKey) {
+    results.ollamaFunction = await testOllamaFunctionAPI();
+    results.ollamaObject = await testOllamaObjectAPI();
+    results.ollamaTools = await testOllamaWithTools();
+  } else {
+    console.log('=== Skipping Ollama tests (no API key) ===\n');
+  }
+
+  // Test OpenRouter if key is available
+  if (hasOpenRouterKey) {
+    results.openrouterFunction = await testOpenRouterFunctionAPI();
+    results.openrouterObject = await testOpenRouterObjectAPI();
+    results.openrouterTools = await testOpenRouterWithTools();
+  } else {
+    console.log('=== Skipping OpenRouter tests (no API key) ===\n');
+  }
+
+  // Summary
+  console.log('==============================');
+  console.log('Test Summary:');
+  console.log(`  Simple API: ${results.simpleAPI ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  Simple API with Model: ${results.simpleAPIWithModel ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  Ollama Function API: ${results.ollamaFunction ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  Ollama Object API: ${results.ollamaObject ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  Ollama Tools: ${results.ollamaTools ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  OpenRouter Function API: ${results.openrouterFunction ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  OpenRouter Object API: ${results.openrouterObject ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log(`  OpenRouter Tools: ${results.openrouterTools ? '✓ Passed' : '✗ Failed/Skipped'}`);
+  console.log('==============================\n');
+
+  if (results.simpleAPI || results.simpleAPIWithModel || results.ollamaFunction || results.ollamaObject || results.ollamaTools || 
+      results.openrouterFunction || results.openrouterObject || results.openrouterTools) {
+    console.log('✓ At least one test passed!');
+  } else {
+    console.log('✗ No tests passed. Check your API keys and try again.');
+  }
+}
+
+main().catch(console.error);