From 9994bb67a1cc1a09f4c5eba70dfbd77c96d0ed76 Mon Sep 17 00:00:00 2001 From: hlohaus <983577+hlohaus@users.noreply.github.com> Date: Wed, 5 Feb 2025 17:07:20 +0100 Subject: Add zerogpu_uuid to demo --- g4f/tools/files.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'g4f/tools/files.py') diff --git a/g4f/tools/files.py b/g4f/tools/files.py index 350bc45e..20eb15c8 100644 --- a/g4f/tools/files.py +++ b/g4f/tools/files.py @@ -431,7 +431,7 @@ async def download_urls( connector=get_connector(proxy=proxy), timeout=ClientTimeout(timeout) ) as session: - async def download_url(url: str) -> str: + async def download_url(url: str, max_depth: int) -> str: try: async with session.get(url) as response: response.raise_for_status() @@ -457,7 +457,7 @@ async def download_urls( except (ClientError, asyncio.TimeoutError) as e: debug.log(f"Download failed: {e.__class__.__name__}: {e}") return None - for filename in await asyncio.gather(*[download_url(url) for url in urls]): + for filename in await asyncio.gather(*[download_url(url, max_depth) for url in urls]): if filename: yield filename else: -- cgit v1.2.3