使用统计

import requests
import json

url = "https://www.openroute.cn/api/v1/chat/completions"
headers = {
    "Authorization": f"Bearer <OPENROUTE_API_KEY>",
    "Content-Type": "application/json"
}
payload = {
    "model": "anthropic/claude-3-opus",
    "messages": [
        {"role": "user", "content": "法国的首都是什么？"}
    ],
    "usage": {
        "include": True
    }
}

response = requests.post(url, headers=headers, data=json.dumps(payload))
print("响应:", response.json()['choices'][0]['message']['content'])
print("使用统计:", response.json()['usage'])

import OpenAI from 'openai';

const openai = new OpenAI({
  baseURL: 'https://www.openroute.cn/api/v1',
  apiKey: '<OPENROUTE_API_KEY>',
});

async function getResponseWithUsage() {
  const response = await openai.chat.completions.create({
    model: 'anthropic/claude-3-opus',
    messages: [
      {
        role: 'user',
        content: '法国的首都是什么？',
      },
    ],
    extra_body: {
      usage: {
        include: true,
      },
    },
  });

  console.log('响应:', response.choices[0].message.content);
  console.log('使用统计:', response.usage);
}

getResponseWithUsage();

from openai import OpenAI

client = OpenAI(
    base_url="https://www.openroute.cn/api/v1",
    api_key="<OPENROUTE_API_KEY>",
)

def chat_completion_with_usage(messages):
    response = client.chat.completions.create(
        model="anthropic/claude-3-opus",
        messages=messages,
        usage={
          "include": True
        },
        stream=True
    )
    return response

for chunk in chat_completion_with_usage([
    {"role": "user", "content": "写一首关于巴黎的俳句。"}
]):
    if hasattr(chunk, 'usage'):
        if hasattr(chunk.usage, 'total_tokens'):
            print(f"\n使用统计:")
            print(f"总 Token: {chunk.usage.total_tokens}")
            print(f"提示 Token: {chunk.usage.prompt_tokens}")
            print(f"完成 Token: {chunk.usage.completion_tokens}")
            print(f"成本: {chunk.usage.cost} 积分")
    elif chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="")

import OpenAI from 'openai';

const openai = new OpenAI({
  baseURL: 'https://www.openroute.cn/api/v1',
  apiKey: '<OPENROUTE_API_KEY>',
});

async function chatCompletionWithUsage(messages) {
  const response = await openai.chat.completions.create({
    model: 'anthropic/claude-3-opus',
    messages,
    usage: {
      include: true,
    },
    stream: true,
  });

  return response;
}

(async () => {
  for await (const chunk of chatCompletionWithUsage([
    { role: 'user', content: '写一首关于巴黎的俳句。' },
  ])) {
    if (chunk.usage) {
      console.log('\n使用统计:');
      console.log(`总 Token: ${chunk.usage.total_tokens}`);
      console.log(`提示 Token: ${chunk.usage.prompt_tokens}`);
      console.log(`完成 Token: ${chunk.usage.completion_tokens}`);
      console.log(`成本: ${chunk.usage.cost} 积分`);
    } else if (chunk.choices[0].delta.content) {
      process.stdout.write(chunk.choices[0].delta.content);
    }
  }
})();