Bedrock

Author: rogervinas

chat-server/build.gradle.kts

@@ -30,11 +30,11 @@ dependencies {
     implementation("org.springframework.ai:spring-ai-mcp-client-spring-boot-starter")
 
     // ollama
-    implementation("org.springframework.ai:spring-ai-ollama-spring-boot-starter")
+    // implementation("org.springframework.ai:spring-ai-ollama-spring-boot-starter")
 
     // bedrock
-    // implementation("org.springframework.ai:spring-ai-bedrock-converse-spring-boot-starter")
-    // implementation("org.springframework.ai:spring-ai-bedrock-ai-spring-boot-starter")
+    implementation("org.springframework.ai:spring-ai-bedrock-converse-spring-boot-starter")
+    implementation("org.springframework.ai:spring-ai-bedrock-ai-spring-boot-starter")
 
     implementation("org.springframework.ai:spring-ai-pgvector-store-spring-boot-starter")
     runtimeOnly("org.postgresql:postgresql")

chat-server/src/main/resources/application-bedrock.yml

@@ -0,0 +1,20 @@
+spring:
+  ai:
+    bedrock:
+      aws:
+        access-key: "${AWS_ACCESS_KEY_ID}"
+        secret-key: "${AWS_SECRET_ACCESS_KEY}"
+        region: "${AWS_REGION:eu-central-1}"
+      converse:
+        chat:
+          enabled: true
+          options:
+            model: "${AWS_BEDROCK_CHAT_MODEL}"
+      cohere:
+        embedding:
+          enabled: true
+          model: "${AWS_BEDROCK_EMBEDDING_MODEL}"
+    vectorstore:
+      pgvector:
+        dimensions: 1024
+        initialize-schema: true

chat-server/src/main/resources/application.yml

@@ -1,6 +1,6 @@
 spring:
   profiles:
-    active: "ollama"
+    active: "bedrock"
   application:
     name: chat-server
   datasource:

chat-server/src/test/kotlin/com/rogervinas/ChatServerApplicationTest.kt

@@ -39,6 +39,7 @@ import java.util.UUID
 @ActiveProfiles("test", "ollama")
 @TestMethodOrder(MethodOrderer.OrderAnnotation::class)
 @Testcontainers
+@DisabledIfCI
 class ChatServerApplicationTest {
 
     companion object {