Function wrap_py_llm_sanitize_response_fn

crates/python/src/py_callable.rs:821–846

Wrap a Python callable `(dict) -> dict` for LLM sanitize response guardrails.

(py_fn: Py<PyAny>)

/// Wrap a Python callable `(dict) -> dict` for LLM sanitize response guardrails.
pub fn wrap_py_llm_sanitize_response_fn(py_fn: Py<PyAny>) -> LlmSanitizeResponseFn {
    Arc::new(move |response: Json| {
        Python::attach(|py| {
            let py_resp = match json_to_py(py, &response) {
                Ok(v) => v,
                Err(e) => {
                    eprintln!(
                        "nemo_relay: json_to_py failed in LLM sanitize response guardrail: {e}"
                    );
                    return response.clone();
                }
            };
            let result = match py_fn.call1(py, (py_resp,)) {
                Ok(v) => v,
                Err(e) => {
                    eprintln!("nemo_relay: LLM sanitize response guardrail callable failed: {e}");
                    return response.clone();
                }
            };
            py_to_json(result.bind(py)).unwrap_or_else(|e| {
                eprintln!("nemo_relay: py_to_json failed in LLM sanitize response guardrail: {e}");
                response.clone()
            })
        })
    })
}

/// Wrap a Python callable `(Event) -> None` for event subscribers.
pub fn wrap_py_event_subscriber(py_fn: Py<PyAny>) -> EventSubscriberFn {