AWS Bedrock

pip install 'lago-agent-sdk[bedrock]'

npm install lago-agent-sdk @aws-sdk/client-bedrock-runtime

import boto3
from lago_agent_sdk import LagoSDK

sdk = LagoSDK(
    api_key="<YOUR_LAGO_API_KEY>",
    default_subscription_id="sub_acme",
)
client = sdk.wrap(boto3.client("bedrock-runtime", region_name="eu-west-1"))

resp = client.converse(
    modelId="eu.amazon.nova-lite-v1:0",
    messages=[{"role": "user", "content": [{"text": "Hello"}]}],
)
sdk.flush()

import { BedrockRuntimeClient, ConverseCommand } from "@aws-sdk/client-bedrock-runtime";
import { LagoSDK } from "lago-agent-sdk";

const sdk = new LagoSDK({
  apiKey: process.env.LAGO_API_KEY!,
  defaultSubscriptionId: "sub_acme",
});
const client = sdk.wrap(new BedrockRuntimeClient({ region: "eu-west-1" }));

await client.send(new ConverseCommand({
  modelId: "eu.amazon.nova-lite-v1:0",
  messages: [{ role: "user", content: [{ text: "Hello" }] }],
}));
await sdk.flush();

import json

resp = client.invoke_model(
    modelId="eu.anthropic.claude-haiku-4-5-v1:0",
    body=json.dumps({
        "anthropic_version": "bedrock-2023-05-31",
        "max_tokens": 200,
        "messages": [{"role": "user", "content": "Hello"}],
    }),
)
payload = json.loads(resp["body"].read())   # works unchanged

import { InvokeModelCommand } from "@aws-sdk/client-bedrock-runtime";

const out = await client.send(new InvokeModelCommand({
  modelId: "eu.anthropic.claude-haiku-4-5-v1:0",
  body: JSON.stringify({
    anthropic_version: "bedrock-2023-05-31",
    max_tokens: 200,
    messages: [{ role: "user", content: "Hello" }],
  }),
}));
const payload = JSON.parse(new TextDecoder().decode(out.body));

resp = client.converse_stream(
    modelId="eu.amazon.nova-lite-v1:0",
    messages=[{"role": "user", "content": [{"text": "Hello"}]}],
)
for event in resp["stream"]:
    ...

import { ConverseStreamCommand } from "@aws-sdk/client-bedrock-runtime";

const out = await client.send(new ConverseStreamCommand({
  modelId: "eu.amazon.nova-lite-v1:0",
  messages: [{ role: "user", content: [{ text: "Hello" }] }],
}));
for await (const event of out.stream!) {
  // ...
}

client.converse(
    modelId="eu.amazon.nova-lite-v1:0",
    messages=[{"role": "user", "content": [{"text": "Hello"}]}],
    extra_lago={
        "subscription": "sub_acme",
        "dimensions": {"feature": "summarize"},
        "mode": "price",     # optional
        "markup": 1.5,       # optional
    },
)

const cmd = new ConverseCommand({
  modelId: "eu.amazon.nova-lite-v1:0",
  messages: [{ role: "user", content: [{ text: "Hello" }] }],
});
(cmd as any).__lago = {
  subscription: "sub_acme",
  dimensions: { feature: "summarize" },
  mode: "price",   // optional
  markup: 1.5,     // optional
};
await client.send(cmd);