from notdiamond.llms.config import LLMConfig
from notdiamond import NotDiamond

client = NotDiamond()

llms = [
    LLMConfig(
        provider="openai",
        model="gpt-3.5-turbo",
        api_key="YOUR_OPENAI_API_KEY",
        temperature=0.5,
        max_tokens=256,
        input_price= 1,  # USD cost per million tokens
        output_price= 0.5,  # USD cost per million tokens
        latency= 0.86,  # Time to first token in seconds
    ),
    LLMConfig(
        provider="anthropic",
        model="claude-3-opus-20240229",
        api_key="YOUR_ANTHROPIC_API_KEY",
        temperature=0.8,
        max_tokens=256,
        input_price= 3,  # USD cost per million tokens
        output_price= 2,  # USD cost per million tokens
        latency= 1.24,  # Time to first token in seconds
    ),
]

result, session_id, provider = client.chat.completions.create(
    messages=[ 
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Concisely explain merge sort."}  # Adjust as desired
    ],
    model=llms,
)

print("Not Diamond session ID: ", session_id)
print("LLM called: ", provider.model)
print("LLM output: ", result.content)

import { NotDiamond } from 'notdiamond';

const client = new NotDiamond({
  apiKey: process.env.NOTDIAMOND_API_KEY,
});

const llms = [
  {
    provider: 'openai',
    model: 'gpt-3.5-turbo',
    contextLength: 4096,
    inputPrice: 1,  // USD cost per million tokens
    outputPrice: 0.5,  // USD cost per million tokens
    latency: 0.86,  // Time to first token in seconds
  },
  {
    provider: 'anthropic',
    model: 'claude-3-opus-20240229',
    contextLength: 100000,
    inputPrice: 3,  // USD cost per million tokens
    outputPrice: 2,  // USD cost per million tokens
    latency: 1.24,  // Time to first token in seconds
  },
];

const messages = [
  { role: 'system', content: 'You are a helpful assistant.' },
  { role: 'user', content: 'Concisely explain merge sort.' },
];

const options = {
  messages,
  llmProviders: llms,
  tradeoff: 'cost',
};

async function main() {
  try {
    const createResult = await client.create(options);

    if ('detail' in createResult) {
      console.error('Error:', createResult.detail);
      return;
    }

    console.log('Not Diamond session ID:', session_id);
    console.log('LLM called:', providers[0].model);
    console.log('LLM output:', createResult.content);
  } catch (error) {
    console.error('An unexpected error occurred:', error);
  }
}

void main();

from notdiamond import NotDiamond

client = NotDiamond(nd_api_url="https://my-api-endpoint.org")

import { NotDiamond } from '../notdiamond';

const client = new NotDiamond({ apiUrl: "https://my-api-endpoint.org" });