File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/test/java/com/embabel/template/agent Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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'" );
You can’t perform that action at this time.
0 commit comments