refactor(g4f/client/__init__.py): Improve provider handling in async_…

…generate method
kqlio67 · Nov 19, 2024 · 6ff94bc · 6ff94bc
1 parent 60f7f5d
commit 6ff94bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/g4f/client/__init__.py b/g4f/client/__init__.py
@@ -254,6 +254,8 @@ async def async_generate(self, prompt: str, model: str = None, provider: Provide
             provider_handler = self.models.get(model, provider or self.provider or BingCreateImages)
         elif isinstance(provider, str):
             provider_handler = convert_to_provider(provider)
+        else:
+            provider_handler = provider
         if provider_handler is None:
             raise ValueError(f"Unknown model: {model}")
         if proxy is None: