Skip to content

Commit 7dd8597

Browse files
CopilotOhYee
andcommitted
fix(mypy): annotate input_data as Any and add wait_until default tests
Co-authored-by: OhYee <13498329+OhYee@users.noreply.github.com>
1 parent 1c6e080 commit 7dd8597

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed

tests/unittests/integration/test_browser_toolset_error_handling.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -520,3 +520,33 @@ def callback(sb):
520520
assert "error" in result
521521
toolset._reset_playwright.assert_not_called()
522522
assert toolset.sandbox is original_sandbox
523+
524+
525+
class TestBrowserNavigationDefaults:
526+
"""测试浏览器导航函数的默认参数 / Tests for browser navigation function defaults"""
527+
528+
def test_browser_navigate_back_default_wait_until_is_domcontentloaded(
529+
self,
530+
):
531+
"""测试 browser_navigate_back 的 wait_until 默认值为 domcontentloaded"""
532+
default = (
533+
BrowserToolSet.browser_navigate_back.args_schema.model_fields[
534+
"wait_until"
535+
].default
536+
)
537+
assert default == "domcontentloaded", (
538+
f"Expected 'domcontentloaded' but got '{default}'"
539+
)
540+
541+
def test_browser_go_forward_default_wait_until_is_domcontentloaded(
542+
self,
543+
):
544+
"""测试 browser_go_forward 的 wait_until 默认值为 domcontentloaded"""
545+
default = (
546+
BrowserToolSet.browser_go_forward.args_schema.model_fields[
547+
"wait_until"
548+
].default
549+
)
550+
assert default == "domcontentloaded", (
551+
f"Expected 'domcontentloaded' but got '{default}'"
552+
)

tests/unittests/integration/test_langchain_agui_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -639,7 +639,7 @@ async def invoke_agent(request: AgentRequest):
639639
tools=tools,
640640
)
641641

642-
input_data = {
642+
input_data: Any = {
643643
"messages": [{
644644
"role": "user",
645645
"content": (

0 commit comments

Comments
 (0)