dev.langchain4j.model.bedrock.internal

Class AbstractBedrockChatModel<T extends BedrockChatModelResponse>

java.lang.Object

dev.langchain4j.model.bedrock.internal.AbstractBedrockChatModel<T>

All Implemented Interfaces:

dev.langchain4j.model.chat.ChatLanguageModel

Direct Known Subclasses:

BedrockAI21LabsChatModel, BedrockAnthropicChatModel, BedrockCohereChatModel, BedrockStabilityAIChatModel, BedrockTitanChatModel

public abstract class AbstractBedrockChatModel<T extends BedrockChatModelResponse>
extends Object
implements dev.langchain4j.model.chat.ChatLanguageModel

Bedrock chat model

Constructor Summary

Constructors

Constructor and Description
AbstractBedrockChatModel()

Method Summary

Modifier and Type	Method and Description
protected String	chatMessageToString(dev.langchain4j.data.message.ChatMessage message) Convert chat message to string
dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage>	generate(List<dev.langchain4j.data.message.ChatMessage> messages)
protected abstract String	getModelId() Get model id
protected abstract Map<String,Object>	getRequestParameters(String prompt) Get request parameters
protected abstract Class<T>	getResponseClassType() Get response class type
protected software.amazon.awssdk.services.bedrockruntime.model.InvokeModelResponse	invoke(String body) Invoke call to the API
protected static Map<String,Object>	of(String key, Object value) Create map with single entry

Methods inherited from class java.lang.Object

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

Methods inherited from interface dev.langchain4j.model.chat.ChatLanguageModel

generate, generate, generate, generate

Constructor Detail

AbstractBedrockChatModel

public AbstractBedrockChatModel()

Method Detail

generate

public dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage> generate(List<dev.langchain4j.data.message.ChatMessage> messages)

Specified by:

generate in interface dev.langchain4j.model.chat.ChatLanguageModel

chatMessageToString

protected String chatMessageToString(dev.langchain4j.data.message.ChatMessage message)

Convert chat message to string

Parameters:

message - chat message

Returns:

string

getRequestParameters

protected abstract Map<String,Object> getRequestParameters(String prompt)

Get request parameters

Parameters:

prompt - prompt

Returns:

request body

getModelId

protected abstract String getModelId()

Get model id

Returns:

model id

getResponseClassType

protected abstract Class<T> getResponseClassType()

Get response class type

Returns:

response class type

invoke

protected software.amazon.awssdk.services.bedrockruntime.model.InvokeModelResponse invoke(String body)

Invoke call to the API

Parameters:

body - body

Returns:

invoke model response

of

protected static Map<String,Object> of(String key,
                                       Object value)

Create map with single entry

Parameters:

key - key

value - value

Returns:

map