Function wrap_py_llm_sanitize_request_fn

crates/python/src/py_callable.rs:591–617

Wrap a Python callable `(LlmRequest) -> LlmRequest` for LLM sanitize request guardrails.

(py_fn: Py<PyAny>)

/// Wrap a Python callable `(LlmRequest) -> LlmRequest` for LLM sanitize request guardrails.
pub fn wrap_py_llm_sanitize_request_fn(py_fn: Py<PyAny>) -> LlmSanitizeRequestFn {
    Arc::new(move |request: LlmRequest| {
        Python::attach(|py| {
            let py_req = PyLLMRequest {
                inner: request.clone(),
            };
            let result = match py_fn.call1(py, (py_req,)) {
                Ok(v) => v,
                Err(e) => {
                    eprintln!("nemo_relay: LLM sanitize request guardrail callable failed: {e}");
                    return request;
                }
            };
            let extracted = result.extract::<PyLLMRequest>(py);
            match extracted {
                Ok(r) => r.inner,
                Err(e) => {
                    eprintln!(
                        "nemo_relay: LLM sanitize request guardrail returned unexpected type \
                         (expected LlmRequest): {e}"
                    );
                    request
                }
            }
        })
    })
}

/// Wrap a Python callable `(LlmRequest) -> Optional[str]` for LLM conditional guardrails.
pub fn wrap_py_llm_conditional_fn(py_fn: Py<PyAny>) -> LlmConditionalFn {