Class RunWithObservabilityTests

tests/flows/test_runner.py:123–199

class RunWithObservabilityTests(unittest.IsolatedAsyncioTestCase):
    async def test_run_config_built_from_cfg(self) -> None:
        cfg = PaperSemanticCfg(
            model="gpt-test",
            max_turns=7,
            workflow_name="custom-name",
            trace_metadata={"k": "v"},
            trace_include_sensitive_data=False,
            tracing_disabled=True,
        )
        agent = MagicMock(name="agent")
        agent.name = "paper_extractor"
        fake_result = MagicMock()
        fake_result.final_output = "OUT"
        with patch(
            "quantmind.flows._runner.Runner.run",
            new=AsyncMock(return_value=fake_result),
        ) as run_mock:
            out = await run_with_observability(
                agent,
                "prompt",
                cfg=cfg,
                memory=None,
                extra_run_hooks=[],
            )
        self.assertEqual(out, "OUT")
        run_mock.assert_awaited_once()
        call = run_mock.await_args
        self.assertIs(call.args[0], agent)
        self.assertEqual(call.args[1], "prompt")
        self.assertEqual(call.kwargs["max_turns"], 7)
        run_cfg = call.kwargs["run_config"]
        self.assertEqual(run_cfg.workflow_name, "custom-name")
        self.assertEqual(run_cfg.trace_metadata, {"k": "v"})
        self.assertFalse(run_cfg.trace_include_sensitive_data)
        self.assertTrue(run_cfg.tracing_disabled)
        # No hooks supplied -> Runner.run sees None.
        self.assertIsNone(call.kwargs["hooks"])

    async def test_workflow_name_falls_back_to_agent_name(self) -> None:
        cfg = PaperSemanticCfg()  # workflow_name = None
        agent = MagicMock()
        agent.name = "paper_extractor"
        fake_result = MagicMock()
        fake_result.final_output = None
        with patch(
            "quantmind.flows._runner.Runner.run",
            new=AsyncMock(return_value=fake_result),
        ) as run_mock:
            await run_with_observability(
                agent, "x", cfg=cfg, memory=None, extra_run_hooks=[]
            )
        self.assertEqual(
            run_mock.await_args.kwargs["run_config"].workflow_name,
            "quantmind.paper_extractor",
        )

    async def test_extra_hooks_forwarded(self) -> None: