diff --git a/forecasting_tools/agents_and_tools/question_generators/question_decomposer.py b/forecasting_tools/agents_and_tools/question_generators/question_decomposer.py index b4be4885..905c50ff 100644 --- a/forecasting_tools/agents_and_tools/question_generators/question_decomposer.py +++ b/forecasting_tools/agents_and_tools/question_generators/question_decomposer.py @@ -170,10 +170,12 @@ async def decompose_into_questions_fast( related_research: str | None, additional_context: str | None, number_of_questions: int = 5, - model: str | GeneralLlm = GeneralLlm.search_context_model( - "openrouter/perplexity/sonar" - ), + model: str | GeneralLlm | None = None, ) -> DecompositionResult: + if model is None: + model = GeneralLlm.search_context_model( + "openrouter/perplexity/sonar" + ) llm = GeneralLlm.to_llm(model) prompt = clean_indents(