Transactional annotation on agent breaks deployment - investigation

Author: igordayen

examples-common/pom.xml

@@ -18,6 +18,20 @@
             <artifactId>embabel-agent-api</artifactId>
         </dependency>
 
+        <!-- H2 Database -->
+        <dependency>
+            <groupId>com.h2database</groupId>
+            <artifactId>h2</artifactId>
+            <scope>runtime</scope>
+        </dependency>
+
+        <!-- Spring Boot Data JPA Starter (includes transaction management) -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-jpa</artifactId>
+        </dependency>
+
+
         <!-- Test Dependencies -->
         <dependency>
             <groupId>com.embabel.agent</groupId>

examples-java/src/main/java/com/embabel/example/JavaAgentShellApplication.java

@@ -20,7 +20,10 @@
 import com.embabel.agent.config.annotation.McpServers;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.autoconfigure.domain.EntityScan;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
+import org.springframework.transaction.annotation.EnableTransactionManagement;
 
 
 /**
@@ -41,6 +44,14 @@
         loggingTheme = LoggingThemes.SEVERANCE,
         mcpServers = {McpServers.DOCKER_DESKTOP}
 )
+/*
+ Unused, but may be required for JPA support
+@@EnableJpaRepositories (basePackages = "com.embabel.example.pydantic.banksupport")
+
+@EntityScan(basePackages = "com.embabel.example.pydantic.banksupport")
+@EnableTransactionManagement(proxyTargetClass = false)
+*/
+
 public class JavaAgentShellApplication {
 
     /**

examples-java/src/main/java/com/embabel/example/pydantic/banksupport/CustomerEntity.java

@@ -0,0 +1,54 @@
+/*
+ * Copyright 2024-2025 Embabel Software, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.embabel.example.pydantic.banksupport;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.Id;
+import org.springframework.ai.tool.annotation.Tool;
+import jakarta.persistence.GenerationType;
+
+//@Entity
+
+/**
+ * Unused JPA Entity.  Keep till further investigation
+ */
+public class CustomerEntity  {
+
+   @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    private String name;
+    private float balance;
+    private float pendingAmount;
+
+    public CustomerEntity() {
+    } // Required by JPA
+
+    public CustomerEntity(Long id, String name, float balance, float pendingAmount) {
+        this.id = id;
+        this.name = name;
+        this.balance = balance;
+        this.pendingAmount = pendingAmount;
+    }
+    @Tool(description = "Find the balance of a customer by id")
+    public float getBalance(boolean includePending) {
+        return includePending ? balance + pendingAmount : balance;
+    }
+
+    // Getters and setters
+}

examples-java/src/main/java/com/embabel/example/pydantic/banksupport/SupportAgent.java

@@ -23,8 +23,11 @@
 import com.embabel.agent.config.models.OpenAiModels;
 import com.fasterxml.jackson.annotation.JsonPropertyDescription;
 import org.springframework.ai.tool.annotation.Tool;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.repository.Repository;
 import org.springframework.lang.Nullable;
+import org.springframework.transaction.annotation.EnableTransactionManagement;
+import org.springframework.transaction.annotation.Transactional;
 
 record Customer(Long id, String name, float balance, float pendingAmount) {
 
@@ -34,7 +37,7 @@ float balance(boolean includePending) {
     }
 }
 
-interface CustomerRepository extends Repository<Customer, Long> {
+interface CustomerRepository /*extends Repository<Customer, Long> */ {
 
     @Nullable
     Customer findById(Long id);
@@ -51,12 +54,22 @@ record SupportOutput(
         @JsonPropertyDescription("Risk level of query") int risk) {
 }
 
+interface SupportAgentSpec {
+
+    SupportOutput supportCustomer(SupportInput supportInput, OperationContext context);
+}
+
+@Transactional
 @Agent(description = "Customer support agent")
-record SupportAgent(CustomerRepository customerRepository) {
+class SupportAgent/*(InMemoryCustomerRepository customerRepository)*/ implements SupportAgentSpec {
+
+    @Autowired
+    CustomerRepository customerRepository;
 
+    @Transactional
     @AchievesGoal(description = "Help bank customer with their query")
     @Action
-    SupportOutput supportCustomer(SupportInput supportInput, OperationContext context) {
+    public SupportOutput supportCustomer(SupportInput supportInput, OperationContext context) {
         var customer = customerRepository.findById(supportInput.customerId());
         if (customer == null) {
             return new SupportOutput("Customer not found with this id", false, 0);
@@ -71,7 +84,7 @@ SupportOutput supportCustomer(SupportInput supportInput, OperationContext contex
                                 In some cases, you may need to block their card. In this case, explain why.
                                 Reply using the customer's name, "%s".
                                 Currencies are in $.
-                                
+                                                                
                                 Their query: [%s]
                                 """.formatted(customer.name(), supportInput.query()),
                         SupportOutput.class);

examples-java/src/main/resources/application.yml

@@ -39,4 +39,15 @@ examples:
         Write a well-structured chapter for a book based on the provided chapter title, goal, and outline.
       backstory: >
         You are an exceptional writer, known for producing engaging, well-researched, and informative content.
-        You excel at transforming complex ideas into readable and well-organized chapters.
+        You excel at transforming complex ideas into readable and well-organized chapters.
+  spring:
+    datasource:
+      url: jdbc:h2:mem:examples;DB_CLOSE_DELAY=-1;MODE=PostgreSQL
+      driver-class-name: org.h2.Driver
+    jpa:
+      hibernate:
+        ddl-auto: create-drop
+    h2:
+      console:
+        enabled: true  # Web console at /h2-console
+