BedrockClient

org.llm4s.llmconnect.provider.BedrockClient
See theBedrockClient companion object
class BedrockClient(config: BedrockConfig, val metrics: MetricsCollector, exchangeLogging: ProviderExchangeLogging)(using val registryService: ModelRegistryService) extends BaseLifecycleLLMClient

org.llm4s.llmconnect.LLMClient for the AWS Bedrock Converse and ConverseStream APIs.

Reaches the models Bedrock hosts - Anthropic Claude, Meta Llama, Mistral, Amazon Nova and Titan, and others - through Bedrock's one Converse interface.

== Authentication ==

Explicit org.llm4s.llmconnect.config.BedrockCredentials when configured (with a session token for temporary credentials); otherwise the config's profile; otherwise the AWS default credential chain (environment variables, ~/.aws/credentials, EC2 instance profile, ECS task role, ...).

== Streaming ==

streamComplete uses ConverseStream, which the AWS SDK offers on its asynchronous client. Its callbacks run on SDK threads; they only enqueue events, and the calling thread drains the queue, so onChunk runs on the caller's thread, in order, and interrupting the caller cancels the request and returns Left(CancelledError).

== Errors ==

ThrottlingException and ServiceQuotaExceededException become RateLimitError, ValidationException ValidationError, AccessDeniedException (and any 401/403) AuthenticationError, other service errors ServiceError, and a failure to reach Bedrock at all NetworkError.

Value parameters

config

region, model, credentials and endpoint.

exchangeLogging

optional provider exchange logging.

metrics

receives per-call latency and token-usage events.

Attributes

Companion
object
Graph
Supertypes
trait LLMClient
trait AutoCloseable
class Object
trait Matchable
class Any
Show all

Members list

Value members

Concrete methods

override def complete(conversation: Conversation, options: CompletionOptions): Result[Completion]

Executes a blocking completion request and returns the full response.

Executes a blocking completion request and returns the full response.

Sends the conversation to the LLM and waits for the complete response. Use when you need the entire response at once or when streaming is not required.

Value parameters

conversation

conversation history including system, user, assistant, and tool messages

options

configuration including temperature, max tokens, tools, etc. (default: CompletionOptions())

Attributes

Returns

Right(Completion) with the model's response, or Left(LLMError) on failure

Definition Classes
override def getContextWindow(): Int

Returns the maximum context window size supported by this model in tokens.

Returns the maximum context window size supported by this model in tokens.

The context window is the total tokens (prompt + completion) the model can process in a single request, including all conversation messages and the generated response.

Attributes

Returns

total context window size in tokens (e.g., 4096, 8192, 128000)

Definition Classes
override def getReserveCompletion(): Int

Returns the number of tokens reserved for the model's completion response.

Returns the number of tokens reserved for the model's completion response.

This value is subtracted from the context window when calculating available tokens for prompts. Corresponds to the max_tokens or completion token limit configured for the model.

Attributes

Returns

number of tokens reserved for completion

Definition Classes
override def streamComplete(conversation: Conversation, options: CompletionOptions, onChunk: StreamedChunk => Unit): Result[Completion]

Executes a streaming completion request, invoking a callback for each chunk as it arrives.

Executes a streaming completion request, invoking a callback for each chunk as it arrives.

Streams the response incrementally, calling onChunk for each token/chunk received. Enables real-time display of responses. Returns the final accumulated completion on success.

Value parameters

conversation

conversation history including system, user, assistant, and tool messages

onChunk

callback invoked for each chunk; called synchronously, avoid blocking operations

options

configuration including temperature, max tokens, tools, etc. (default: CompletionOptions())

Attributes

Returns

Right(Completion) with the complete accumulated response, or Left(LLMError) on failure

Definition Classes

Inherited methods

override def close(): Unit

Releases resources and closes connections to the LLM provider.

Releases resources and closes connections to the LLM provider.

Call when the client is no longer needed. After calling close(), the client should not be used. Default implementation is a no-op; override if managing resources like connections or thread pools.

Attributes

Definition Classes
BaseLifecycleLLMClient -> LLMClient -> AutoCloseable
Inherited from:
BaseLifecycleLLMClient
def completeStructured[A](conversation: Conversation, schema: ObjectSchema[A], options: CompletionOptions)(implicit reader: Reader[A]): Result[A]

Sends the conversation and parses the response into a typed value using the provided schema.

Sends the conversation and parses the response into a typed value using the provided schema.

Sets ResponseFormat.JsonSchema on the options so providers that support native structured output (OpenAI, Gemini) enforce the schema at generation time. Anthropic falls back to a best-effort system-prompt instruction, which is not schema-enforced. Because models may wrap JSON in markdown code fences or surround it with prose, the response is normalised (fence stripped, first balanced {...} or [...] extracted) before being deserialised with uPickle into the expected type A.

Type parameters

A

target type; must have a corresponding upickle.default.Reader[A]

Value parameters

conversation

conversation history

options

additional completion options (default: CompletionOptions())

reader

implicit uPickle reader for deserialising the JSON into A

schema

JSON-Schema description of the expected response object

Attributes

Returns

Right(A) on success, or Left(LLMError) when the provider call fails or the JSON cannot be parsed

Inherited from:
LLMClient
protected def completeWithMetrics(operation: => Result[Completion]): Result[Completion]

Validates that the client is open, executes the operation, and records standard completion metrics (latency, token usage, estimated cost).

Validates that the client is open, executes the operation, and records standard completion metrics (latency, token usage, estimated cost).

Use this in complete and streamComplete implementations to avoid repeating the lifecycle-check + metrics-wrapping boilerplate.

An interrupted call - one that throws InterruptedException, or fails while the thread is interrupted - is returned as Left(CancelledError) with the interrupt flag kept, whatever the provider SDK did with it. A call made with the flag already set returns Left(CancelledError) at once, without running operation: an SDK that ignores the flag would otherwise send the (billed) request anyway.

Value parameters

operation

The provider-specific completion logic to execute. Called only when the client is open and the thread is not interrupted.

Attributes

Returns

The completion result with metrics recorded as a side-effect.

Inherited from:
BaseLifecycleLLMClient

Calculates available token budget for prompts after accounting for completion reserve and headroom.

Calculates available token budget for prompts after accounting for completion reserve and headroom.

Formula: (contextWindow - reserveCompletion) * (1 - headroom)

Headroom provides a safety margin for tokenization variations and message formatting overhead.

Value parameters

headroom

safety margin as percentage of prompt budget (default: HeadroomPercent.Standard ~10%)

Attributes

Returns

maximum tokens available for prompt content

Inherited from:
LLMClient
def validate(): Result[Unit]

Validates client configuration and connectivity to the LLM provider.

Validates client configuration and connectivity to the LLM provider.

May perform checks such as verifying API credentials, testing connectivity, and validating configuration. Default implementation returns success; override for provider-specific validation.

Attributes

Returns

Right(()) if validation succeeds, Left(LLMError) with details on failure

Inherited from:
LLMClient
protected def validateNotClosed: Result[Unit]

Attributes

Inherited from:
BaseLifecycleLLMClient
protected def withMetrics[A](provider: String, model: String, operation: => Result[A], extractUsage: A => Option[TokenUsage], extractCost: A => Option[Double]): Result[A]

Executes operation and records metrics for the call.

Executes operation and records metrics for the call.

Latency and outcome (success or classified error) are recorded for every call regardless of result. Token counts and cost are recorded only on success — a Left result emits an org.llm4s.metrics.Outcome.Error event whose kind is derived from the org.llm4s.error.LLMError subtype via ErrorKind.fromLLMError.

Value parameters

extractCost

Extracts the pre-computed cost (USD) from a successful result; return None to skip cost recording.

extractUsage

Extracts prompt/completion token counts from a successful result; return None to skip token recording.

model

Model identifier forwarded to the collector.

operation

The LLM call to time and observe.

provider

Provider label forwarded to the collector (e.g. "openai").

Attributes

Returns

The result of operation, unchanged.

Inherited from:
MetricsRecording

Givens