~ | Merge pull request xtekky#1092 from ostix360/main

Add timeout
Lin-jun-xiang · Oct 18, 2023 · 591e60e · 591e60e
2 parents 2e68a4b + 24f7495
commit 591e60e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/g4f/Provider/retry_provider.py b/g4f/Provider/retry_provider.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import asyncio
 import random
 from typing import List, Type, Dict
 from ..typing import CreateResult, Messages
@@ -68,7 +69,11 @@ async def create_async(
         self.exceptions: Dict[str, Exception] = {}
         for provider in providers:
             try:
-                return await provider.create_async(model, messages, **kwargs)
+                return await asyncio.wait_for(provider.create_async(model, messages, **kwargs), timeout=60)
+            except asyncio.TimeoutError as e:
+                self.exceptions[provider.__name__] = e
+                if logging:
+                    print(f"{provider.__name__}: TimeoutError: {e}")
             except Exception as e:
                 self.exceptions[provider.__name__] = e
                 if logging: