From a591c5d409fd28b2b31f3df819c7ef88bec6d794 Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Sat, 14 Dec 2024 19:56:36 +0100 Subject: Add Clarity Token to Copilot, (#2482) * Add Clarity Token to Copilot, Ignore disabled providers in gui, Read cookie files after upload * Add more chat templates to HuggingFace --- g4f/Provider/Mhystical.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) (limited to 'g4f/Provider/Mhystical.py') diff --git a/g4f/Provider/Mhystical.py b/g4f/Provider/Mhystical.py index 2aa98ebc..14412c07 100644 --- a/g4f/Provider/Mhystical.py +++ b/g4f/Provider/Mhystical.py @@ -4,6 +4,7 @@ import json import logging from aiohttp import ClientSession from ..typing import AsyncResult, Messages +from ..requests.raise_for_status import raise_for_status from .base_provider import AsyncGeneratorProvider, ProviderModelMixin from .helper import format_prompt @@ -67,17 +68,10 @@ class Mhystical(AsyncGeneratorProvider, ProviderModelMixin): "messages": [{"role": "user", "content": format_prompt(messages)}] } async with session.post(cls.api_endpoint, json=data, headers=headers, proxy=proxy) as response: - if response.status == 400: - yield "Error: API key is missing" - elif response.status == 429: - yield "Error: Rate limit exceeded" - elif response.status == 500: - yield "Error: Internal server error" - else: - response.raise_for_status() - response_text = await response.text() - filtered_response = cls.filter_response(response_text) - yield filtered_response + await raise_for_status(response) + response_text = await response.text() + filtered_response = cls.filter_response(response_text) + yield filtered_response @staticmethod def filter_response(response_text: str) -> str: -- cgit v1.2.3