Add WikiAgent

Author: johnsonr

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

@@ -0,0 +1,98 @@
+/*
+ * Copyright 2024-2025 Embabel Software, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.embabel.example.wikipedia;
+
+import com.embabel.agent.api.annotation.*;
+import com.embabel.agent.api.common.OperationContext;
+import com.embabel.agent.core.CoreToolGroups;
+import com.embabel.common.ai.model.BuildableLlmOptions;
+import com.embabel.common.ai.model.LlmOptions;
+import org.springframework.beans.factory.annotation.Value;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+record ResearchSubject(String name) {
+}
+
+record FocusAreas(List<FocusArea> focusAreas) {}
+
+record FocusArea(String topic) {
+}
+
+record ResearchReport(String subject, String summary) {
+}
+
+/**
+ * Demonstrates an agent that interacts with the user
+ */
+@Agent(description = "Find information from wikipedia according to the user's request.")
+public class WikiAgent {
+
+    private final int wordCount;
+    private final BuildableLlmOptions llm;
+
+    public WikiAgent(
+            @Value("${wiki.wordcount:200}") int wordCount,
+            @Value("${wiki.llm:gpt-4.1-nano}") String model) {
+        this.wordCount = wordCount;
+        this.llm = LlmOptions.fromModel(model);
+    }
+
+    @Action
+    FocusAreas focusAreas(ResearchSubject researchSubject, OperationContext context) {
+        return context.promptRunner()
+                .withLlm(llm.withTemperature(.6))
+                .withToolGroup(CoreToolGroups.WEB)
+                .createObject("""
+                    What are some interesting topics to explore about %s?
+                    Please provide a list of focus areas, each as a separate line.
+                    Go beyond the obvious: emphasize quirky and lesser-known aspects.
+                    Use Wikipedia as the source of information.
+                    """.formatted(researchSubject.name()), FocusAreas.class);
+    }
+
+    @Action
+    FocusArea findFocusArea(ResearchSubject researchSubject, FocusAreas focusAreas) {
+        return WaitFor.formSubmission(
+                """
+                    Great, there are lots of interesting things to explore about %s.
+                    Choices:
+                        %s
+                    Please select one of the topics above to focus on.
+                    """.formatted(researchSubject.name(),
+                                focusAreas.focusAreas().stream().map(FocusArea::topic)
+                                        .collect(Collectors.joining("\n"))),
+                FocusArea.class);
+    }
+
+    @Action
+    @AchievesGoal(description = "Find information from wikipedia according to the user's request.",
+            export = @Export(remote = true, name="wikiSearch", startingInputTypes = {ResearchSubject.class})
+    )
+    ResearchReport performResearch(ResearchSubject researchSubject, FocusArea focusArea, OperationContext context) {
+        return context.promptRunner()
+                .withLlm(llm.withTemperature(.8))
+                .withToolGroup(CoreToolGroups.WEB)
+                .createObject("""
+                    Research the following subject on Wikipedia, with the given focus area, and summarize it
+                    in %d words:
+                    Subject: %s
+                    Focus Area: %s
+                    """.formatted(wordCount, researchSubject.name(), focusArea.topic()),
+                        ResearchReport.class);
+    }
+}