add openssa.l2.task.abstract.AbstractTask.decompose(...)

aitomatic · Apr 23, 2024 · 22d8da0 · 22d8da0
1 parent 790d50b
commit 22d8da0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/openssa/l2/task/abstract.py b/openssa/l2/task/abstract.py
@@ -12,6 +12,7 @@
 from .status import TaskStatus
 
 if TYPE_CHECKING:
+    from openssa.l2.planning.abstract.plan import APlan
     from openssa.l2.planning.abstract.planner import APlanner
     from openssa.l2.resource.abstract import AResource
 
@@ -69,5 +70,10 @@ def from_dict_or_str(cls, dict_or_str: TaskDict | str, /) -> Self:
 
         raise TypeError(f'*** {dict_or_str} IS NEITHER A DICTIONARY NOR A STRING ***')
 
+    def decompose(self) -> APlan:
+        """Decompose task into modular plan."""
+        assert self.dynamic_decomposer, '*** MISSING DYNAMIC DECOMPOSER ***'
+        return self.dynamic_decomposer.plan(problem=self.ask, resources=self.resources)
+
 
 ATask: TypeVar = TypeVar('ATask', bound=AbstractTask, covariant=False, contravariant=False)