AWS Bedrock (Beta)

npm install @raindrop-ai/bedrock @aws-sdk/client-bedrock-runtime

pnpm add @raindrop-ai/bedrock @aws-sdk/client-bedrock-runtime

pip install raindrop-bedrock boto3

import { createRaindropBedrock } from "@raindrop-ai/bedrock";
import { BedrockRuntimeClient, ConverseCommand } from "@aws-sdk/client-bedrock-runtime";

const raindrop = createRaindropBedrock({
  writeKey: "your-write-key",
  userId: "user-123",
});

const client = new BedrockRuntimeClient({ region: "us-east-1" });
const wrapped = raindrop.wrap(client);

const response = await wrapped.send(
  new ConverseCommand({
    modelId: "anthropic.claude-3-5-sonnet-20241022-v2:0",
    messages: [
      { role: "user", content: [{ text: "What is the capital of France?" }] },
    ],
  }),
);

console.log(response.output?.message?.content?.[0]?.text);
await raindrop.flush();

import boto3
from raindrop_bedrock import RaindropBedrock

raindrop = RaindropBedrock(
    api_key="your-write-key",
    user_id="user-123",
)

client = boto3.client("bedrock-runtime", region_name="us-east-1")
raindrop.wrap(client)

response = client.converse(
    modelId="anthropic.claude-3-5-sonnet-20241022-v2:0",
    messages=[
        {"role": "user", "content": [{"text": "What is the capital of France?"}]},
    ],
)

print(response["output"]["message"]["content"][0]["text"])
raindrop.flush()

const raindrop = createRaindropBedrock({
  writeKey: "your-write-key",       // Optional: your Raindrop write key (omit to disable telemetry)
  endpoint: "...",          // Optional: custom API endpoint
  userId: "user-123",      // Optional: associate events with a user
  convoId: "convo-456",    // Optional: conversation/thread ID
  projectId: "support-prod", // Optional: route events to a specific project (slug)
  debug: false,             // Optional: enable verbose logging
});

raindrop = RaindropBedrock(
    api_key="your-write-key",           # Optional: your Raindrop API key (omit to disable telemetry)
    user_id="user-123",        # Optional: associate events with a user
    convo_id="convo-456",      # Optional: conversation/thread ID
    project_id="support-prod", # Optional: route events to a specific project (slug)
    tracing_enabled=True,      # Optional: enable Raindrop tracing
    bypass_otel_for_tools=True,# Optional: bypass OpenTelemetry for tools
    debug=False,               # Optional: enable verbose logging
)

const raindrop = createRaindropBedrock({
  writeKey: "your-write-key",
  projectId: "support-prod",
});

raindrop = RaindropBedrock(
    api_key="your-write-key",
    project_id="support-prod",
)

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

const response = await wrapped.send(
  new InvokeModelCommand({
    modelId: "anthropic.claude-3-5-sonnet-20241022-v2:0",
    contentType: "application/json",
    body: JSON.stringify({
      anthropic_version: "bedrock-2023-05-31",
      messages: [{ role: "user", content: "Hello!" }],
      max_tokens: 256,
    }),
  }),
);

const result = JSON.parse(new TextDecoder().decode(response.body));
console.log(result.content[0].text);
await raindrop.flush();

import json

response = client.invoke_model(
    modelId="anthropic.claude-3-5-sonnet-20241022-v2:0",
    contentType="application/json",
    body=json.dumps({
        "anthropic_version": "bedrock-2023-05-31",
        "messages": [{"role": "user", "content": "Hello!"}],
        "max_tokens": 256,
    }),
)

result = json.loads(response["body"].read())
print(result["content"][0]["text"])
raindrop.flush()

raindrop.users.identify({ userId: "user-123", traits: { plan: "pro", company: "Acme" } });

raindrop.identify(user_id="user-123", traits={"plan": "pro", "company": "Acme"})

raindrop.signals.track({
  eventId: "evt_abc123",
  name: "thumbs_up",
  type: "feedback",
  sentiment: "POSITIVE",
  comment: "Great answer!",
});

raindrop.track_signal(
    event_id="evt_abc123",
    name="thumbs_up",
    signal_type="feedback",
    sentiment="POSITIVE",
    comment="Great answer!",
)

await raindrop.flush();     // flush pending data
await raindrop.shutdown();  // flush + release resources

raindrop.flush()     # flush pending data
raindrop.shutdown()  # flush + release resources