1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17import { BedrockRuntimeClient, ConverseCommand } from "@aws-sdk/client-bedrock-runtime";
const DEFAULT_MODEL_ID = "amazon.nova-lite-v1:0";
const bedrockClient = new BedrockRuntimeClient({ region: process.env.AWS_REGION || "us-east-1" });
export const bedrock = {
invoke: async (prompt: string, modelId?: string): Promise<string> => {
const command = new ConverseCommand({
modelId: modelId || DEFAULT_MODEL_ID,
messages: [{ role: "user", content: [{ text: prompt }] }],
inferenceConfig: { maxTokens: 1024 },
});
const response = await bedrockClient.send(command);
return response.output?.message?.content?.[0]?.text ?? '';
},
};