diff --git a/src/askui/models/shared/__init__.py b/src/askui/models/shared/__init__.py index 8168ab16..4df27a7b 100644 --- a/src/askui/models/shared/__init__.py +++ b/src/askui/models/shared/__init__.py @@ -1,11 +1,20 @@ from .android_base_tool import AndroidBaseTool from .computer_base_tool import ComputerBaseTool -from .playwright_base_tool import PlaywrightBaseTool from .tool_tags import ToolTags +try: + from .playwright_base_tool import PlaywrightBaseTool + + _PLAYWRIGHT_AVAILABLE = True +except ImportError: + _PLAYWRIGHT_AVAILABLE = False + + __all__ = [ "AndroidBaseTool", "ComputerBaseTool", - "PlaywrightBaseTool", "ToolTags", ] + +if _PLAYWRIGHT_AVAILABLE: + __all__ += ["PlaywrightBaseTool"]