moves decompose modules to appropriate path

Author: Tulio Coppola

cli/decompose/decompose.py

@@ -51,9 +51,9 @@ def run(
     backend_req_timeout: Annotated[
         int,
         typer.Option(
-            help='Time (in seconds) for timeout to be passed on the model inference requests. Defaults to "3600"'
+            help='Time (in seconds) for timeout to be passed on the model inference requests. Defaults to "300"'
         ),
-    ] = 3600,
+    ] = 300,
     backend_endpoint: Annotated[
         str | None,
         typer.Option(

cli/decompose/pipeline.py

@@ -7,17 +7,19 @@
 from mellea.backends.ollama import OllamaModelBackend
 from mellea.backends.openai import OpenAIBackend
 from mellea.backends.types import ModelOption
-from mellea.helpers.prompt_modules import (
+from mellea.helpers.prompt_modules.decompose import (
     constraint_extractor,
     subtask_constraint_assign,
     subtask_list,
     subtask_prompt_generator,
 )
-from mellea.helpers.prompt_modules.subtask_constraint_assign import (
+from mellea.helpers.prompt_modules.decompose.subtask_constraint_assign import (
     SubtaskPromptConstraintsItem,
 )
-from mellea.helpers.prompt_modules.subtask_list import SubtaskItem
-from mellea.helpers.prompt_modules.subtask_prompt_generator import SubtaskPromptItem
+from mellea.helpers.prompt_modules.decompose.subtask_list import SubtaskItem
+from mellea.helpers.prompt_modules.decompose.subtask_prompt_generator import (
+    SubtaskPromptItem,
+)
 
 
 class DecompSubtasksResult(TypedDict):
@@ -47,7 +49,7 @@ def decompose(
     user_input_variable: list[str] | None = None,
     model_id: str = "mistral-small3.2:latest",
     backend: DecompBackend = DecompBackend.ollama,
-    backend_req_timeout: int = 3600,
+    backend_req_timeout: int = 300,
     backend_endpoint: str | None = None,
     backend_api_key: str | None = None,
 ) -> DecompPipelineResult:

mellea/helpers/prompt_modules/__init__.py

@@ -1,8 +0,0 @@
-from .constraint_extractor import constraint_extractor as constraint_extractor
-from .subtask_constraint_assign import (
-    subtask_constraint_assign as subtask_constraint_assign,
-)
-from .subtask_list import subtask_list as subtask_list
-from .subtask_prompt_generator import (
-    subtask_prompt_generator as subtask_prompt_generator,
-)

mellea/helpers/prompt_modules/decompose/__init__.py

@@ -0,0 +1,8 @@
+from .constraint_extractor import constraint_extractor as constraint_extractor
+from .subtask_constraint_assign import (
+    subtask_constraint_assign as subtask_constraint_assign,
+)
+from .subtask_list import subtask_list as subtask_list
+from .subtask_prompt_generator import (
+    subtask_prompt_generator as subtask_prompt_generator,
+)