test fix

microsoft · Dec 10, 2024 · b5c5a10 · b5c5a10
1 parent 2a7a87f
commit b5c5a10
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/python/packages/autogen-ext/pyproject.toml b/python/packages/autogen-ext/pyproject.toml
@@ -78,7 +78,10 @@ testpaths = ["tests"]
 include = "../../shared_tasks.toml"
 
 [tool.poe.tasks]
-test = "pytest -n auto"
+test = [
+    "playwright install",
+    "pytest -n auto"
+]
 mypy = "mypy --config-file ../../pyproject.toml --exclude src/autogen_ext/runtimes/grpc/protos --exclude tests/protos src tests"
 
 [tool.mypy]

diff --git a/python/packages/autogen-ext/src/autogen_ext/agents/web_surfer/_multimodal_web_surfer.py b/python/packages/autogen-ext/src/autogen_ext/agents/web_surfer/_multimodal_web_surfer.py
@@ -103,7 +103,7 @@ class MultimodalWebSurfer(BaseChatAgent):
             import asyncio
             from autogen_agentchat.ui import Console
             from autogen_agentchat.teams import RoundRobinGroupChat
-            from autogen_ext.models import OpenAIChatCompletionClient
+            from autogen_ext.models.openai import OpenAIChatCompletionClient
             from autogen_ext.agents.web_surfer import MultimodalWebSurfer
 
 

diff --git a/python/packages/autogen-ext/src/autogen_ext/agents/web_surfer/playwright_controller.py b/python/packages/autogen-ext/src/autogen_ext/agents/web_surfer/playwright_controller.py
@@ -58,7 +58,7 @@ def __init__(
         self.to_resize_viewport = to_resize_viewport
         self._page_script: str = ""
         self.last_cursor_position: Tuple[float, float] = (0.0, 0.0)
-        self._markdown_converter: Optional[MarkItDown] | None = None
+        self._markdown_converter: Optional[Any] | None = None
 
         # Read page_script
         with open(os.path.join(os.path.abspath(os.path.dirname(__file__)), "page_script.js"), "rt") as fh: