Merge pull request #86 from askui/refactor/make-agents-more-composeable

refactor!: make agents more composable

Author: adi-wan-askui

README.md

@@ -287,6 +287,7 @@ Here's how to create and use custom models:
 import functools
 from askui import (
     ActModel,
+    ActSettings,
     GetModel,
     LocateModel,
     Locator,
@@ -310,6 +311,8 @@ class MyActModel(ActModel):
         messages: list[MessageParam],
         model_choice: str,
         on_message: OnMessageCb | None = None,
+        tools: list[Tool] | None = None,
+        settings: ActSettings | None = None,
     ) -> None:
         # Implement custom act logic, e.g.:
         # - Use a different AI model
@@ -389,6 +392,8 @@ class DynamicActModel(ActModel):
         messages: list[MessageParam],
         model_choice: str,
         on_message: OnMessageCb | None = None,
+        tools: list[Tool] | None = None,
+        settings: ActSettings | None = None,
     ) -> None:
         pass
 

src/askui/__init__.py

@@ -31,13 +31,16 @@
     ToolUseBlockParam,
     UrlImageSourceParam,
 )
+from .models.shared.settings import ActSettings, MessageSettings
+from .models.shared.tools import Tool
 from .models.types.response_schemas import ResponseSchema, ResponseSchemaBase
 from .retry import ConfigurableRetry, Retry
 from .tools import ModifierKey, PcKey
 from .utils.image_utils import ImageSource, Img
 
 __all__ = [
     "ActModel",
+    "ActSettings",
     "Base64ImageSourceParam",
     "CacheControlEphemeralParam",
     "CitationCharLocationParam",
@@ -52,6 +55,7 @@
     "LocateModel",
     "Locator",
     "MessageParam",
+    "MessageSettings",
     "Model",
     "ModelChoice",
     "ModelComposition",
@@ -67,6 +71,7 @@
     "Retry",
     "TextBlockParam",
     "TextCitationParam",
+    "Tool",
     "ToolResultBlockParam",
     "ToolUseBlockParam",
     "UrlImageSourceParam",