Handoff example

Author: johnsonr

examples-java/src/main/java/com/embabel/example/handoff/Handoff.java

@@ -0,0 +1,43 @@
+package com.embabel.example.handoff;
+
+import com.embabel.agent.api.annotation.*;
+import com.embabel.agent.api.common.OperationContext;
+import com.embabel.agent.domain.io.UserInput;
+import com.embabel.common.ai.model.LlmOptions;
+import com.embabel.example.horoscope.StarPerson;
+import com.embabel.example.wikipedia.ResearchSubject;
+
+record Choice(String choice) {
+
+}
+
+record AdventureLog(String log) {
+}
+
+@Agent(description = "Choose your own adventure handoff agent")
+public class Handoff {
+
+    /**
+     * This action is get user input out of the way in selecting the adventure by handoff
+     * @param userInput
+     * @param operationContext
+     * @return
+     */
+    @Action
+    Choice makeChoice(UserInput userInput, OperationContext operationContext) {
+        return WaitFor.formSubmission("Please choose your own adventure", Choice.class);
+    }
+
+    @Action
+    @AchievesGoal(description = "Choose your own adventure",
+        export = @Export(remote = true))
+    AdventureLog chooseAdventure(Choice choice, OperationContext operationContext) {
+        return operationContext.promptRunner()
+                .withLlm(LlmOptions.fromModel("gpt-4.1-nano"))
+                .withHandoffs(StarPerson.class, ResearchSubject.class)
+                .createObject("""
+                    Based on the user's input, decide what to do. You might do something related to a horoscope or some fact checking.
+                    What they said: %s
+                    """.formatted(choice.choice()), AdventureLog.class);
+    }
+}

examples-java/src/main/java/com/embabel/example/horoscope/StarNewsFinder.java

@@ -15,10 +15,7 @@
  */
 package com.embabel.example.horoscope;
 
-import com.embabel.agent.api.annotation.AchievesGoal;
-import com.embabel.agent.api.annotation.Action;
-import com.embabel.agent.api.annotation.Agent;
-import com.embabel.agent.api.annotation.WaitFor;
+import com.embabel.agent.api.annotation.*;
 import com.embabel.agent.api.common.PromptRunner;
 import com.embabel.agent.config.models.OpenAiModels;
 import com.embabel.agent.core.CoreToolGroups;
@@ -118,7 +115,11 @@ public RelevantNewsStories findNewsStories(StarPerson person, Horoscope horoscop
     // The @AchievesGoal annotation indicates that completing this action
     // achieves the given goal, so the agent can be complete
     @AchievesGoal(
-            description = "Write an amusing writeup for the target person based on their horoscope and current news stories"
+            description = "Write an amusing writeup for the target person based on their horoscope and current news stories",
+            export = @Export(
+                    remote = true,
+                    name = "starNewsWriteup",
+                    startingInputTypes = {StarPerson.class, UserInput.class})
     )
     @Action
     public Writeup writeup(

examples-java/src/main/java/com/embabel/example/wikipedia/ResearchSubject.java

@@ -0,0 +1,4 @@
+package com.embabel.example.wikipedia;
+
+public record ResearchSubject(String name) {
+}

examples-java/src/main/java/com/embabel/example/wikipedia/WikiAgent.java

@@ -25,9 +25,6 @@
 import java.util.List;
 import java.util.stream.Collectors;
 
-record ResearchSubject(String name) {
-}
-
 record FocusAreas(List<FocusArea> focusAreas) {}
 
 record FocusArea(String topic) {