Skip to content

Commit

Permalink
Updates SummarizePage test
Browse files Browse the repository at this point in the history
Now that page summaries are in their own conversation turn, we need to adjust the indexes we check in the SummarizePage test.
  • Loading branch information
jonathansampson committed Oct 17, 2024
1 parent 423ede2 commit aa38d53
Showing 1 changed file with 4 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -384,7 +384,10 @@ TEST_F(EngineConsumerOAIUnitTest, SummarizePage) {
EXPECT_EQ(*messages[1].GetDict().Find("role"), "user");
EXPECT_EQ(*messages[1].GetDict().Find("content"),
"This is the text of a web page:\n<page>\nThis is a "
"page.\n</page>\n\nTell me more about this page");
"page.\n</page>\n\n");
EXPECT_EQ(*messages[2].GetDict().Find("role"), "user");
EXPECT_EQ(*messages[2].GetDict().Find("content"),
"Tell me more about this page");
std::move(completed_callback)
.Run(EngineConsumer::GenerationResult(""));
});
Expand Down

0 comments on commit aa38d53

Please sign in to comment.