refactor: moves prompt_modules to cli scope

Signed-off-by: Tulio Coppola <[email protected]>

Author: Tulio Coppola

cli/decompose/pipeline.py

@@ -7,19 +7,16 @@
 from mellea.backends.ollama import OllamaModelBackend
 from mellea.backends.openai import OpenAIBackend
 from mellea.backends.types import ModelOption
-from mellea.helpers.prompt_modules.decompose import (
+
+from .prompt_modules import (
     constraint_extractor,
     subtask_constraint_assign,
     subtask_list,
     subtask_prompt_generator,
 )
-from mellea.helpers.prompt_modules.decompose.subtask_constraint_assign import (
-    SubtaskPromptConstraintsItem,
-)
-from mellea.helpers.prompt_modules.decompose.subtask_list import SubtaskItem
-from mellea.helpers.prompt_modules.decompose.subtask_prompt_generator import (
-    SubtaskPromptItem,
-)
+from .prompt_modules.subtask_constraint_assign import SubtaskPromptConstraintsItem
+from .prompt_modules.subtask_list import SubtaskItem
+from .prompt_modules.subtask_prompt_generator import SubtaskPromptItem
 
 
 class DecompSubtasksResult(TypedDict):

mellea/helpers/prompt_modules/decompose/__init__.py renamed to cli/decompose/prompt_modules/__init__.py

@@ -4,12 +4,9 @@
 
 from mellea import MelleaSession
 from mellea.backends.types import ModelOption
-from mellea.helpers.prompt_modules._prompt_modules import (
-    PromptModule,
-    PromptModuleString,
-)
 from mellea.stdlib.instruction import Instruction
 
+from .._prompt_modules import PromptModule, PromptModuleString
 from ._exceptions import BackendGenerationError, TagExtractionError
 from ._prompt import get_system_prompt, get_user_prompt