Fix include in mocks.py

This commit is contained in:
hlohaus 2025-02-03 21:24:37 +01:00
parent 9c8aedbeb1
commit b86bf0dcf7
2 changed files with 2 additions and 2 deletions

View file

@ -1,5 +1,5 @@
from g4f.providers.base_provider import AbstractProvider, AsyncProvider, AsyncGeneratorProvider
from g4f.image import ImageResponse
from g4f.providers.response import ImageResponse
from g4f.errors import MissingAuthError
class ProviderMock(AbstractProvider):

View file

@ -137,7 +137,7 @@ async def get_nodriver(
timeout: int = 120,
browser_executable_path=None,
**kwargs
) -> Browser:
) -> tuple[Browser, callable]:
if not has_nodriver:
raise MissingRequirementsError('Install "nodriver" and "platformdirs" package | pip install -U nodriver platformdirs')
user_data_dir = user_config_dir(f"g4f-{user_data_dir}") if has_platformdirs else None