def wrap_gen_fn()

in deepseek_vl2/serve/app_modules/gradio_utils.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def wrap_gen_fn(gen_fn):
    @wraps(gen_fn)
    def wrapped_gen_fn(prompt, *args, **kwargs):
        try:
            yield from gen_fn(prompt, *args, **kwargs)
        except gr.Error as g_err:
            raise g_err
        except Exception as e:
            raise gr.Error(f"Failed to generate text: {e}") from e

    return wrapped_gen_fn