Upgrade to 0.1.4

Author: johnsonr

pom.xml

@@ -18,7 +18,7 @@
 
  <properties>
  <java.version>21</java.version>
- <embabel-agent.version>0.1.3</embabel-agent.version>
+ <embabel-agent.version>0.1.4</embabel-agent.version>
  </properties>
 
  <dependencies>

src/test/java/com/embabel/template/agent/WriteAndReviewAgentTest.java

@@ -20,7 +20,7 @@ void testWriteAndReviewAgent() {
  var agent = new WriteAndReviewAgent(200, 400);
  agent.craftStory(new UserInput("Tell me a story about a brave knight", Instant.now()), context);
 
- String prompt = promptRunner.getLlmInvocations().getFirst().getPrompt();
+ var prompt = promptRunner.getLlmInvocations().getFirst().getMessages().getFirst().getContent();
  assertTrue(prompt.contains("knight"), "Expected prompt to contain 'knight'");
 
  }
@@ -34,8 +34,9 @@ void testReview() {
  context.expectResponse("A thrilling tale of bravery and adventure!");
  agent.reviewStory(userInput, story, context);
  var llmInvocation = context.getLlmInvocations().getFirst();
- assertTrue(llmInvocation.getPrompt().contains("knight"), "Expected prompt to contain 'knight'");
- assertTrue(llmInvocation.getPrompt().contains("review"), "Expected prompt to contain 'review'");
+ var prompt = llmInvocation.getMessages().getFirst().getContent();
+ assertTrue(prompt.contains("knight"), "Expected prompt to contain 'knight'");
+ assertTrue(prompt.contains("review"), "Expected prompt to contain 'review'");
  }
 
 }