Tool group selection

Author: johnsonr

pom.xml

@@ -36,6 +36,12 @@
             <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>

src/main/java/com/embabel/coding/ReferenceConfiguration.java

@@ -0,0 +1,18 @@
+package com.embabel.coding;
+
+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
+public class ReferenceConfiguration {
+
+    @Bean
+    public RagService ragService() {
+        return new LuceneRagService(
+                "code ref",
+                "code description"
+        );
+    }
+}

src/main/java/com/embabel/coding/Tyrell.java

@@ -1,11 +1,10 @@
 package com.embabel.coding;
 
 import com.embabel.agent.api.annotation.*;
+import com.embabel.agent.api.common.LlmReference;
 import com.embabel.agent.api.common.OperationContext;
-import com.embabel.agent.api.common.ToolObject;
 import com.embabel.agent.domain.library.code.SoftwareProject;
 import com.embabel.agent.spi.ToolGroupResolver;
-import com.embabel.agent.tools.common.LlmReference;
 import com.embabel.coding.tools.api.ApiReference;
 import com.embabel.coding.tools.git.RepositoryReferenceProvider;
 import com.embabel.coding.tools.jvm.ClassGraphApiReferenceExtractor;
@@ -42,6 +41,7 @@ public Tyrell(Config config, ToolGroupResolver toolGroupResolver) {
                 .cloneRepository("https://github.com/embabel/embabel-agent-examples.git");
         references.add(embabelApiReference);
         references.add(examplesReference);
+        references.add(softwareProject);
     }
 
 
@@ -77,16 +77,12 @@ AgentCreationRequest requestAgentDetails() {
     public AgentCreationResult createAgent(AgentCreationRequest request, OperationContext embabel) {
         return embabel.ai()
                 .withLlm(config.codingLlm)
-                .withPromptElements(references)
-                // TODO need withToolObjects method in API
-                .withToolObject(new ToolObject(references.get(0)).withNamingStrategy(n -> "api_ref_" + n))
-                .withToolObject(new ToolObject(references.get(1)).withNamingStrategy(n -> "examples_repo_" + n))
-                .withToolObject(new ToolObject(softwareProject).withNamingStrategy(n -> "this_project_" + n))
-                .withPromptElements(softwareProject)
+                .withReferences(references)
                 .withTemplate("coding/creator")
                 .createObject(
                         AgentCreationResult.class,
                         Map.of(
+                                "thisProject", softwareProject.getName().replace("-", "_"),
                                 "package", request.pkg(),
                                 "purpose", request.purpose(),
                                 "toolGroups", toolGroupResolver.availableToolGroups()

src/main/java/com/embabel/joke/agent/JokeAgent.java

@@ -9,4 +9,5 @@
 #embabel.models.embeddingServices.cheapest=nomic-embed-text:latest
 #
 #embabel.agent-platform.ranking.llm=llama3.1:8b
-embabel.tyrell.coding-llm.model=claude-opus-4-20250514
+embabel.tyrell.coding-llm.model=claude-3-7-sonnet-latest
+#embabel.tyrell.coding-llm.model=claude-3-5-haiku-latest

src/main/resources/application.properties

@@ -1,30 +1,28 @@
 Create a new Embabel agent in Java within this project.
 This project uses normal Maven conventions.
 
-The package should be {{ pgk }}
-
-The purpose of the agent is {{ purpose }}.
-
-Choose an appropriate class name. Refer to API documentation and examples for how to build an agent.
+Choose an appropriate class name within the package.
+Refer to API documentation and examples for how to build an agent.
 
 Start by considering the data flow between the agent's @Action methods.
 These types should be captured in records.
 
 For each @Action method, consider what tool groups may be required.
 Be parsimonious in your choice of tool groups, including only what's necessary.
-The available tool groups are:
-
-{% for tool_group in tool_groups %}
-    - {{ tool_group.role }}: {{ tool_group.description }}
+The available tool groups are the following (use the role string in code for withToolGroup):
+Role | Description
+---- | -----------
+{% for group in toolGroups %}
+    {{ group.role }} | {{ group.description }}
 {% endfor %}
 
 Ue modern Java with vars and records.
 
 Use the build tool to check that your build works and fix it if it doesn't.
 
-You should also create the following artifacts:
-
-1. Unit test for the new agent. Use WriteAndReviewAgentTest in this repository as a guide
+{#You should also create the following artifacts:#}
+{##}
+{#1. Unit test for the new agent. Use WriteAndReviewAgentTest in this repository as a guide#}
 
 {#2. Integration test for the new agent. Use#}
 
@@ -33,7 +31,18 @@ DO NOT MODIFY ANY FILES BESIDES THE NEW FILES YOU'VE CREATED
 DO NOT MODIFY THE pom.xml
 
 Tools provided will give you access to reference repositories.
-The tool names beginning with "this_project_" will give you access to this repository
+The tool names beginning with "{{ thisProject }}" will give you access to this repository
+
+This project uses normal Maven project structure, you will add source code under src/main/java and tests under src/test/java.
+Note that the examples repo has a different structure.
 
 Create the simplest possible implementation that meets the requirements.
-Do not invent extra steps.
+Do not invent extra steps or actions.
+
+# PACKAGE
+
+The package should be {{ package }}
+
+# PURPOSE OF THE AGENT
+
+{{ purpose }}