From 40ada1e5ef8b1bfb60be5bebccb81681e83af3da Mon Sep 17 00:00:00 2001
From: alexheifetz <aheifetz@rocketmail.com>
Date: Fri, 8 Aug 2025 01:42:09 -0400
Subject: [PATCH 1/9] Update embabel-agent to stable version 0.1.0

---
 pom.xml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pom.xml b/pom.xml
index 0d4956b..730d6b4 100644
--- a/pom.xml
+++ b/pom.xml
@@ -18,7 +18,7 @@
 
     <properties>
         <java.version>21</java.version>
-        <embabel-agent.version>0.1.0-SNAPSHOT</embabel-agent.version>
+        <embabel-agent.version>0.1.0</embabel-agent.version>
     </properties>
 
     <dependencies>

From 3e4aeb8cd5551b0e27c92cf502979b92780deea4 Mon Sep 17 00:00:00 2001
From: alexheifetz <aheifetz@rocketmail.com>
Date: Sun, 10 Aug 2025 21:59:07 -0400
Subject: [PATCH 2/9] Update to work with agent-0.1.1-SNAP dev version

---
 .../java/com/embabel/template/agent/WriteAndReviewAgent.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
index 250eed4..39169e0 100644
--- a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
+++ b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
@@ -143,7 +143,7 @@ ReviewedStory reviewStory(UserInput userInput, Story story, OperationContext con
     Story craftStory(UserInput userInput, OperationContext context) {
         PromptRunner runner = context.promptRunner()
             // Higher temperature for more creative output
-            .withLlm(LlmOptions.fromCriteria(AutoModelSelectionCriteria.INSTANCE, 0.9))
+            .withLlm(LlmOptions.fromCriteria(AutoModelSelectionCriteria.INSTANCE))
             .withPromptContributor(Personas.WRITER);
 
         return runner.createObject(String.format("""

From 7920c9b1c8b7de64509caf12332451c30385ed40 Mon Sep 17 00:00:00 2001
From: alexheifetz <aheifetz@rocketmail.com>
Date: Sun, 10 Aug 2025 22:01:11 -0400
Subject: [PATCH 3/9] Switch to agent-0.1.1-SNAP

---
 pom.xml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pom.xml b/pom.xml
index 730d6b4..7b4210c 100644
--- a/pom.xml
+++ b/pom.xml
@@ -18,7 +18,7 @@
 
     <properties>
         <java.version>21</java.version>
-        <embabel-agent.version>0.1.0</embabel-agent.version>
+        <embabel-agent.version>0.1.1-SNAPSHOT</embabel-agent.version>
     </properties>
 
     <dependencies>

From 1362c1744f543b69441031dec93ecdf14f00d432 Mon Sep 17 00:00:00 2001
From: alexheifetz <aheifetz@rocketmail.com>
Date: Sun, 10 Aug 2025 22:10:17 -0400
Subject: [PATCH 4/9] Fix broken test.

---
 .../com/embabel/template/agent/WriteAndReviewAgentTest.java  | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java b/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java
index d3db4ee..7d2fa76 100644
--- a/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java
+++ b/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java
@@ -24,10 +24,7 @@ void testWriteAndReviewAgent() {
 
         String prompt = promptRunner.getLlmInvocations().getFirst().getPrompt();
         assertTrue(prompt.contains("knight"), "Expected prompt to contain 'knight'");
-
-        var temp = promptRunner.getLlmInvocations().getFirst().getInteraction().getLlm().getTemperature();
-        assertEquals(0.9, temp, 0.01,
-                "Expected temperature to be 0.9: Higher for more creative output");
+        
     }
 
     @Test

From 5b134c908060629f787500aff551d87ad1e92916 Mon Sep 17 00:00:00 2001
From: alexheifetz <aheifetz@rocketmail.com>
Date: Wed, 13 Aug 2025 01:59:33 -0400
Subject: [PATCH 5/9] Fixed compile errors.

---
 .../com/embabel/template/agent/WriteAndReviewAgent.java   | 8 ++------
 .../embabel/template/agent/WriteAndReviewAgentTest.java   | 2 +-
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
index 39169e0..6b7897d 100644
--- a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
+++ b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
@@ -41,17 +41,13 @@ abstract class Personas {
             "Roald Dahl",
             "A creative storyteller who loves to weave imaginative tales that are a bit unconventional",
             "Quirky",
-            "Create memorable stories that captivate the reader's imagination.",
-            "",
-            PromptContributionLocation.BEGINNING
+            "Create memorable stories that captivate the reader's imagination."
     );
     static final Persona REVIEWER = Persona.create(
             "Media Book Review",
             "New York Times Book Reviewer",
             "Professional and insightful",
-            "Help guide readers toward good stories",
-            "",
-            PromptContributionLocation.BEGINNING
+            "Help guide readers toward good stories"
     );
 }
 
diff --git a/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java b/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java
index 7d2fa76..b37485a 100644
--- a/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java
+++ b/src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java
@@ -24,7 +24,7 @@ void testWriteAndReviewAgent() {
 
         String prompt = promptRunner.getLlmInvocations().getFirst().getPrompt();
         assertTrue(prompt.contains("knight"), "Expected prompt to contain 'knight'");
-        
+
     }
 
     @Test

From 98d7bca2bdcd24c796580b225a4119e4f599cffd Mon Sep 17 00:00:00 2001
From: Rod Johnson <johnsonroda@gmail.com>
Date: Mon, 25 Aug 2025 12:13:03 +1000
Subject: [PATCH 6/9] Update API

---
 .idea/jarRepositories.xml                     |  5 ++++
 .../template/agent/WriteAndReviewAgent.java   | 25 ++++++++-----------
 2 files changed, 16 insertions(+), 14 deletions(-)

diff --git a/.idea/jarRepositories.xml b/.idea/jarRepositories.xml
index d288ac2..003cdc5 100644
--- a/.idea/jarRepositories.xml
+++ b/.idea/jarRepositories.xml
@@ -31,5 +31,10 @@
       <option name="name" value="JBoss Community repository" />
       <option name="url" value="https://repository.jboss.org/nexus/content/repositories/public/" />
     </remote-repository>
+    <remote-repository>
+      <option name="id" value="embabel-releases" />
+      <option name="name" value="embabel-releases" />
+      <option name="url" value="https://repo.embabel.com/artifactory/libs-release" />
+    </remote-repository>
   </component>
 </project>
\ No newline at end of file
diff --git a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
index 6b7897d..6ab2b6b 100644
--- a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
+++ b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
@@ -20,13 +20,10 @@
 import com.embabel.agent.api.annotation.Agent;
 import com.embabel.agent.api.annotation.Export;
 import com.embabel.agent.api.common.OperationContext;
-import com.embabel.agent.api.common.PromptRunner;
 import com.embabel.agent.domain.io.UserInput;
 import com.embabel.agent.domain.library.HasContent;
 import com.embabel.agent.prompt.persona.Persona;
-import com.embabel.common.ai.model.AutoModelSelectionCriteria;
 import com.embabel.common.ai.model.LlmOptions;
-import com.embabel.common.ai.prompt.PromptContributionLocation;
 import com.embabel.common.core.types.Timestamped;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Profile;
@@ -108,8 +105,9 @@ class WriteAndReviewAgent {
             export = @Export(remote = true, name = "writeAndReviewStory"))
     @Action
     ReviewedStory reviewStory(UserInput userInput, Story story, OperationContext context) {
-        String review = context.promptRunner()
-                .withLlm(LlmOptions.fromCriteria(AutoModelSelectionCriteria.INSTANCE))
+        var review = context
+                .ai()
+                .withAutoLlm()
                 .withPromptContributor(Personas.REVIEWER)
                 .generateText(String.format("""
                                 You will be given a short story to review.
@@ -137,12 +135,11 @@ ReviewedStory reviewStory(UserInput userInput, Story story, OperationContext con
 
     @Action
     Story craftStory(UserInput userInput, OperationContext context) {
-        PromptRunner runner = context.promptRunner()
-            // Higher temperature for more creative output
-            .withLlm(LlmOptions.fromCriteria(AutoModelSelectionCriteria.INSTANCE))
-            .withPromptContributor(Personas.WRITER);
-
-        return runner.createObject(String.format("""
+        return context.ai()
+                // Higher temperature for more creative output
+                .withLlm(LlmOptions.withAutoLlm().withTemperature(.7))
+                .withPromptContributor(Personas.WRITER)
+                .createObject(String.format("""
                                 Craft a short story in %d words or less.
                                 The story should be engaging and imaginative.
                                 Use the user's input as inspiration if possible.
@@ -151,8 +148,8 @@ Story craftStory(UserInput userInput, OperationContext context) {
                                 # User input
                                 %s
                                 """,
-            storyWordCount,
-            userInput.getContent()
-        ).trim(), Story.class);
+                        storyWordCount,
+                        userInput.getContent()
+                ).trim(), Story.class);
     }
 }
\ No newline at end of file

From b566915c4f4ede575187925caa06f288093c31d8 Mon Sep 17 00:00:00 2001
From: Rod Johnson <johnsonroda@gmail.com>
Date: Mon, 25 Aug 2025 12:28:50 +1000
Subject: [PATCH 7/9] Add InjectedDemo

---
 README.md                                     |  8 ++++++
 .../java/com/embabel/template/DemoShell.java  | 14 ++++++++++
 .../template/agent/WriteAndReviewAgent.java   | 12 ++++-----
 .../template/injected/InjectedDemo.java       | 27 +++++++++++++++++++
 4 files changed, 55 insertions(+), 6 deletions(-)
 create mode 100644 src/main/java/com/embabel/template/DemoShell.java
 create mode 100644 src/main/java/com/embabel/template/injected/InjectedDemo.java

diff --git a/README.md b/README.md
index c359214..06e8383 100644
--- a/README.md
+++ b/README.md
@@ -26,6 +26,8 @@ Add your magic here!
 
 Illustrates:
 
+- An injected demo showing how any Spring component can be injected with an Embabel `Ai` instance to enable it to
+  perform LLM operations.
 - A simple agent
 - Unit tests for an agent verifying prompts and hyperparameters
 
@@ -47,6 +49,12 @@ When the Embabel shell comes up, use the story agent like this:
 x "Tell me a story about...[your topic]"
 ```
 
+Try the `InjectedDemo` command to see simple, non-agent use:
+
+```java
+animal
+```
+
 ## A2A Support
 
 Embabel integrates with the [A2A](https://github.com/google-a2a/A2A) protocol, allowing you to connect to other
diff --git a/src/main/java/com/embabel/template/DemoShell.java b/src/main/java/com/embabel/template/DemoShell.java
new file mode 100644
index 0000000..c5976e0
--- /dev/null
+++ b/src/main/java/com/embabel/template/DemoShell.java
@@ -0,0 +1,14 @@
+package com.embabel.template;
+
+import com.embabel.template.injected.InjectedDemo;
+import org.springframework.shell.standard.ShellComponent;
+import org.springframework.shell.standard.ShellMethod;
+
+@ShellComponent
+record DemoShell(InjectedDemo injectedDemo) {
+
+    @ShellMethod("Invent an animal")
+    String animal() {
+        return injectedDemo.inventAnimal().toString();
+    }
+}
diff --git a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
index 6ab2b6b..18ca0d6 100644
--- a/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
+++ b/src/main/java/com/embabel/template/agent/WriteAndReviewAgent.java
@@ -23,6 +23,7 @@
 import com.embabel.agent.domain.io.UserInput;
 import com.embabel.agent.domain.library.HasContent;
 import com.embabel.agent.prompt.persona.Persona;
+import com.embabel.agent.prompt.persona.RoleGoalBackstory;
 import com.embabel.common.ai.model.LlmOptions;
 import com.embabel.common.core.types.Timestamped;
 import org.springframework.beans.factory.annotation.Value;
@@ -34,12 +35,11 @@
 import java.time.format.DateTimeFormatter;
 
 abstract class Personas {
-    static final Persona WRITER = Persona.create(
-            "Roald Dahl",
-            "A creative storyteller who loves to weave imaginative tales that are a bit unconventional",
-            "Quirky",
-            "Create memorable stories that captivate the reader's imagination."
-    );
+    static final RoleGoalBackstory WRITER = RoleGoalBackstory
+            .withRole("Creative Storyteller")
+            .andGoal("Write engaging and imaginative stories")
+            .andBackstory("Has a PhD in French literature; used to work in a circus");
+
     static final Persona REVIEWER = Persona.create(
             "Media Book Review",
             "New York Times Book Reviewer",
diff --git a/src/main/java/com/embabel/template/injected/InjectedDemo.java b/src/main/java/com/embabel/template/injected/InjectedDemo.java
new file mode 100644
index 0000000..0392f80
--- /dev/null
+++ b/src/main/java/com/embabel/template/injected/InjectedDemo.java
@@ -0,0 +1,27 @@
+package com.embabel.template.injected;
+
+import com.embabel.agent.api.common.Ai;
+import org.springframework.stereotype.Component;
+
+/**
+ * Demonstrate injection of Embabel's OperationContext into a Spring component.
+ *
+ * @param ai Embabel AI helper, injected by Spring
+ */
+@Component
+public record InjectedDemo(Ai ai) {
+
+    public record Animal(String name, String species) {
+    }
+
+    public Animal inventAnimal() {
+        return ai
+                .withDefaultLlm()
+                .createObject("""
+                                You just woke up in a magical forest.
+                                Invent a fictional animal.
+                                The animal should have a name and a species.
+                                """,
+                        Animal.class);
+    }
+}

From 281850a1d6ee02e2718fe5294b1449b4de526a3e Mon Sep 17 00:00:00 2001
From: Rod Johnson <johnsonroda@gmail.com>
Date: Mon, 25 Aug 2025 17:12:13 +1000
Subject: [PATCH 8/9] Upgrade to 0.1.1

---
 pom.xml                                                       | 4 ++--
 .../com/embabel/{template => }/ProjectNameApplication.java    | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)
 rename src/main/java/com/embabel/{template => }/ProjectNameApplication.java (97%)

diff --git a/pom.xml b/pom.xml
index 7b4210c..cb108aa 100644
--- a/pom.xml
+++ b/pom.xml
@@ -18,7 +18,7 @@
 
     <properties>
         <java.version>21</java.version>
-        <embabel-agent.version>0.1.1-SNAPSHOT</embabel-agent.version>
+        <embabel-agent.version>0.1.1</embabel-agent.version>
     </properties>
 
     <dependencies>
@@ -66,7 +66,7 @@
             <snapshots>
                 <enabled>false</enabled>
             </snapshots>
-        </repository>     
+        </repository>
         <repository>
             <id>embabel-snapshots</id>
             <url>https://repo.embabel.com/artifactory/libs-snapshot</url>
diff --git a/src/main/java/com/embabel/template/ProjectNameApplication.java b/src/main/java/com/embabel/ProjectNameApplication.java
similarity index 97%
rename from src/main/java/com/embabel/template/ProjectNameApplication.java
rename to src/main/java/com/embabel/ProjectNameApplication.java
index 85f9ad4..9f89218 100644
--- a/src/main/java/com/embabel/template/ProjectNameApplication.java
+++ b/src/main/java/com/embabel/ProjectNameApplication.java
@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package com.embabel.template;
+package com.embabel;
 
 import com.embabel.agent.config.annotation.EnableAgentShell;
 import com.embabel.agent.config.annotation.EnableAgents;

From 43b29278ac82b3838f947cddf151b0871266d46d Mon Sep 17 00:00:00 2001
From: Rod Johnson <johnsonroda@gmail.com>
Date: Mon, 25 Aug 2025 17:16:07 +1000
Subject: [PATCH 9/9] Remove integration test superclass

---
 .../EmbabelMockitoIntegrationTest.java        | 139 ------------------
 .../WriteAndReviewAgentIntegrationTest.java   |   4 +-
 2 files changed, 2 insertions(+), 141 deletions(-)
 delete mode 100644 src/test/java/com/embabel/agent/testing/integration/EmbabelMockitoIntegrationTest.java

diff --git a/src/test/java/com/embabel/agent/testing/integration/EmbabelMockitoIntegrationTest.java b/src/test/java/com/embabel/agent/testing/integration/EmbabelMockitoIntegrationTest.java
deleted file mode 100644
index 787b39f..0000000
--- a/src/test/java/com/embabel/agent/testing/integration/EmbabelMockitoIntegrationTest.java
+++ /dev/null
@@ -1,139 +0,0 @@
-package com.embabel.agent.testing.integration;
-
-import com.embabel.agent.core.AgentPlatform;
-import com.embabel.agent.spi.LlmInteraction;
-import com.embabel.agent.spi.LlmOperations;
-import org.mockito.ArgumentCaptor;
-import org.mockito.ArgumentMatcher;
-import org.mockito.Mockito;
-import org.mockito.stubbing.OngoingStubbing;
-import org.mockito.verification.VerificationMode;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.boot.test.context.SpringBootTest;
-import org.springframework.test.context.TestPropertySource;
-import org.springframework.test.context.bean.override.mockito.MockitoBean;
-
-import static org.mockito.ArgumentMatchers.*;
-import static org.mockito.Mockito.verify;
-import static org.mockito.Mockito.when;
-
-@SpringBootTest
-@TestPropertySource(properties = {
-        "embabel.agent.llm.default-model=test-model",
-        "embabel.agent.verbosity.debug=true",
-        "spring.shell.interactive.enabled=false",
-        "spring.shell.noninteractive.enabled=false"
-})
-public class EmbabelMockitoIntegrationTest {
-
-    @Autowired
-    protected AgentPlatform agentPlatform;
-
-    @MockitoBean
-    protected LlmOperations llmOperations;
-
-    // Stubbing methods
-    protected <T> OngoingStubbing<T> whenCreateObject(String prompt, Class<T> outputClass, LlmInteraction llmInteraction) {
-        // Mock the lower level LLM operation to create an object
-        // that will ultimately be called
-        return when(llmOperations.createObject(prompt, llmInteraction, eq(outputClass), any(), any()));
-    }
-
-    protected <T> OngoingStubbing<T> whenCreateObject(String prompt, Class<T> outputClass) {
-        return when(llmOperations.createObject(prompt, any(), eq(outputClass), any(), any()));
-    }
-
-    protected OngoingStubbing<String> whenGenerateText(String prompt, LlmInteraction llmInteraction) {
-        return when(llmOperations.createObject(prompt, llmInteraction, eq(String.class), any(), any()));
-    }
-
-    protected OngoingStubbing<String> whenGenerateText(String prompt) {
-        return when(llmOperations.createObject(prompt, any(), eq(String.class), any(), any()));
-    }
-
-    // Verification methods
-    protected <T> void verifyCreateObject(String prompt, Class<T> outputClass, LlmInteraction llmInteraction) {
-        verify(llmOperations).createObject(eq(prompt), eq(llmInteraction), eq(outputClass), any(), any());
-    }
-
-    protected <T> void verifyCreateObject(String prompt, Class<T> outputClass) {
-        verify(llmOperations).createObject(eq(prompt), any(), eq(outputClass), any(), any());
-    }
-
-    protected <T> void verifyCreateObject(String prompt, Class<T> outputClass, LlmInteraction llmInteraction, VerificationMode mode) {
-        verify(llmOperations, mode).createObject(eq(prompt), eq(llmInteraction), eq(outputClass), any(), any());
-    }
-
-    protected <T> void verifyCreateObject(String prompt, Class<T> outputClass, VerificationMode mode) {
-        verify(llmOperations, mode).createObject(eq(prompt), any(), eq(outputClass), any(), any());
-    }
-
-    protected void verifyGenerateText(String prompt, LlmInteraction llmInteraction) {
-        verify(llmOperations).createObject(eq(prompt), eq(llmInteraction), eq(String.class), any(), any());
-    }
-
-    protected void verifyGenerateText(String prompt) {
-        verify(llmOperations).createObject(eq(prompt), any(), eq(String.class), any(), any());
-    }
-
-    protected void verifyGenerateText(String prompt, LlmInteraction llmInteraction, VerificationMode mode) {
-        verify(llmOperations, mode).createObject(eq(prompt), eq(llmInteraction), eq(String.class), any(), any());
-    }
-
-    protected void verifyGenerateText(String prompt, VerificationMode mode) {
-        verify(llmOperations, mode).createObject(eq(prompt), any(), eq(String.class), any(), any());
-    }
-
-    // Verification methods with argument matchers
-    protected <T> void verifyCreateObjectMatching(ArgumentMatcher<String> promptMatcher, Class<T> outputClass) {
-        verify(llmOperations).createObject(argThat(promptMatcher), any(), eq(outputClass), any(), any());
-    }
-
-    protected <T> void verifyCreateObjectMatching(ArgumentMatcher<String> promptMatcher, Class<T> outputClass, ArgumentMatcher<LlmInteraction> llmInteractionMatcher) {
-        verify(llmOperations).createObject(argThat(promptMatcher),
-                argThat(llmInteractionMatcher),
-                eq(outputClass), any(), any());
-    }
-
-    protected <T> void verifyCreateObjectMatching(ArgumentMatcher<String> promptMatcher, Class<T> outputClass, LlmInteraction llmInteraction) {
-        verify(llmOperations).createObject(argThat(promptMatcher), eq(llmInteraction), eq(outputClass), any(), any());
-    }
-
-    protected <T> void verifyCreateObjectMatching(ArgumentMatcher<String> promptMatcher, Class<T> outputClass, VerificationMode mode) {
-        verify(llmOperations, mode).createObject(argThat(promptMatcher), any(), eq(outputClass), any(), any());
-    }
-
-    protected void verifyGenerateTextMatching(ArgumentMatcher<String> promptMatcher) {
-        verify(llmOperations).createObject(argThat(promptMatcher), any(), eq(String.class), any(), any());
-    }
-
-    protected void verifyGenerateTextMatching(ArgumentMatcher<String> promptMatcher, LlmInteraction llmInteraction) {
-        Mockito.verify(llmOperations).createObject(argThat(promptMatcher), eq(llmInteraction), eq(String.class), any(), any());
-    }
-
-    protected void verifyGenerateTextMatching(ArgumentMatcher<String> promptMatcher, VerificationMode mode) {
-        Mockito.verify(llmOperations, mode).createObject(argThat(promptMatcher), any(), eq(String.class), any(), any());
-    }
-
-    // Convenience verification methods
-    protected void verifyNoInteractions() {
-        Mockito.verifyNoInteractions(llmOperations);
-    }
-
-    protected void verifyNoMoreInteractions() {
-        Mockito.verifyNoMoreInteractions(llmOperations);
-    }
-
-    // Argument captor helpers
-    protected ArgumentCaptor<String> capturePrompt() {
-        return ArgumentCaptor.forClass(String.class);
-    }
-
-    protected ArgumentCaptor<LlmInteraction> captureLlmInteraction() {
-        return ArgumentCaptor.forClass(LlmInteraction.class);
-    }
-
-    protected <T> ArgumentCaptor<Class<T>> captureOutputClass() {
-        return ArgumentCaptor.forClass(Class.class);
-    }
-}
\ No newline at end of file
diff --git a/src/test/java/com/embabel/template/agent/WriteAndReviewAgentIntegrationTest.java b/src/test/java/com/embabel/template/agent/WriteAndReviewAgentIntegrationTest.java
index 9ea9036..a3ece1e 100644
--- a/src/test/java/com/embabel/template/agent/WriteAndReviewAgentIntegrationTest.java
+++ b/src/test/java/com/embabel/template/agent/WriteAndReviewAgentIntegrationTest.java
@@ -13,7 +13,7 @@
  * This will run under Spring Boot against an AgentPlatform instance
  * that has loaded all our agents.
  */
-class StoryWriterIntegrationTest extends EmbabelMockitoIntegrationTest {
+class WriteAndReviewAgentIntegrationTest extends EmbabelMockitoIntegrationTest {
 
     @Test
     void shouldExecuteCompleteWorkflow() {
@@ -39,7 +39,7 @@ void shouldExecuteCompleteWorkflow() {
                 "Expected review to match: " + reviewedStoryResult);
 
         verifyCreateObjectMatching(prompt -> prompt.contains("Craft a short story"), Story.class,
-                llm -> llm.getLlm().getTemperature() == 0.9 && llm.getToolGroups().isEmpty());
+                llm -> llm.getLlm().getTemperature() == 0.7 && llm.getToolGroups().isEmpty());
         verifyGenerateTextMatching(prompt -> prompt.contains("You will be given a short story to review"));
         verifyNoMoreInteractions();
     }