Amazon Bedrock

pip install 'niadra[bedrock]'   # boto3 1.40 ou mais novo, abaixo da 2

npm install @niadra/sdk @aws-sdk/client-bedrock-runtime   # @niadra/sdk/bedrock

"""Amazon Bedrock's Converse API with the customer's context."""

import boto3

from niadra import Niadra, phone
from niadra.integrations.bedrock import wrap

niadra = Niadra(channel="chat")
bedrock = wrap(boto3.client("bedrock-runtime"))

with niadra.conversation("thread-81", subject=phone("+5511912345678")) as conversation:
    conversation.customer("Where is my replacement lid?")
    response = bedrock.converse(
        modelId="us.anthropic.claude-sonnet-4-5-20250929-v1:0",
        system=[{"text": "You are Acme's agent."}],
        messages=[{"role": "user", "content": [{"text": "Where is my replacement lid?"}]}],
    )
    print(response["output"]["message"]["content"][0]["text"])

import { BedrockRuntimeClient, ConverseCommand } from "@aws-sdk/client-bedrock-runtime";
import { Niadra, handles } from "@niadra/sdk";
import { wrapBedrock } from "@niadra/sdk/bedrock";

const niadra = new Niadra();
const client = new BedrockRuntimeClient({});

/** One customer message in; `userId` comes from your session, never from the model. */
export async function reply(userId: string, chatId: string, text: string): Promise<string> {
  const convo = niadra.conversation({ subject: handles.appUserId(userId), channel: "web_chat", conversation_id: chatId });
  const bedrock = wrapBedrock(client, convo);
  const response = await bedrock.send(
    new ConverseCommand({
      modelId: "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
      system: [{ text: "You are Acme's support agent. Be brief." }],
      messages: [{ role: "user", content: [{ text }] }],
    }),
  );
  return response.output?.message?.content?.map((block) => block.text ?? "").join("") ?? "";
}