Skip to content

Commit c71630a

Browse files
committed
Avoid IntelliJ warnings on unused method
1 parent e2d9523 commit c71630a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ void testWriteAndReviewAgent() {
1818
context.expectResponse(new WriteAndReviewAgent.Story("One upon a time Sir Galahad . . "));
1919

2020
var agent = new WriteAndReviewAgent(200, 400);
21-
agent.craftStory(new UserInput("Tell me a story about a brave knight", Instant.now()), context);
21+
var story = agent.craftStory(new UserInput("Tell me a story about a brave knight", Instant.now()), context);
2222

2323
var prompt = promptRunner.getLlmInvocations().getFirst().getMessages().getFirst().getContent();
2424
assertTrue(prompt.contains("knight"), "Expected prompt to contain 'knight'");
@@ -32,7 +32,7 @@ void testReview() {
3232
var story = new WriteAndReviewAgent.Story("Once upon a time, Sir Galahad...");
3333
var context = FakeOperationContext.create();
3434
context.expectResponse("A thrilling tale of bravery and adventure!");
35-
agent.reviewStory(userInput, story, context);
35+
var review = agent.reviewStory(userInput, story, context);
3636
var llmInvocation = context.getLlmInvocations().getFirst();
3737
var prompt = llmInvocation.getMessages().getFirst().getContent();
3838
assertTrue(prompt.contains("knight"), "Expected prompt to contain 'knight'");

0 commit comments

Comments
 (0)