Skip to content

Commit 6524b5c

Browse files
committed
Use Claude 3 Haiku model if LLMCLI_MODEL=haiku env is set
(Experimental, no promises this stays)
1 parent 1ba58d7 commit 6524b5c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

main.go

+3
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,9 @@ func run(ctx context.Context, args runArgs) error {
121121
const oldClaudeModelId = "anthropic.claude-3-sonnet-20240229-v1:0"
122122
const claudeModelId = "anthropic.claude-3-5-sonnet-20240620-v1:0"
123123
var modelId = claudeModelId
124+
if os.Getenv("LLMCLI_MODEL") == "haiku" {
125+
modelId = "anthropic.claude-3-haiku-20240307-v1:0"
126+
}
124127
for i := range contentBlocks {
125128
if doc, ok := contentBlocks[i].(*types.ContentBlockMemberDocument); ok {
126129
// https://docs.aws.amazon.com/bedrock/latest/userguide/conversation-inference.html#conversation-inference-supported-models-features

0 commit comments

Comments
 (0)