Apply override only to local generation, do not propagate overridden …

…Runner to child components

sammo/components.py

@@ -87,9 +87,6 @@ async def _call(
         dynamic_context: frozendict | None,
         priority: int = 0,
     ) -> LLMResult:
-        if self._override_runner is not None:
-            runner = self._override_runner
-
         y = await self._child(runner, context, dynamic_context)
         parents = [y]
         if self._history:
@@ -98,8 +95,14 @@ async def _call(
             history = previous_turn.history
         else:
             history = None
+
         try:
-            result = await runner.generate_text(
+            if self._override_runner is not None:
+                runner_for_generation = self._override_runner
+            else:
+                runner_for_generation = runner
+
+            result = await runner_for_generation.generate_text(
                 y.value,
                 priority=priority,
                 system_prompt=self._system_prompt,

sammo/components_test.py

@@ -35,3 +35,13 @@ async def test_override_runner():
     res = await res2(runner2, dict())
     assert runner2.prompt_log[0] == "I got test1"
     assert res.value == "test2"
+
+
+@pytest.mark.asyncio
+async def test_child_runner_not_overridden():
+    runner1 = MockedRunner("test1")
+    runner2 = MockedRunner("test2")
+    res2 = GenerateText(Template("I got {{res1}}", res1=GenerateText("Get test1")), runner=runner2)
+    res = await res2(runner1, dict())
+    assert runner2.prompt_log[0] == "I got test1"
+    assert res.value == "test2"