diff options
Diffstat (limited to 'g4f')
| -rw-r--r-- | g4f/client/client.py | 28 |
1 files changed, 21 insertions, 7 deletions
diff --git a/g4f/client/client.py b/g4f/client/client.py index 8e195213..63358302 100644 --- a/g4f/client/client.py +++ b/g4f/client/client.py @@ -154,14 +154,29 @@ class AsyncClient(Client): stacklevel=2 ) super().__init__(*args, **kwargs) + self.chat = Chat(self) + self._images = Images(self) + self.completions = Completions(self) - async def chat_complete(self, *args, **kwargs): - """Legacy method that redirects to async_create""" - return await self.chat.completions.async_create(*args, **kwargs) + @property + def images(self) -> 'Images': + return self._images + + async def async_create(self, *args, **kwargs) -> Union['ChatCompletion', AsyncIterator['ChatCompletionChunk']]: + response = await super().async_create(*args, **kwargs) + async for result in response: + return result - async def create_image(self, *args, **kwargs): - """Legacy method that redirects to async_generate""" - return await self.images.async_generate(*args, **kwargs) + async def async_generate(self, *args, **kwargs) -> 'ImagesResponse': + return await super().async_generate(*args, **kwargs) + + async def _fetch_image(self, url: str) -> bytes: + async with ClientSession() as session: + async with session.get(url) as resp: + if resp.status == 200: + return await resp.read() + else: + raise Exception(f"Failed to fetch image from {url}, status code {resp.status}") class Completions: def __init__(self, client: Client, provider: ProviderType = None): @@ -531,4 +546,3 @@ class Images: async def create_variation(self, image: Union[str, bytes], model: str = None, response_format: str = "url", **kwargs): # Existing implementation, adjust if you want to support b64_json here as well pass - |