Function wrap_py_llm_exec_fn

crates/python/src/py_callable.rs:719–737

Wrap a Python callable `(LlmRequest) -> dict` for LLM execution. Supports both sync and async Python callables.

(
    py_fn: Py<PyAny>,
)

/// Wrap a Python callable `(LlmRequest) -> dict` for LLM execution.
/// Supports both sync and async Python callables.
pub fn wrap_py_llm_exec_fn(
    py_fn: Py<PyAny>,
) -> Box<dyn Fn(LlmRequest) -> Pin<Box<dyn Future<Output = FlowResult<Json>> + Send>> + Send + Sync>
{
    let py_fn = std::sync::Arc::new(py_fn);
    Box::new(move |request: LlmRequest| {
        let py_fn = py_fn.clone();
        Box::pin(async move {
            resolve_json_or_future(Python::attach(|py| {
                let py_req = PyLLMRequest { inner: request };
                let result = py_fn
                    .call1(py, (py_req,))
                    .map_err(|e: PyErr| FlowError::Internal(e.to_string()))?;
                split_json_or_future(py, result)
            }))
            .await
        })
    })
}

/// Wrap a Python async generator `(LlmRequest) -> AsyncIterator[Any]` for LLM
/// stream execution.