Support AWS_SESSION_TOKEN in BedrockClient

[tiulpin]

• includes changes from zarechneva/enable-bedrock-tests
• fixes integration tests and provides proper auth settings for bedrock integration

[Copilot]

Pull Request Overview

This PR adds support for AWS session tokens in the BedrockClient to enable the use of temporary AWS credentials. The changes primarily focus on enabling Bedrock integration tests that were previously disabled.

• Adds awsSessionToken parameter to BedrockLLMClient constructors and related factory functions
• Updates integration tests to use proper AWS environment variables and enables previously disabled Bedrock tests
• Adds a smoke test to verify Bedrock credentials before running full integration tests

Reviewed Changes

Copilot reviewed 9 out of 10 changed files in this pull request and generated 1 comment.

Show a summary per file

File	Description
SimplePromptExecutors.jvm.kt	Adds awsSessionToken parameter to simpleBedrockExecutor function
BedrockLLMClient.kt	Updates constructor to accept and use awsSessionToken for authentication
TestUtils.kt	Fixes AWS environment variable name and adds session token utility function
SingleLLMPromptExecutorIntegrationTest.kt	Enables Bedrock tests and updates to use session token
BedrockCredentialsSmokeTest.kt	New smoke test to verify Bedrock credentials
env.template.properties	Adds AWS credential environment variables template
build.gradle.kts	Adds AWS SDK dependencies for Bedrock tests
libs.versions.toml	Adds bedrock and sts SDK library definitions
heavy-tests.yml	Updates workflow to run Bedrock smoke test

Comments suppressed due to low confidence (1)

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/utils/TestUtils.kt:42

• The warning message is misleading. When AWS_SESSION_TOKEN is not set, null is returned, not a 'default session token'. Consider changing to 'using null (no session token)' or removing the message entirely since returning null for optional session tokens is expected behavior.

                println("WARNING: environment variable `AWS_SESSION_TOKEN` is not set, using default session token")

[github-actions]

Qodana for JVM

368 new problems were found

Inspection name	Severity	Problems
Check Kotlin and Java source code coverage	🔶 Warning	349
Unused import directive	🔶 Warning	15
Vulnerable imported dependency	🔶 Warning	4

@@ Code coverage @@
+ 64% total lines covered
7978 lines analyzed, 5180 lines covered
# Calculated according to the filters of your coverage tool

☁️ View the detailed Qodana report

Contact Qodana team

Contact us at [email protected]

• Or via our issue tracker: https://jb.gg/qodana-issue
• Or share your feedback: https://jb.gg/qodana-discussions