From 474cb011c4a58beb6a56e1a1fd4e5b3e322622cf Mon Sep 17 00:00:00 2001 From: Davor Runje Date: Tue, 28 Jan 2025 22:43:56 +0100 Subject: [PATCH] polishing --- autogen/tools/experimental/browser_use/browser_use.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/autogen/tools/experimental/browser_use/browser_use.py b/autogen/tools/experimental/browser_use/browser_use.py index 92d002c4b..08e57d393 100644 --- a/autogen/tools/experimental/browser_use/browser_use.py +++ b/autogen/tools/experimental/browser_use/browser_use.py @@ -6,6 +6,7 @@ from pydantic import BaseModel +from ....doc_utils import export_module from ....import_utils import optional_import_block, require_optional_import from ... import Depends, Tool @@ -19,8 +20,7 @@ __all__ = ["BrowserUseResult", "BrowserUseTool"] -# todo: add export_module decorator -# @export_module("autogen.tools.experimental.browser_use") +@export_module("autogen.tools.experimental.browser_use") class BrowserUseResult(BaseModel): """The result of using the browser to perform a task. @@ -44,8 +44,7 @@ def inner(_x: T = x) -> T: @require_optional_import(["langchain_openai", "browser_use"], "browser-use") -# todo: add export_module decorator -# @export_module("autogen.tools.experimental") +@export_module("autogen.tools.experimental") class BrowserUseTool(Tool): """BrowserUseTool is a tool that uses the browser to perform a task."""