Fix stage predefined function overloads

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "uv_build"
 [project]
 name = "draive"
 description = "Framework designed to simplify and accelerate the development of LLM-based applications."
-version = "0.99.0"
+version = "0.99.1"
 readme = "README.md"
 maintainers = [
   { name = "Kacper Kaliński", email = "kacper.kalinski@miquido.com" },

src/draive/guardrails/safety/regex.py

@@ -68,7 +68,7 @@ def _is_actionable_command(
     )
     for separator in (".", "!", "?", "\n"):
         if separator in normalized_prefix:
-            normalized_prefix = normalized_prefix.split(separator)[-1]
+            normalized_prefix = normalized_prefix.rsplit(separator, maxsplit=1)[-1]
 
     clause: str = normalized_prefix.strip()
     if not clause:

src/draive/stages/stage.py

@@ -100,6 +100,7 @@ def predefined(
         element: ModelContextElement,
         /,
         *elements: ModelContextElement,
+        result: Multimodal | None = None,
         meta: Meta | MetaValues | None = None,
     ) -> Self: ...
 
@@ -110,6 +111,7 @@ def predefined(
         element: Multimodal,
         /,
         *elements: Multimodal,
+        result: Multimodal | None = None,
         meta: Meta | MetaValues | None = None,
     ) -> Self: ...
 

tests/test_aws_state_observability.py

@@ -307,9 +307,5 @@ async def metric_putting(
         scope.record_scope_end(error)
         await asyncio.gather(*tasks)
 
-    exception_events = [
-        event
-        for event in recorded_events
-        if event["detail_type"] == "exception"
-    ]
+    exception_events = [event for event in recorded_events if event["detail_type"] == "exception"]
     assert len(exception_events) == 1