1 files changed, 9 insertions, 0 deletions
diff --git a/src/main/java/dev/dnpm/oshelper/atc/services/AgentCodeService.java b/src/main/java/dev/dnpm/oshelper/atc/services/AgentCodeService.java
index f1dc2e2..6540023 100644
--- a/src/main/java/dev/dnpm/oshelper/atc/services/AgentCodeService.java
+++ b/src/main/java/dev/dnpm/oshelper/atc/services/AgentCodeService.java
@@ -25,6 +25,7 @@ import dev.dnpm.oshelper.atc.AgentCode;
 import java.util.Comparator;
 import java.util.List;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import org.jspecify.annotations.NullMarked;
@@ -100,4 +101,12 @@ public interface AgentCodeService {
    * @return A list with agent codes
    */
   List<AgentCode> findAgentCodes(String query, int size);
+
+  /**
+   * Queries source for agents with code matching code string.
+   *
+   * @param code The code string
+   * @return An optional agent code if found
+   */
+  Optional<AgentCode> getAgentCode(String code);
 }