UNPKG

notdiamond

Version:
143 lines (108 loc) 4.23 kB
# NotDiamond Node API Library This library provides convenient access to the NotDiamond model routing API from TypeScript or JavaScript. NotDiamond helps you select the best AI model for your specific use case, optimizing for factors like cost, latency, or performance. ## Installation ```sh npm install notdiamond ``` For Deno users: ```ts import { NotDiamond } from 'https://deno.land/x/notdiamond@v1.0.0/mod.ts'; ``` ## Quick Start To use the API, you need to sign up for a NotDiamond account and obtain an API key. [Sign up here](https://app.notdiamond.ai). ### Basic Usage Here's a simple example of how to use NotDiamond to select the best model between GPT-4o, Claude 3.5 Sonnet, and Gemini 1.5 Pro, while optimizing for latency and outputting the raw text: ```typescript import { NotDiamond } from 'notdiamond'; const notDiamond = new NotDiamond({ // Optional - automatically loads from environment variable apiKey: process.env.NOTDIAMOND_API_KEY, }); async function basicExample() { // 1. Select the best model const result = await notDiamond.create({ // Define the user's message messages: [{ content: 'What is 12x12?', role: 'user' }], // Specify the LLM providers and models to choose from llmProviders: [ { provider: 'openai', model: 'gpt-4o-2024-05-13' }, { provider: 'anthropic', model: 'claude-3-5-sonnet-20240620' }, { provider: 'google', model: 'gemini-1.5-pro-latest' }, ], // Set the optimization criteria to latency tradeoff: 'latency', }); // 2. Handle potential errors if ('detail' in result) { console.error('Error:', result.detail); return; } // 3. Log the results // Display the text response console.log('LLM output:', result.content); // Display the selected provider(s) console.log('Selected providers:', result.providers); // Show the unique session ID for this request console.log('Session ID:', result.session_id); } basicExample(); ``` ### Advanced Usage with Tools You can also use NotDiamond with custom tools: ```typescript import { NotDiamond, Tool } from 'notdiamond'; const notDiamond = new NotDiamond(); // Define custom tools for the AI to use const tools: Tool[] = [ { type: 'function', function: { name: 'multiply', description: 'Multiplies two numbers.', parameters: { type: 'object', properties: { a: { type: 'integer' }, b: { type: 'integer' }, }, required: ['a', 'b'], }, }, }, ]; async function toolCallingExample() { const result = await notDiamond.modelSelect({ messages: [{ content: 'What is 12x12?', role: 'user' }], llmProviders: [ { provider: 'openai', model: 'gpt-4-1106-preview' }, { provider: 'anthropic', model: 'claude-3-sonnet-20240229' }, ], // Include custom tools in the request tools: tools, // Optimize for cost instead of latency tradeoff: 'cost', // Allow selection of up to 2 models maxModelDepth: 2, }); if ('detail' in result) { console.error('Error:', result.detail); return; } console.log('Selected providers:', result.providers); console.log('Session ID:', result.session_id); } toolCallingExample(); ``` ## API Reference For a complete API reference and more detailed documentation, please check our [API Documentation](https://notdiamond.readme.io/docs/what-is-not-diamond#getting-started). ## Key Concepts - **llmProviders**: An array of AI providers and models you want NotDiamond to choose from. - **tradeoff**: The factor to optimize for (e.g., 'latency', 'cost', 'performance'). - **tools**: Custom functions that the AI can use to perform specific tasks. - **maxModelDepth**: The maximum number of models to include in the response (for advanced use cases). ## Error Handling NotDiamond uses typed responses. If there's an error, the response will have a `detail` property with the error message. Always check for this property when handling responses. ## Support If you encounter any issues or have questions, please [open an issue](https://github.com/Not-Diamond/notdiamond-node) on our GitHub repository. ## License This library is released under the MIT License.