diff options
author | kqlio67 <kqlio67@users.noreply.github.com> | 2024-11-06 20:53:18 +0100 |
---|---|---|
committer | kqlio67 <kqlio67@users.noreply.github.com> | 2024-11-06 20:53:18 +0100 |
commit | 18b309257c56b73f680debfd8eec1b12231c2698 (patch) | |
tree | f44c02b56916547e55f5ab5ea0f61bba27d44b55 /g4f/Provider/nexra/NexraFluxPro.py | |
parent | Update (g4f/Provider/Allyfy.py) (diff) | |
download | gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.tar gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.tar.gz gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.tar.bz2 gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.tar.lz gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.tar.xz gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.tar.zst gpt4free-18b309257c56b73f680debfd8eec1b12231c2698.zip |
Diffstat (limited to 'g4f/Provider/nexra/NexraFluxPro.py')
-rw-r--r-- | g4f/Provider/nexra/NexraFluxPro.py | 70 |
1 files changed, 0 insertions, 70 deletions
diff --git a/g4f/Provider/nexra/NexraFluxPro.py b/g4f/Provider/nexra/NexraFluxPro.py deleted file mode 100644 index cfb26385..00000000 --- a/g4f/Provider/nexra/NexraFluxPro.py +++ /dev/null @@ -1,70 +0,0 @@ -from __future__ import annotations - -import json -import requests -from ...typing import CreateResult, Messages -from ..base_provider import ProviderModelMixin, AbstractProvider -from ...image import ImageResponse - -class NexraFluxPro(AbstractProvider, ProviderModelMixin): - url = "https://nexra.aryahcr.cc/documentation/flux-pro/en" - api_endpoint = "https://nexra.aryahcr.cc/api/image/complements" - working = True - - default_model = 'flux' - models = [default_model] - model_aliases = { - "flux-pro": "flux", - } - - @classmethod - def get_model(cls, model: str) -> str: - if model in cls.models: - return model - elif model in cls.model_aliases: - return cls.model_aliases[model] - else: - return cls.default_model - - @classmethod - def create_completion( - cls, - model: str, - messages: Messages, - proxy: str = None, - response: str = "url", # base64 or url - **kwargs - ) -> CreateResult: - model = cls.get_model(model) - - headers = { - 'Content-Type': 'application/json' - } - - data = { - "prompt": messages[-1]["content"], - "model": model, - "response": response - } - - response = requests.post(cls.api_endpoint, headers=headers, json=data) - - result = cls.process_response(response) - yield result - - @classmethod - def process_response(cls, response): - if response.status_code == 200: - try: - content = response.text.strip() - content = content.lstrip('_') - data = json.loads(content) - if data.get('status') and data.get('images'): - image_url = data['images'][0] - return ImageResponse(images=[image_url], alt="Generated Image") - else: - return "Error: No image URL found in the response" - except json.JSONDecodeError as e: - return f"Error: Unable to decode JSON response. Details: {str(e)}" - else: - return f"Error: {response.status_code}, Response: {response.text}" |