diff options
author | Tekky <98614666+xtekky@users.noreply.github.com> | 2024-10-15 11:51:53 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-10-15 11:51:53 +0200 |
commit | 5ed3467d07181e876d957984c16782d687abd3b5 (patch) | |
tree | 23bd0fd3481d81fca70ac3c7842cb7ffa8f6497f /g4f/Provider/nexra/NexraImageURL.py | |
parent | Merge pull request #2268 from yjg30737/patch-1 (diff) | |
parent | Updated(docs/client.md) (diff) | |
download | gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.tar gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.tar.gz gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.tar.bz2 gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.tar.lz gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.tar.xz gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.tar.zst gpt4free-5ed3467d07181e876d957984c16782d687abd3b5.zip |
Diffstat (limited to 'g4f/Provider/nexra/NexraImageURL.py')
-rw-r--r-- | g4f/Provider/nexra/NexraImageURL.py | 46 |
1 files changed, 0 insertions, 46 deletions
diff --git a/g4f/Provider/nexra/NexraImageURL.py b/g4f/Provider/nexra/NexraImageURL.py deleted file mode 100644 index 13d70757..00000000 --- a/g4f/Provider/nexra/NexraImageURL.py +++ /dev/null @@ -1,46 +0,0 @@ -from __future__ import annotations -from aiohttp import ClientSession -import json -from ...typing import AsyncResult, Messages -from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin -from ..helper import format_prompt -from ...image import ImageResponse - -class NexraImageURL(AsyncGeneratorProvider, ProviderModelMixin): - label = "Image Generation Provider" - api_endpoint = "https://nexra.aryahcr.cc/api/image/complements" - models = ['dalle', 'dalle2', 'dalle-mini', 'emi', 'sdxl-turbo', 'prodia'] - - @classmethod - async def create_async_generator( - cls, - model: str, - messages: Messages, - proxy: str = None, - **kwargs - ) -> AsyncResult: - headers = { - "Content-Type": "application/json", - } - - async with ClientSession(headers=headers) as session: - prompt = format_prompt(messages) - data = { - "prompt": prompt, - "model": model, - "response": "url" - } - - async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response: - response.raise_for_status() - response_text = await response.text() - - cleaned_response = response_text.lstrip('_') - response_json = json.loads(cleaned_response) - - images = response_json.get("images") - if images and len(images) > 0: - image_response = ImageResponse(images[0], alt="Generated Image") - yield image_response - else: - yield "No image URL found." |