Function run_with_observability

quantmind/flows/_runner.py:17–59

Build `RunConfig` + composed hooks, run the agent, return final output. Args: agent: The Agents SDK ``Agent`` to invoke. input: Prompt string or pre-built input items. cfg: Flow configuration. Tracing fields and ``max_turns`` are forwarded to the SDK; ``workf

(
    agent: Agent[Any],
    input: str | list[Any],
    *,
    cfg: BaseFlowCfg,
    memory: object | None = None,
    extra_run_hooks: list[RunHooks[Any]],
)

async def run_with_observability(
    agent: Agent[Any],
    input: str | list[Any],
    *,
    cfg: BaseFlowCfg,
    memory: object | None = None,
    extra_run_hooks: list[RunHooks[Any]],
) -> Any:
    """Build `RunConfig` + composed hooks, run the agent, return final output.

    Args:
        agent: The Agents SDK ``Agent`` to invoke.
        input: Prompt string or pre-built input items.
        cfg: Flow configuration. Tracing fields and ``max_turns`` are
            forwarded to the SDK; ``workflow_name`` falls back to
            ``"quantmind.<agent.name>"`` when unset.
        memory: PR6 ``Memory`` placeholder. Currently unused at runtime;
            the value is forwarded to the trajectory-archive stub so PR6
            can wire it in without changing call sites.
        extra_run_hooks: User-supplied hooks. Composed with any
            memory-derived hooks (none in PR5) into a single
            ``RunHooks`` instance.

    Returns:
        ``RunResult.final_output`` typed by the agent&#x27;s ``output_type``.
    """
    workflow_name = cfg.workflow_name or f"quantmind.{agent.name}"
    run_cfg = RunConfig(
        workflow_name=workflow_name,
        trace_metadata=cfg.trace_metadata,
        trace_include_sensitive_data=cfg.trace_include_sensitive_data,
        tracing_disabled=cfg.tracing_disabled,
    )
    hooks = _compose_hooks(_collect_hooks(memory, extra_run_hooks))
    result = await Runner.run(
        agent,
        input,
        run_config=run_cfg,
        hooks=hooks,
        max_turns=cfg.max_turns,
    )
    _archive_run_artifacts(cfg, memory, result)
    return result.final_output


def _collect_hooks(