From 4a52a71956a8d46fcb7294ac71734504bb09bcc2 Mon Sep 17 00:00:00 2001 From: S. Solomon Darnell Date: Fri, 28 Mar 2025 21:52:21 -0500 Subject: two version of R2R are here --- .../unstructured_client/_hooks/registration.py | 48 ++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 .venv/lib/python3.12/site-packages/unstructured_client/_hooks/registration.py (limited to '.venv/lib/python3.12/site-packages/unstructured_client/_hooks/registration.py') diff --git a/.venv/lib/python3.12/site-packages/unstructured_client/_hooks/registration.py b/.venv/lib/python3.12/site-packages/unstructured_client/_hooks/registration.py new file mode 100644 index 00000000..f2a7fc60 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/unstructured_client/_hooks/registration.py @@ -0,0 +1,48 @@ +"""Registration of custom, human-written hooks.""" + +from .custom import ( + CleanServerUrlSDKInitHook, + LoggerHook, + SuggestDefiningUrlIf401AfterErrorHook, + SplitPdfHook, +) +from .types import Hooks + + +# This file is only ever generated once on the first generation and then is free to be modified. +# Any hooks you wish to add should be registered in the init_hooks function. Feel free to define +# them in this file or in separate files in the hooks folder. + + +def init_hooks(hooks: Hooks): + # pylint: disable=unused-argument + """Add hooks by calling `hooks.register__hook` with an instance of that hook. + + Hooks are registered per SDK instance, and are valid for the lifetime of the SDK instance + """ + + # Initialize custom hooks + clean_server_url_hook = CleanServerUrlSDKInitHook() + suggest_defining_url_hook = SuggestDefiningUrlIf401AfterErrorHook() + logger_hook = LoggerHook() + split_pdf_hook = SplitPdfHook() + + # NOTE: logger_hook should stay registered last as logs the status of + # request and whether it will be retried which can be changed by e.g. split_pdf_hook + + # Register SDK Init hooks + hooks.register_sdk_init_hook(clean_server_url_hook) + hooks.register_sdk_init_hook(logger_hook) + hooks.register_sdk_init_hook(split_pdf_hook) + + # Register Before Request hooks + hooks.register_before_request_hook(split_pdf_hook) + + # Register After Error hooks + hooks.register_after_success_hook(split_pdf_hook) + hooks.register_after_success_hook(logger_hook) + + # Register After Error hooks + hooks.register_after_error_hook(suggest_defining_url_hook) + hooks.register_after_error_hook(split_pdf_hook) + hooks.register_after_error_hook(logger_hook) -- cgit v1.2.3