unique_toolkit 0.5.43__tar.gz → 0.5.44__tar.gz

{unique_toolkit-0.5.43 → unique_toolkit-0.5.44}/CHANGELOG.md

@@ -5,7 +5,8 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-
+## [0.5.44] - 2024-12-18
+- Add `event_constructor` to `verify_signature_and_construct_event` to allow for custom event construction.
 
 ## [0.5.43] - 2024-12-13
 - Add `Prompt` class to handle templated prompts that can be formatted into LanguageModelSystemMessage and LanguageModelUserMessage.

{unique_toolkit-0.5.43 → unique_toolkit-0.5.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.43
+Version: 0.5.44
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,7 +100,8 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-
+## [0.5.44] - 2024-12-18
+- Add `event_constructor` to `verify_signature_and_construct_event` to allow for custom event construction.
 
 ## [0.5.43] - 2024-12-13
 - Add `Prompt` class to handle templated prompts that can be formatted into LanguageModelSystemMessage and LanguageModelUserMessage.

{unique_toolkit-0.5.43 → unique_toolkit-0.5.44}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.5.43"
+version = "0.5.44"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.5.43 → unique_toolkit-0.5.44}/unique_toolkit/app/verification.py

@@ -1,4 +1,5 @@
 import logging
+from typing import Callable, TypeVar
 
 import unique_sdk
 
@@ -11,12 +12,16 @@ class WebhookVerificationError(Exception):
     pass
 
 
+T = TypeVar("T")
+
+
 def verify_signature_and_construct_event(
     headers: dict[str, str],
     payload: bytes,
     endpoint_secret: str,
     logger: logging.Logger = logging.getLogger(__name__),
-):
+    event_constructor: Callable[..., T] = Event,
+) -> T:
     """
     Verify the signature of a webhook and construct an event object.
 
@@ -25,18 +30,14 @@ def verify_signature_and_construct_event(
         payload (bytes): The raw payload of the webhook request.
         endpoint_secret (str): The secret used to verify the webhook signature.
         logger (logging.Logger): A logger instance for logging messages.
-
+        event_constructor (Callable[..., T]): A callable that constructs an event object.
     Returns:
-        Union[Event, Tuple[Dict[str, bool], int]]:
-            If successful, returns an Event object.
-            If unsuccessful, returns a tuple with an error response and HTTP status code.
+        T: The constructed event object.
 
     Raises:
         WebhookVerificationError: If there's an error during verification or event construction.
     """
 
-    # Only verify the event if there is an endpoint secret defined
-    # Otherwise use the basic event deserialized with json
     sig_header = headers.get("X-Unique-Signature")
     timestamp = headers.get("X-Unique-Created-At")
 
@@ -52,7 +53,7 @@ def verify_signature_and_construct_event(
             endpoint_secret,
         )
         logger.info("✅  Webhook signature verification successful.")
-        return Event(**event)
+        return event_constructor(**event)
     except unique_sdk.SignatureVerificationError as e:
         logger.error("⚠️  Webhook signature verification failed. " + str(e))
         raise WebhookVerificationError(f"Signature verification failed: {str(e)}")