unique_toolkit 0.5.42__py3-none-any.whl → 0.5.44__py3-none-any.whl

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)}")

unique_toolkit/language_model/__init__.py

@@ -1,5 +1,8 @@
 from .infos import LanguageModel as LanguageModel
 from .infos import LanguageModelName as LanguageModelName
+from .prompt import (
+    Prompt as Prompt,
+)
 from .schemas import (
     LanguageModelAssistantMessage as LanguageModelAssistantMessage,
 )

unique_toolkit/language_model/prompt.py

@@ -0,0 +1,124 @@
+from string import Template
+from typing import Any
+
+from unique_toolkit.language_model.schemas import (
+    LanguageModelSystemMessage,
+    LanguageModelUserMessage,
+)
+
+
+class Prompt:
+    """
+    A class for handling templated prompts that can be formatted into LanguageModelSystemMessage and LanguageModelUserMessage.
+
+    This class wraps a string template and provides methods to format it with variables and
+    convert it into different types of language model messages. It uses Python's string.Template
+    for variable substitution.
+
+    Usage:
+        # Create a prompt with a template and set variables
+        prompt = Prompt("Hello, ${name}!", name="World")
+
+        # Substitute the template with variables and return the formatted content
+        content = prompt.substitute(name="World")
+
+        # Get the formatted content
+        content = prompt.content  # Returns "Hello, World!"
+
+        # Get the template
+        template = prompt.template
+
+        # Convert to language model messages
+        system_msg = prompt.to_system_msg()
+        user_msg = prompt.to_user_msg()
+        user_msg_with_images = prompt.to_user_msg_with_images(images=["IMAGE_IN_BASE64"])
+
+    Properties:
+        template: Returns the underlying template string
+        content: Returns the current formatted content string
+
+    Methods:
+        substitute(**kwargs): Substitutes the template with the given variables
+        to_user_msg(): Converts the prompt to a LanguageModelUserMessage
+        to_user_msg_with_images(images): Converts the prompt to a LanguageModelUserMessage with images
+    """
+
+    def __init__(self, template: str, **kwargs):
+        self._template = Template(template)
+        self._content = self._template.template
+        if kwargs:
+            self._content = self._template.substitute(**kwargs)
+
+    @property
+    def template(self):
+        """
+        Returns the template string.
+
+        Returns:
+            str: The template string.
+        """
+        return self._template
+
+    @property
+    def content(self):
+        """
+        Returns the formatted content string.
+
+        Returns:
+            str: The formatted content string.
+        """
+        return self._content
+
+    def substitute(self, **kwargs: Any) -> str:
+        """
+        Substitutes the template with the given kwargs. Raises KeyError if a required parameter is missing.
+
+        Args:
+            **kwargs: Keyword arguments to substitute into the template.
+
+        Returns:
+            str: The substituted template string.
+
+        Raises:
+            KeyError: If a required parameter in the template is missing from kwargs.
+        """
+        self._content = self._template.substitute(**kwargs)
+        return self._content
+
+    def to_system_msg(self) -> LanguageModelSystemMessage:
+        """
+        Returns a LanguageModelSystemMessage with the content of the prompt.
+
+        Returns:
+            LanguageModelSystemMessage: The formatted prompt.
+        """
+        return LanguageModelSystemMessage(content=self._content)
+
+    def to_user_msg(self) -> LanguageModelUserMessage:
+        """
+        Returns a LanguageModelUserMessage with the content of the prompt.
+
+        Returns:
+            LanguageModelUserMessage: The formatted prompt.
+        """
+        return LanguageModelUserMessage(content=self._content)
+
+    def to_user_msg_with_images(self, images: list[str]) -> LanguageModelUserMessage:
+        """
+        Returns a LanguageModelUserMessage with the content of the prompt and the images.
+
+        Args:
+            images: List of images in base64 format.
+
+        Returns:
+            LanguageModelUserMessage: The formatted prompt with images.
+        """
+        return LanguageModelUserMessage(
+            content=[
+                {"type": "text", "text": self._content},
+                *[
+                    {"type": "image_url", "imageUrl": {"url": image}}
+                    for image in images
+                ],
+            ]
+        )

{unique_toolkit-0.5.42.dist-info → unique_toolkit-0.5.44.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.42
+Version: 0.5.44
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,6 +100,12 @@ 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.
+
 ## [0.5.42] - 2024-12-11
 - Update `LanguageModelTokenLimits` with fix avoiding floats for token
 

{unique_toolkit-0.5.42.dist-info → unique_toolkit-0.5.44.dist-info}/RECORD

@@ -9,7 +9,7 @@ unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-
 unique_toolkit/app/performance/async_tasks.py,sha256=H0l3OAcosLwNHZ8d2pd-Di4wHIXfclEvagi5kfqLFPA,1941
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
 unique_toolkit/app/schemas.py,sha256=6RY7Ex-B3pOnFILlitHEi9sqJvupbpdxlN9xt33qRsM,1571
-unique_toolkit/app/verification.py,sha256=UZqTHg3PX_QxMjeLH_BVBYoMVqMnMpeMoqvyTBKDqj8,1996
+unique_toolkit/app/verification.py,sha256=mffa6wm0i4hJbwzofePrkaia46xumMzECwQ0T3eKAx0,1929
 unique_toolkit/chat/__init__.py,sha256=1prdTVfLOf6NgU-Aa1VIO-XiR6OYuRm51LaVRfKDCqc,267
 unique_toolkit/chat/schemas.py,sha256=IHOnb3pWlRlSPoEUWBTl6LK8YeMdlg2iXi9ghyBeiLw,1495
 unique_toolkit/chat/service.py,sha256=lFcwCKHRHVLkMIO1cBHwH_Py0uZ4mmvgkUwQuZkpsxI,21438
@@ -35,12 +35,13 @@ unique_toolkit/evaluators/hallucination/service.py,sha256=k8qro5Lw4Ak58m4HYp3G4H
 unique_toolkit/evaluators/hallucination/utils.py,sha256=507BsX1mFTEne1-LdRCNMgBj-IXSFvBj1t3BPe1UkGs,7639
 unique_toolkit/evaluators/output_parser.py,sha256=eI72qkzK1dZyUvnfP2SOAQCGBj_-PwX5wy_aLPMsJMY,883
 unique_toolkit/evaluators/schemas.py,sha256=Jaue6Uhx75X1CyHKWj8sT3RE1JZXTqoLtfLt2xQNCX8,2507
-unique_toolkit/language_model/__init__.py,sha256=YuhyczGPj6w9xX-sOVUhmozvzIFxcckHFEkeMBecr5s,1784
+unique_toolkit/language_model/__init__.py,sha256=hgk5yiFF4SpIcE2QSoki9YknFxmcKnq2LCJ1cK9de9I,1830
 unique_toolkit/language_model/infos.py,sha256=kQK6F3r8xTN7oT6b39J7rxW-Y4iPXjx_Fr9bCOVQdm0,12509
+unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/schemas.py,sha256=MBwEAFnnCYjSpbZDUYLrqkCc-0bsarsepSQEj1v6YEY,6991
 unique_toolkit/language_model/service.py,sha256=brNCPRA0XxgqHi2rI5i2lyFCkUiw4MNMe1VaR3UgWmY,15500
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
-unique_toolkit-0.5.42.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.5.42.dist-info/METADATA,sha256=N2FElgrmwsR2zsNLdpgWp3OtAoF26gPw3-3bO0uaZtU,14861
-unique_toolkit-0.5.42.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.42.dist-info/RECORD,,
+unique_toolkit-0.5.44.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.44.dist-info/METADATA,sha256=IXHGnl23ctzUc3Pm2BXVwDoHyScGkgd7dBawlOkEAZ4,15154
+unique_toolkit-0.5.44.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.44.dist-info/RECORD,,