Updated Orchestrator Config

Author: raghav-2002-os

src/magentic_ui/teams/orchestrator/_orchestrator.py

@@ -185,8 +185,6 @@ def __init__(
         # Setup internal variables
         self._setup_internals()
 
-        self.enhanced_plan = True
-
     def _setup_internals(self) -> None:
         """
         Setup internal variables used in orchestrator
@@ -800,7 +798,7 @@ async def _orchestrate_step_planning(
                 await self._handle_relevant_plan_from_memory(context=context)
 
             # create a first plan
-            if self.enhanced_plan:
+            if self._config.enhanced_planning:
                 user_query = context[1].content
                 assert isinstance(user_query, str)
                 context.append(
@@ -881,7 +879,7 @@ async def _orchestrate_step_planning(
                 if self._config.retrieve_relevant_plans == "hint":
                     await self._handle_relevant_plan_from_memory(context=context)
 
-                if self.enhanced_plan:
+                if self._config.enhanced_planning:
                     user_query = context[1].content
                     assert isinstance(user_query, str)
                     context.append(

src/magentic_ui/teams/orchestrator/orchestrator_config.py

@@ -44,3 +44,4 @@ class OrchestratorConfig(BaseModel):
     max_replans: Union[int, None] = 3
     no_overwrite_of_task: bool = False
     sentinel_tasks: bool = False
+    enhanced_planning: bool = True