Function wrap_py_tool_conditional_fn

crates/python/src/py_callable.rs:212–237

Wrap a Python callable `(str, Json) -> Optional[str]` for tool conditional guardrails.

(py_fn: Py<PyAny>)

/// Wrap a Python callable `(str, Json) -> Optional[str]` for tool conditional guardrails.
pub fn wrap_py_tool_conditional_fn(py_fn: Py<PyAny>) -> ToolConditionalFn {
    Arc::new(move |name: &str, args: &Json| {
        Python::attach(|py| {
            let py_args = json_to_py(py, args).map_err(|e| {
                FlowError::Internal(format!(
                    "tool conditional json_to_py failed for '{name}': {e}"
                ))
            })?;
            let result = py_fn.call1(py, (name, py_args)).map_err(|e| {
                FlowError::Internal(format!(
                    "Python tool conditional callable failed for '{name}': {e}"
                ))
            })?;
            let bound = result.bind(py);
            if bound.is_none() {
                Ok(None)
            } else {
                bound.extract::<String>().map(Some).map_err(|e| {
                    FlowError::Internal(format!(
                        "tool conditional guardrail for '{name}' returned unexpected type (expected str or None): {e}"
                    ))
                })
            }
        })
    })
}

/// Wrap a Python callable `(str, Json) -> Json` for tool request intercepts.
pub fn wrap_py_tool_request_intercept_fn(py_fn: Py<PyAny>) -> ToolInterceptFn {