Introduce PlannerFactory

Author: johnsonr

embabel-agent-api/src/main/kotlin/com/embabel/agent/core/support/ConcurrentAgentProcess.kt

@@ -18,6 +18,7 @@ package com.embabel.agent.core.support
 
 import com.embabel.agent.api.common.PlatformServices
 import com.embabel.agent.core.*
+import com.embabel.agent.spi.PlannerFactory
 import com.embabel.plan.WorldState
 import com.embabel.plan.goap.GoapWorldState
 import kotlinx.coroutines.future.await
@@ -37,6 +38,7 @@ open class ConcurrentAgentProcess(
     processOptions: ProcessOptions,
     blackboard: Blackboard,
     platformServices: PlatformServices,
+    plannerFactory: PlannerFactory,
     timestamp: Instant = Instant.now(),
     val callbacks: List<AgentProcessCallback> = emptyList(),
 ) : SimpleAgentProcess(
@@ -46,6 +48,7 @@ open class ConcurrentAgentProcess(
     processOptions = processOptions,
     blackboard = blackboard,
     platformServices = platformServices,
+    plannerFactory = plannerFactory,
     timestamp = timestamp,
 ) {
     override fun formulateAndExecutePlan(worldState: WorldState): AgentProcess {

embabel-agent-api/src/main/kotlin/com/embabel/agent/core/support/DefaultAgentPlatform.kt

@@ -17,19 +17,18 @@ package com.embabel.agent.core.support
 
 import com.embabel.agent.api.common.Asyncer
 import com.embabel.agent.channel.OutputChannel
-import com.embabel.agent.spi.config.spring.AgentPlatformProperties
 import com.embabel.agent.core.*
 import com.embabel.agent.event.AgentDeploymentEvent
 import com.embabel.agent.event.AgentProcessCreationEvent
 import com.embabel.agent.event.AgenticEventListener
 import com.embabel.agent.spi.*
+import com.embabel.agent.spi.config.spring.AgentPlatformProperties
+import com.embabel.agent.spi.support.GoapPlannerFactory
 import com.embabel.agent.spi.support.InMemoryAgentProcessRepository
 import com.embabel.agent.spi.support.InMemoryContextRepository
 import com.embabel.agent.spi.support.SpringContextPlatformServices
 import com.embabel.common.textio.template.TemplateRenderer
 import com.fasterxml.jackson.databind.ObjectMapper
-import com.fasterxml.jackson.dataformat.yaml.YAMLFactory
-import com.fasterxml.jackson.module.kotlin.registerKotlinModule
 import org.slf4j.LoggerFactory
 import org.springframework.beans.factory.annotation.Autowired
 import org.springframework.beans.factory.annotation.Value
@@ -64,7 +63,7 @@ open class DefaultAgentPlatform(
 
     private val logger = LoggerFactory.getLogger(DefaultAgentPlatform::class.java)
 
-    private val yamlObjectMapper = ObjectMapper(YAMLFactory()).registerKotlinModule()
+    private val plannerFactory: PlannerFactory = GoapPlannerFactory
 
     private val agents: MutableMap<String, Agent> = ConcurrentHashMap()
 
@@ -176,6 +175,7 @@ open class DefaultAgentPlatform(
                 id = id,
                 parentId = null,
                 processOptions = processOptions,
+                plannerFactory = plannerFactory,
             )
 
             AgentPlatformProperties.ProcessType.CONCURRENT -> ConcurrentAgentProcess(
@@ -185,6 +185,7 @@ open class DefaultAgentPlatform(
                 id = id,
                 parentId = null,
                 processOptions = processOptions,
+                plannerFactory = plannerFactory,
                 callbacks = callbacks,
             )
         }
@@ -212,6 +213,7 @@ open class DefaultAgentPlatform(
             }",
             parentId = parentAgentProcess.id,
             processOptions = processOptions,
+            plannerFactory = plannerFactory,
         )
         logger.debug("👶 Creating child process {} from {}", childAgentProcess.id, parentAgentProcess.id)
         agentProcessRepository.save(childAgentProcess)

embabel-agent-api/src/main/kotlin/com/embabel/agent/core/support/SimpleAgentProcess.kt

@@ -19,11 +19,11 @@ import com.embabel.agent.api.common.PlatformServices
 import com.embabel.agent.core.*
 import com.embabel.agent.event.AgentProcessPlanFormulatedEvent
 import com.embabel.agent.event.GoalAchievedEvent
+import com.embabel.agent.spi.PlannerFactory
 import com.embabel.common.util.indentLines
 import com.embabel.plan.Plan
 import com.embabel.plan.Planner
 import com.embabel.plan.WorldState
-import com.embabel.plan.goap.AStarGoapPlanner
 import com.embabel.plan.goap.WorldStateDeterminer
 import java.time.Instant
 
@@ -34,6 +34,7 @@ open class SimpleAgentProcess(
     processOptions: ProcessOptions,
     blackboard: Blackboard,
     platformServices: PlatformServices,
+    plannerFactory: PlannerFactory,
     timestamp: Instant = Instant.now(),
 ) : AbstractAgentProcess(
     id = id,
@@ -52,7 +53,7 @@ open class SimpleAgentProcess(
 
     override val worldStateDeterminer: WorldStateDeterminer = BlackboardWorldStateDeterminer(processContext)
 
-    override val planner: Planner<*, *, *> = AStarGoapPlanner(worldStateDeterminer)
+    override val planner: Planner<*, *, *> = plannerFactory.createPlanner(processOptions, worldStateDeterminer)
 
     override fun recordLlmInvocation(llmInvocation: LlmInvocation) {
         _llmInvocations.add(llmInvocation)

embabel-agent-api/src/main/kotlin/com/embabel/agent/spi/PlannerFactory.kt

@@ -0,0 +1,34 @@
+/*
+ * 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.agent.spi
+
+import com.embabel.agent.core.ProcessOptions
+import com.embabel.plan.Plan
+import com.embabel.plan.Planner
+import com.embabel.plan.PlanningSystem
+import com.embabel.plan.WorldState
+import com.embabel.plan.goap.WorldStateDeterminer
+
+/**
+ * Pluggable planner factory
+ */
+fun interface PlannerFactory {
+
+    fun createPlanner(
+        processOptions: ProcessOptions,
+        worldStateDeterminer: WorldStateDeterminer,
+    ): Planner<out PlanningSystem, out WorldState, out Plan>
+}

embabel-agent-api/src/main/kotlin/com/embabel/agent/spi/support/GoapPlannerFactory.kt

@@ -0,0 +1,37 @@
+/*
+ * 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.agent.spi.support
+
+import com.embabel.agent.core.ProcessOptions
+import com.embabel.agent.spi.PlannerFactory
+import com.embabel.plan.Plan
+import com.embabel.plan.Planner
+import com.embabel.plan.PlanningSystem
+import com.embabel.plan.WorldState
+import com.embabel.plan.goap.AStarGoapPlanner
+import com.embabel.plan.goap.WorldStateDeterminer
+
+/**
+ * PlannerFactory that always creates AStarGoapPlanner
+ */
+object GoapPlannerFactory : PlannerFactory {
+    override fun createPlanner(
+        processOptions: ProcessOptions,
+        worldStateDeterminer: WorldStateDeterminer,
+    ): Planner<out PlanningSystem, out WorldState, out Plan> {
+        return AStarGoapPlanner(worldStateDeterminer)
+    }
+}

embabel-agent-api/src/main/kotlin/com/embabel/agent/testing/integration/IntegrationTestUtils.kt

@@ -17,7 +17,6 @@ package com.embabel.agent.testing.integration
 
 import com.embabel.agent.api.common.PlatformServices
 import com.embabel.agent.channel.DevNullOutputChannel
-import com.embabel.agent.spi.config.spring.AgentPlatformProperties.ProcessType
 import com.embabel.agent.core.*
 import com.embabel.agent.core.support.DefaultAgentPlatform
 import com.embabel.agent.core.support.InMemoryBlackboard
@@ -26,7 +25,9 @@ import com.embabel.agent.event.AgenticEventListener
 import com.embabel.agent.spi.LlmOperations
 import com.embabel.agent.spi.OperationScheduler
 import com.embabel.agent.spi.ToolGroupResolver
+import com.embabel.agent.spi.config.spring.AgentPlatformProperties.ProcessType
 import com.embabel.agent.spi.support.ExecutorAsyncer
+import com.embabel.agent.spi.support.GoapPlannerFactory
 import com.embabel.agent.spi.support.RegistryToolGroupResolver
 import com.embabel.agent.spi.support.SpringContextPlatformServices
 import com.embabel.agent.testing.common.EventSavingAgenticEventListener
@@ -93,6 +94,7 @@ object IntegrationTestUtils {
             blackboard = InMemoryBlackboard(),
             processOptions = ProcessOptions(),
             platformServices = platformServices ?: dummyPlatformServices(),
+            plannerFactory = GoapPlannerFactory,
         )
     }
 

embabel-agent-api/src/test/java/com/embabel/agent/api/annotation/support/PackageVisibleTests.java

@@ -28,6 +28,7 @@
 import com.embabel.agent.core.ProcessOptions;
 import com.embabel.agent.core.support.InMemoryBlackboard;
 import com.embabel.agent.core.support.SimpleAgentProcess;
+import com.embabel.agent.spi.support.GoapPlannerFactory;
 import org.junit.jupiter.api.Test;
 
 import java.time.Instant;
@@ -77,6 +78,7 @@ public void actionInvocationWithJavaPackageVisibleParameters() {
                         processOptions,
                         new InMemoryBlackboard(),
                         platformServices,
+                        GoapPlannerFactory.INSTANCE,
                         Instant.now()
                 )
         );
@@ -109,6 +111,7 @@ public void conditionInvocationWithJavaPackageVisibleConditionMethod() {
                         processOptions,
                         new InMemoryBlackboard(),
                         platformServices,
+                        GoapPlannerFactory.INSTANCE,
                         Instant.now()
                 )
         );

embabel-agent-api/src/test/kotlin/com/embabel/agent/api/annotation/support/AgentMetadataReaderActionTest.kt

@@ -27,6 +27,7 @@ import com.embabel.agent.domain.io.UserInput
 import com.embabel.agent.event.AgenticEventListener.Companion.DevNull
 import com.embabel.agent.spi.LlmInteraction
 import com.embabel.agent.spi.LlmOperations
+import com.embabel.agent.spi.support.GoapPlannerFactory
 import com.embabel.agent.support.Dog
 import com.embabel.agent.testing.integration.IntegrationTestUtils
 import com.embabel.agent.testing.integration.IntegrationTestUtils.dummyAgentProcessRunning
@@ -312,6 +313,7 @@ class AgentMetadataReaderActionTest {
                     platformServices = platformServices,
                     processOptions = ProcessOptions(),
                     blackboard = InMemoryBlackboard(),
+                    plannerFactory = GoapPlannerFactory,
                     parentId = null,
                 ),
                 platformServices = platformServices,
@@ -436,6 +438,7 @@ class AgentMetadataReaderActionTest {
                     platformServices = platformServices,
                     processOptions = ProcessOptions(),
                     blackboard = InMemoryBlackboard(),
+                    plannerFactory = GoapPlannerFactory,
                     parentId = null,
                 ),
                 platformServices = platformServices,

embabel-agent-api/src/test/kotlin/com/embabel/agent/api/dsl/AgentBuilderTest.kt

@@ -22,6 +22,7 @@ import com.embabel.agent.core.ProcessContext
 import com.embabel.agent.core.ProcessOptions
 import com.embabel.agent.core.support.InMemoryBlackboard
 import com.embabel.agent.core.support.SimpleAgentProcess
+import com.embabel.agent.spi.support.GoapPlannerFactory
 import com.embabel.agent.support.Dog
 import com.embabel.agent.testing.integration.IntegrationTestUtils.dummyPlatformServices
 import com.embabel.common.core.types.Semver
@@ -109,6 +110,7 @@ class AgentBuilderTest {
                     processOptions = ProcessOptions(blackboard = blackboard),
                     platformServices = platformServices,
                     id = "test",
+                    plannerFactory = GoapPlannerFactory,
                     parentId = null,
                 ),
                 processOptions = ProcessOptions(blackboard = blackboard),
@@ -138,6 +140,7 @@ class AgentBuilderTest {
                     processOptions = ProcessOptions(blackboard = blackboard),
                     platformServices = platformServices,
                     id = "test",
+                    plannerFactory = GoapPlannerFactory,
                     parentId = null,
                 ),
                 processOptions = ProcessOptions(blackboard = blackboard),

embabel-agent-api/src/test/kotlin/com/embabel/agent/core/support/ConcurrentAgentProcessTest.kt

@@ -15,24 +15,18 @@
  */
 package com.embabel.agent.core.support
 
-import com.embabel.agent.api.annotation.AchievesGoal
-import com.embabel.agent.api.annotation.Action
-import com.embabel.agent.api.annotation.confirm
 import com.embabel.agent.api.annotation.support.AgentMetadataReader
-import com.embabel.agent.api.annotation.waitFor
 import com.embabel.agent.api.common.StuckHandler
 import com.embabel.agent.api.common.StuckHandlerResult
 import com.embabel.agent.api.common.StuckHandlingResultCode
 import com.embabel.agent.api.dsl.Frog
-import com.embabel.agent.api.dsl.agent
 import com.embabel.agent.api.dsl.evenMoreEvilWizard
 import com.embabel.agent.core.*
 import com.embabel.agent.core.hitl.ConfirmationRequest
 import com.embabel.agent.domain.io.UserInput
-import com.embabel.agent.domain.library.Person
 import com.embabel.agent.event.ObjectAddedEvent
 import com.embabel.agent.event.ObjectBoundEvent
-import com.embabel.agent.support.Dog
+import com.embabel.agent.spi.support.GoapPlannerFactory
 import com.embabel.agent.support.SimpleTestAgent
 import com.embabel.agent.testing.common.EventSavingAgenticEventListener
 import com.embabel.agent.testing.integration.IntegrationTestUtils.dummyPlatformServices
@@ -58,6 +52,7 @@ class ConcurrentAgentProcessTest {
                 blackboard = InMemoryBlackboard(),
                 platformServices = dummyPlatformServices(),
                 parentId = null,
+                plannerFactory = GoapPlannerFactory,
             )
             assertThrows<IOException> {
                 jacksonObjectMapper().writeValueAsString(cap)
@@ -99,6 +94,7 @@ class ConcurrentAgentProcessTest {
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
                 parentId = null,
+                plannerFactory = GoapPlannerFactory,
             )
             val agentStatus = agentProcess.tick()
             assertEquals(AgentProcessStatusCode.WAITING, agentStatus.status)
@@ -117,6 +113,7 @@ class ConcurrentAgentProcessTest {
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
                 parentId = null,
+                plannerFactory = GoapPlannerFactory,
             )
             val agentStatus = agentProcess.run()
             assertEquals(AgentProcessStatusCode.WAITING, agentStatus.status)
@@ -191,6 +188,7 @@ class ConcurrentAgentProcessTest {
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
                 parentId = null,
+                plannerFactory = GoapPlannerFactory,
             )
             return agentProcess.run()
         }
@@ -211,6 +209,7 @@ class ConcurrentAgentProcessTest {
                 processOptions = ProcessOptions(),
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
+                plannerFactory = GoapPlannerFactory,
                 parentId = null,
             )
             val person = LocalPerson("John")
@@ -229,6 +228,7 @@ class ConcurrentAgentProcessTest {
                 processOptions = ProcessOptions(),
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
+                plannerFactory = GoapPlannerFactory,
                 parentId = null,
             )
             val person = LocalPerson("John")
@@ -246,6 +246,7 @@ class ConcurrentAgentProcessTest {
                 processOptions = ProcessOptions(),
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
+                plannerFactory = GoapPlannerFactory,
                 parentId = null,
             )
             val person = LocalPerson("John")
@@ -264,6 +265,7 @@ class ConcurrentAgentProcessTest {
                 processOptions = ProcessOptions(),
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
+                plannerFactory = GoapPlannerFactory,
                 parentId = null,
             )
             val person = LocalPerson("John")
@@ -290,6 +292,7 @@ class ConcurrentAgentProcessTest {
                 processOptions = ProcessOptions(),
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
+                plannerFactory = GoapPlannerFactory,
                 parentId = null,
             )
             assertEquals(0, agentProcess.toolsStats.toolsStats.size, "No tools called yet")
@@ -310,6 +313,7 @@ class ConcurrentAgentProcessTest {
                 processOptions = ProcessOptions(),
                 blackboard = blackboard,
                 platformServices = dummyPlatformServices,
+                plannerFactory = GoapPlannerFactory,
                 parentId = null,
             )
             assertEquals(AgentProcessStatusCode.NOT_STARTED, agentProcess.status)