Summaries

Author: johnsonr

src/main/java/com/embabel/grouper/agent/Grouper.java

@@ -131,16 +131,18 @@ Model.Positioning evolvePositioning(
         logger.info("Evolving positioning based on FocusGroupRun {}", focusGroupRun);
         // TODO Should handle > 1 message
         var messageVariants = focusGroupRun.positioning.messageVariants().getFirst();
-        var newMessageWordings = config.nextCreative()
+        var creativeControl = config.nextCreative()
                 .promptRunner(ai)
                 .withPromptContributor(messageVariants.message())
-                .creating(NewMessageWordings.class)
+                .creating(CreativeControl.class)
                 .fromPrompt("""
                         Given the objectives, consider
                         the following feedback:
                         %s
                         
-                        Create new message wordings we could try.
+                        1. Summarize the feedback in no more than %d words.
+                        
+                        2. Create new message wordings we could try.
                         
                         Be creative. Try to break through!
                         
@@ -150,13 +152,15 @@ Model.Positioning evolvePositioning(
                         %s
                         """.formatted(
                         focusGroupRun.infoString(true, 1),
+                        config.findingsWordCount(),
                         config.maxVariants(),
                         bestScoringVariants)
                 );
-        logger.info("New wordings: {}", newMessageWordings);
+        logger.info("Creative input: {}", creativeControl);
+        bestScoringVariants.addFinding(creativeControl.summary);
         var newMessageVariants = new Model.MessageVariants(
                 messageVariants.message(),
-                newMessageWordings.wordings().toArray(new String[0])
+                creativeControl.wordings().toArray(new String[0])
         );
 
         return new Model.Positioning(List.of(newMessageVariants));
@@ -171,11 +175,12 @@ Model.BestScoringVariants results(
             OperationContext context) {
         return bestScoringVariants;
     }
-}
 
+    private record CreativeControl(
+            String summary,
+            List<String> wordings
+    ) {
+    }
+}
 
-record NewMessageWordings(
-        List<String> wordings
-) {
 
-}

src/main/java/com/embabel/grouper/agent/Grouper2.java

@@ -16,6 +16,7 @@ public record GrouperConfig(
         int maxVariants,
         int maxIterations,
         double minMessageScore,
+        int findingsWordCount,
         List<Actor<RoleGoalBackstory>> creatives
 ) implements Predicate<FocusGroupRun> {
 

src/main/java/com/embabel/grouper/agent/GrouperConfig.java

@@ -177,6 +177,7 @@ public record MessageVariantScore(
     public static class BestScoringVariants {
         private Vector<MessageVariantScore> bestVariants = Vector.empty();
         private final GrouperConfig config;
+        private Vector<String> findings = Vector.empty();
 
         public BestScoringVariants(GrouperConfig config) {
             this.config = config;
@@ -186,7 +187,8 @@ public List<Model.MessageVariantScore> bestVariants() {
             return bestVariants.asJava();
         }
 
-        public void updateFrom(FocusGroupRun focusGroupRun, GrouperConfig config) {
+        public void updateFrom(FocusGroupRun focusGroupRun,
+                               GrouperConfig config) {
             var newScores = Vector.ofAll(
                     focusGroupRun.positioning.messageVariants().stream()
                             .flatMap(mv -> mv.expressions().stream())
@@ -202,13 +204,23 @@ public void updateFrom(FocusGroupRun focusGroupRun, GrouperConfig config) {
                     .take(config.maxVariants());
         }
 
+        public void addFinding(String finding) {
+            findings = findings.append(finding);
+        }
+
         @NotNull
         @Override
         public String toString() {
-            return bestVariants
+            var variants = bestVariants
                     .sorted(Comparator.comparingDouble(Model.MessageVariantScore::normalizedScore).reversed())
                     .map(mv -> "%.2f: %s".formatted(config.decisionScore(mv), mv.messageVariant().wording()))
                     .collect(Collectors.joining("\n"));
+            return """
+                    %s
+                    
+                    Findings:
+                    - %s
+                    """.formatted(variants, findings.collect(Collectors.joining("\n- ")));
 
         }
     }

src/main/java/com/embabel/grouper/domain/Model.java

@@ -42,3 +42,4 @@ grouper:
   max-variants: 10
   max-iterations: 10
   min-message-score: 0.9
+  findings-word-count: 60