First version

Author: johnsonr

README.md

@@ -16,76 +16,4 @@
 
     
 
-# Generated agent
-
-Java agent repository generated by the Embabel project creator.
-
-Starting point for your own agent development using the [Embabel framework](https://github.com/embabel/embabel-agent).
-
-Add your magic here!
-
-Illustrates:
-
-- An injected demo showing how any Spring component can be injected with an Embabel `Ai` instance to enable it to
-  perform LLM operations.
-- A simple agent
-- Unit tests for an agent verifying prompts and hyperparameters
-
-# To run
-
-Run the shell script to start Embabel under Spring Shell:
-
-```bash
-./scripts/shell.sh
-```
-
-There is a single example agent, `WriteAndReviewAgent`.
-It uses one LLM with a high temperature and creative persona to write a story based on your input,
-then another LLM with a low temperature and different persona to review the story.
-
-When the Embabel shell comes up, use the story agent like this:
-
-```
-x "Tell me a story about...[your topic]"
-```
-
-Try the `InjectedDemo` command to see simple, non-agent use:
-
-```java
-animal
-```
-
-## A2A Support
-
-Embabel integrates with the [A2A](https://github.com/google-a2a/A2A) protocol, allowing you to connect to other
-A2A-enabled agents and
-services.
-
-> Embabel agents can be exposed to A2A with zero developer effort.
-
-Check out the `a2a` branch of this repository to try A2A support.
-
-You'll need the following environment variable:
-
-- `GOOGLE_STUDIO_API_KEY`: Your Google Studio API key, which is used for Gemini.
-
-Start the Google A2A web interface using Docker compose:
-
-```bash
-docker compose up
-```
-
-Go to the web interface running within the container at `http://localhost:12000/`.
-
-Connect to your agent at `host.docker.internal:8080/a2a`. Note that `localhost:8080/a2a` won't work as the server
-cannot access it when running in a Docker container.
-
-Your agent will have automatically been exported to A2A. Add it in the UI, and start a chat.
-You should see something like this:
-
-<img src="images/a2a_ui.jpg" alt="A2A UI" width="600">
-
-## Contributors
-
-[![Embabel contributors](https://contrib.rocks/image?repo=embabel/java-agent-template)](https://github.com/embabel/java-agent-template/graphs/contributors)
-
+# Talk to the Docs

pom.xml

@@ -9,16 +9,16 @@
         <version>3.5.3</version>
         <relativePath/> <!-- Lookup parent from repository -->
     </parent>
-    <groupId>com.embabel.template</groupId>
-    <artifactId>ProjectName</artifactId>
+    <groupId>com.embabel.agent</groupId>
+    <artifactId>guide</artifactId>
     <version>0.1.0-SNAPSHOT</version>
     <packaging>jar</packaging>
-    <name>My first agent</name>
-    <description>Generated agent project</description>
+    <name>Talk to the docs</name>
+    <description>Embabel talk to the docs</description>
 
     <properties>
         <java.version>21</java.version>
-        <embabel-agent.version>0.1.1</embabel-agent.version>
+        <embabel-agent.version>0.1.3-SNAPSHOT</embabel-agent.version>
     </properties>
 
     <dependencies>
@@ -30,6 +30,18 @@
             <version>${embabel-agent.version}</version>
         </dependency>
 
+        <dependency>
+            <groupId>com.embabel.agent</groupId>
+            <artifactId>embabel-agent-starter-shell</artifactId>
+            <version>${embabel-agent.version}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>com.embabel.agent</groupId>
+            <artifactId>embabel-agent-rag-lucene</artifactId>
+            <version>${embabel-agent.version}</version>
+        </dependency>
+
         <dependency>
             <groupId>com.embabel.agent</groupId>
             <artifactId>embabel-agent-test</artifactId>

…/com/embabel/ProjectNameApplication.java …n/java/com/embabel/GuideApplication.javasrc/main/java/com/embabel/ProjectNameApplication.java renamed to src/main/java/com/embabel/GuideApplication.java src/main/java/com/embabel/ProjectNameApplication.java renamed to src/main/java/com/embabel/GuideApplication.java

@@ -24,9 +24,9 @@
 
 @SpringBootApplication
 @EnableAgentShell
-@EnableAgents(loggingTheme = LoggingThemes.STAR_WARS)
-class ProjectNameApplication {
+@EnableAgents(loggingTheme = LoggingThemes.SEVERANCE)
+class GuideApplication {
     public static void main(String[] args) {
-        SpringApplication.run(ProjectNameApplication.class, args);
+        SpringApplication.run(GuideApplication.class, args);
     }
 }

src/main/java/com/embabel/guide/ConversationManager.java

@@ -0,0 +1,30 @@
+package com.embabel.guide;
+
+import com.embabel.agent.api.common.Ai;
+import com.embabel.agent.rag.Ingester;
+import com.embabel.chat.ChatSession;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Service;
+
+@Service
+public class ConversationManager {
+
+    private final Ai ai;
+    private final Ingester ingester;
+
+    private final Logger logger = LoggerFactory.getLogger(ConversationManager.class);
+
+    public ConversationManager(
+            Ai ai,
+            Ingester ingester) {
+        this.ingester = ingester;
+        this.ai = ai;
+        var ingestionResult = ingester.ingest("data/index.md");
+        logger.info("Ingestion result: {}", ingestionResult);
+    }
+    
+    public ChatSession guide() {
+        return new Guide(ai);
+    }
+}

src/main/java/com/embabel/guide/Guide.java

@@ -0,0 +1,38 @@
+package com.embabel.guide;
+
+import com.embabel.agent.api.common.Ai;
+import com.embabel.agent.rag.tools.RagOptions;
+import com.embabel.chat.*;
+import org.jetbrains.annotations.NotNull;
+
+/**
+ * GuideLoader will have loaded content
+ */
+public class Guide implements ChatSession {
+
+    private final Ai ai;
+
+    private Conversation conversation = new InMemoryConversation();
+
+    public Guide(Ai ai) {
+        this.ai = ai;
+    }
+
+    @NotNull
+    @Override
+    public Conversation getConversation() {
+        return conversation;
+    }
+
+    @Override
+    public void respond(@NotNull UserMessage userMessage, @NotNull MessageListener messageListener) {
+        conversation = conversation.withMessage(userMessage);
+        final var assistantMessage = ai
+                .withLlmByRole("docs")
+                .withRagTools(new RagOptions().withSimilarityThreshold(.0).withTopK(8))
+                .withTemplate("guide_system")
+                .respondWithSystemPrompt(conversation);
+        conversation = conversation.withMessage(assistantMessage);
+        messageListener.onMessage(assistantMessage);
+    }
+}

src/main/java/com/embabel/guide/GuideShell.java

@@ -0,0 +1,29 @@
+package com.embabel.guide;
+
+import com.embabel.agent.event.logging.personality.severance.LumonColorPalette;
+import com.embabel.agent.shell.TerminalServices;
+import org.springframework.shell.standard.ShellComponent;
+import org.springframework.shell.standard.ShellMethod;
+
+@ShellComponent
+public class GuideShell {
+
+    private final TerminalServices terminalServices;
+    private final ConversationManager conversationManager;
+
+    public GuideShell(
+            ConversationManager conversationManager,
+            TerminalServices terminalServices) {
+        this.conversationManager = conversationManager;
+        this.terminalServices = terminalServices;
+    }
+
+    @ShellMethod("talk to docs")
+    public String talk() {
+        var guide = conversationManager.guide();
+        return terminalServices.chat(
+                guide,
+                "Welcome to the Guide! How can I assist you today?",
+                LumonColorPalette.INSTANCE);
+    }
+}

src/main/java/com/embabel/guide/RagConfig.java

@@ -0,0 +1,18 @@
+package com.embabel.guide;
+
+import com.embabel.agent.rag.RagService;
+import com.embabel.agent.rag.lucene.LuceneRagService;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+class RagConfig {
+
+    @Bean
+    RagService ragService() {
+        return new LuceneRagService(
+                "docs",
+                "data/index.md"
+        );
+    }
+}