diff --git a/cli/decompose/pipeline.py b/cli/decompose/pipeline.py index a574d35d6..a9b169392 100644 --- a/cli/decompose/pipeline.py +++ b/cli/decompose/pipeline.py @@ -116,7 +116,7 @@ def decompose( ).parse() constraint_validation_strategies: dict[str, Literal["code", "llm"]] = { - cons_key: validation_decision.generate(m_session, cons_key).parse() + cons_key: validation_decision.generate(m_session, cons_key).parse() or "llm" for cons_key in task_prompt_constraints } @@ -142,7 +142,9 @@ def decompose( constraints=[ { "constraint": cons_str, - "validation_strategy": constraint_validation_strategies[cons_str], + "validation_strategy": constraint_validation_strategies.get( + cons_str, "llm" + ), } for cons_str in subtask_data.constraints ], diff --git a/docs/examples/m_decompose/python/python_decompose_example.py b/docs/examples/m_decompose/python/python_decompose_example.py index a1b951cc3..21faf6444 100644 --- a/docs/examples/m_decompose/python/python_decompose_example.py +++ b/docs/examples/m_decompose/python/python_decompose_example.py @@ -32,7 +32,7 @@ def run_decompose(task_prompt: str) -> DecompPipelineResult: result = decompose( task_prompt=task_prompt, - model_id="granite4:micro", + model_id="granite3.3:8b", # Note micro will not properly create tags, need 8b backend=DecompBackend.ollama, # Use Ollama backend backend_req_timeout=300, # 5 minute timeout ) @@ -84,7 +84,9 @@ def generate_python_script( print("\nšŸ“ Generating executable Python script...") # Load the template from the CLI decompose directory - cli_decompose_dir = Path(__file__).parent.parent.parent.parent / "cli" / "decompose" + cli_decompose_dir = ( + Path(__file__).parent.parent.parent.parent.parent / "cli" / "decompose" + ) environment = Environment( loader=FileSystemLoader(cli_decompose_dir), autoescape=False )