Package move and add Pydantic example

Author: johnsonr

examples-java/src/main/java/com/embabel/example/crew/README.md

@@ -0,0 +1 @@
+# Agents based on Crew AI samples

…babel/example/bookwriter/BookWriter.java …/example/crew/bookwriter/BookWriter.javaexamples-java/src/main/java/com/embabel/example/bookwriter/BookWriter.java renamed to examples-java/src/main/java/com/embabel/example/crew/bookwriter/BookWriter.java examples-java/src/main/java/com/embabel/example/bookwriter/BookWriter.java renamed to examples-java/src/main/java/com/embabel/example/crew/bookwriter/BookWriter.java

@@ -1,4 +1,4 @@
-package com.embabel.example.bookwriter;
+package com.embabel.example.crew.bookwriter;
 
 import com.embabel.agent.api.annotation.*;
 import com.embabel.agent.api.common.OperationContext;

…com/embabel/example/bookwriter/README.md …mbabel/example/crew/bookwriter/README.mdexamples-java/src/main/java/com/embabel/example/bookwriter/README.md renamed to examples-java/src/main/java/com/embabel/example/crew/bookwriter/README.md examples-java/src/main/java/com/embabel/example/bookwriter/README.md renamed to examples-java/src/main/java/com/embabel/example/crew/bookwriter/README.md

@@ -0,0 +1 @@
+# Agents based on Pydantic AI samples

examples-java/src/main/java/com/embabel/example/pydantic/README.md

@@ -0,0 +1,27 @@
+package com.embabel.example.pydantic.banksupport;
+
+import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Service
+class InMemoryCustomerRepository implements CustomerRepository {
+
+    private final List<Customer> customers = new ArrayList<>();
+
+    {
+        // Pre-populate with customer
+        customers.add(new Customer(123L, "John", 100.0f, 27.0f));
+    }
+
+    @Override
+    public Customer findById(Long id) {
+        return customers.stream()
+                .filter(customer -> customer.id().equals(id))
+                .findFirst()
+                .orElse(null);
+    }
+
+
+}

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

@@ -0,0 +1,17 @@
+# Pydantic Bank Support Example
+
+See https://ai.pydantic.dev/examples/bank-support/#running-the-example
+
+Our version demonstrates:
+
+- Integration with Spring Data for production-quality data access
+- A true domain model with LLM tools on domain objects.
+
+Run within Spring Shell with:
+
+```bash
+bank-support --id 123 --query "What's my balance?"
+
+bank-support --id 123 --query "What's my balance, including pending amounts. Also, I have lost my card. Can you help?"
+```
+

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

@@ -0,0 +1,65 @@
+package com.embabel.example.pydantic.banksupport;
+
+
+import com.embabel.agent.api.annotation.AchievesGoal;
+import com.embabel.agent.api.annotation.Action;
+import com.embabel.agent.api.annotation.Agent;
+import com.embabel.agent.api.common.OperationContext;
+import com.embabel.agent.config.models.OpenAiModels;
+import com.fasterxml.jackson.annotation.JsonPropertyDescription;
+import org.springframework.ai.tool.annotation.Tool;
+import org.springframework.data.repository.Repository;
+import org.springframework.lang.Nullable;
+
+record Customer(Long id, String name, float balance, float pendingAmount) {
+
+    @Tool(description = "Find the balance of a customer by id")
+    float balance(boolean includePending) {
+        return includePending ? balance + pendingAmount : balance;
+    }
+}
+
+interface CustomerRepository extends Repository<Customer, Long> {
+
+    @Nullable
+    Customer findById(Long id);
+}
+
+record SupportInput(
+        @JsonPropertyDescription("Customer ID") Long customerId,
+        @JsonPropertyDescription("Query from the customer") String query) {
+}
+
+record SupportOutput(
+        @JsonPropertyDescription("Advice returned to the customer") String advice,
+        @JsonPropertyDescription("Whether to block their card or not") boolean blockCard,
+        @JsonPropertyDescription("Risk level of query") int risk) {
+}
+
+@Agent(description = "Customer support agent")
+record SupportAgent(CustomerRepository customerRepository) {
+
+    @AchievesGoal(description = "Help bank customer with their query")
+    @Action
+    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);
+        }
+        return context.ai()
+                .withLlm(OpenAiModels.GPT_41_MINI)
+                .withToolObject(customer)
+                .createObject(
+                        """
+                                You are a support agent in our bank, give the
+                                customer support and judge the risk level of their query.
+                                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 $.
+                                
+                                There query: [%s]
+                                """.formatted(customer.name(), supportInput.query()),
+                        SupportOutput.class);
+    }
+
+}

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

@@ -0,0 +1,31 @@
+package com.embabel.example.pydantic.banksupport;
+
+import com.embabel.agent.api.common.autonomy.AgentInvocation;
+import com.embabel.agent.core.AgentPlatform;
+import com.embabel.agent.core.ProcessOptions;
+import org.springframework.shell.standard.ShellComponent;
+import org.springframework.shell.standard.ShellMethod;
+import org.springframework.shell.standard.ShellOption;
+
+@ShellComponent
+public record SupportAgentShellCommands(
+        AgentPlatform agentPlatform
+) {
+
+    @ShellMethod("Get bank support for a customer query")
+    public String bankSupport(
+            @ShellOption(value = "id", help = "customer id", defaultValue = "123") Long id,
+            @ShellOption(value = "query", help = "customer query", defaultValue = "What's my balance, including pending amounts?") String query
+    ) {
+        var supportInput = new SupportInput(id, query);
+        System.out.println("Support input: " + supportInput);
+        var invocation = AgentInvocation
+                .builder(agentPlatform)
+                .options(ProcessOptions.builder().verbosity(v -> v.showPrompts(true)).build())
+                .build(SupportOutput.class);
+        var result = invocation.invoke(supportInput);
+        return result.toString();
+    }
+
+
+}