Package com.tnsai.llm.providers

Class BedrockClient

java.lang.Object

com.tnsai.llm.providers.BedrockClient

All Implemented Interfaces:

com.tnsai.llm.LLMClient

public class BedrockClient
extends Object
implements com.tnsai.llm.LLMClient

AWS Bedrock LLM client implementation. Supports Anthropic Claude 3 models via Amazon Bedrock.

Configuration:

• AWS Credentials (via environment variables, profile, or IAM role)
• AWS_REGION - Default: us-east-1

Supported Models (tested):

• anthropic.claude-3-sonnet-20240229-v1:0
• anthropic.claude-3-haiku-20240307-v1:0
• anthropic.claude-3-opus-20240229-v1:0

Constructor Summary

Constructors

Constructor	Description
BedrockClient(String modelId)	Creates a BedrockClient with the specified model ID.
BedrockClient(String modelId, float temperature)
BedrockClient(String modelId, float temperature, Float topP, Integer maxTokens, String region)
BedrockClient(String modelId, float temperature, Integer maxTokens, String region)

Method Summary

Modifier and Type	Method	Description
com.tnsai.llm.ChatResponse	chat(String message, Optional<String> systemPrompt, Optional<List<Map<String,Object>>> history, Optional<List<Map<String,Object>>> tools)
Optional<Integer>	getMaxTokens()
String	getModel()
float	getTemperature()
Optional<Float>	getTopP()
Stream<String>	streamChat(String message, Optional<String> systemPrompt, Optional<List<Map<String,Object>>> history, Optional<List<Map<String,Object>>> tools)

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface com.tnsai.llm.LLMClient

chat, chat, chat, chat, chat, getCapabilities, getProviderName, listModels, streamChat, streamChat, streamChatWithHandler, streamChatWithHandler, streamChatWithSpec, streamChatWithSpec

Constructor Details

BedrockClient

public BedrockClient(String modelId)

Creates a BedrockClient with the specified model ID. Uses default region (us-east-1) and standard AWS credential chain.

Parameters:

modelId - The Bedrock model ID (e.g., "anthropic.claude-3-sonnet-20240229-v1:0")

BedrockClient

public BedrockClient(String modelId,
 float temperature)

BedrockClient

public BedrockClient(String modelId,
 float temperature,
 Integer maxTokens,
 String region)

BedrockClient

public BedrockClient(String modelId,
 float temperature,
 Float topP,
 Integer maxTokens,
 String region)

Method Details

getModel

public String getModel()

Specified by:

getModel in interface com.tnsai.llm.LLMClient

getTemperature

public float getTemperature()

Specified by:

getTemperature in interface com.tnsai.llm.LLMClient

getTopP

public Optional<Float> getTopP()

Specified by:

getTopP in interface com.tnsai.llm.LLMClient

getMaxTokens

public Optional<Integer> getMaxTokens()

Specified by:

getMaxTokens in interface com.tnsai.llm.LLMClient

chat

public com.tnsai.llm.ChatResponse chat(String message,
 Optional<String> systemPrompt,
 Optional<List<Map<String,Object>>> history,
 Optional<List<Map<String,Object>>> tools)

Specified by:

chat in interface com.tnsai.llm.LLMClient

streamChat

public Stream<String> streamChat(String message,
 Optional<String> systemPrompt,
 Optional<List<Map<String,Object>>> history,
 Optional<List<Map<String,Object>>> tools)

Specified by:

streamChat in interface com.tnsai.llm.LLMClient