Function wrap_py_finalizer_fn

crates/python/src/py_callable.rs:802–818

Wrap a Python callable `() -> Any` as a finalizer for streaming LLM calls. The finalizer is called once when the stream is fully consumed. Its return value is converted from a Python object to `serde_json::Value` (Json) and used as the aggregated response.

(py_fn: Py<PyAny>)

/// value is converted from a Python object to `serde_json::Value` (Json) and
/// used as the aggregated response.
pub fn wrap_py_finalizer_fn(py_fn: Py<PyAny>) -> Box<dyn FnOnce() -> Json + Send> {
    Box::new(move || {
        Python::attach(|py| {
            let result = match py_fn.call0(py) {
                Ok(v) => v,
                Err(e) => {
                    eprintln!("nemo_relay: Python finalizer callable failed: {e}");
                    return Json::Null;
                }
            };
            py_to_json(result.bind(py)).unwrap_or_else(|e| {
                eprintln!("nemo_relay: py_to_json failed in finalizer: {e}");
                Json::Null
            })
        })
    })
}

/// Wrap a Python callable `(dict) -> dict` for LLM sanitize response guardrails.
pub fn wrap_py_llm_sanitize_response_fn(py_fn: Py<PyAny>) -> LlmSanitizeResponseFn {