Skip to content

Commit 8110f38

Browse files
committed
Fix broken tests
1 parent 0a9fda8 commit 8110f38

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

examples-java/src/test/java/com/embabel/example/horoscope/StarNewsFinderTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,9 @@ void writeupPromptMustContainKeyData() {
5555
RelevantNewsStories relevantNewsStories = new RelevantNewsStories(Arrays.asList(cockatoos, emus));
5656
Horoscope horoscope = new Horoscope("This is a good day for you");
5757

58-
starNewsFinder.writeup(starPerson, relevantNewsStories, horoscope, context);
58+
starNewsFinder.writeup(starPerson, relevantNewsStories, horoscope, context.ai());
5959

60-
var prompt = context.getLlmInvocations().getFirst().getPrompt();
60+
var prompt = context.getLlmInvocations().getFirst().getMessages().getFirst().getContent();
6161
var toolGroups = context.getLlmInvocations().getFirst().getInteraction().getToolGroups();
6262

6363

examples-kotlin/src/test/kotlin/com/embabel/example/horoscope/StarNewsFinderTest.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ class StarNewsFinderTest {
7676
context = context
7777
)
7878

79-
val prompt = context.llmInvocations.first().prompt
79+
val prompt = context.llmInvocations.first().messages.first().content
8080
val toolGroups = context.llmInvocations.first().interaction.toolGroups
8181

8282
assertTrue(prompt.contains(starPerson.name))

0 commit comments

Comments
 (0)