diff options
Diffstat (limited to 'g4f/Provider/MagickPen.py')
| -rw-r--r-- | g4f/Provider/MagickPen.py | 41 |
1 files changed, 30 insertions, 11 deletions
diff --git a/g4f/Provider/MagickPen.py b/g4f/Provider/MagickPen.py index 0f476eca..eab70536 100644 --- a/g4f/Provider/MagickPen.py +++ b/g4f/Provider/MagickPen.py @@ -12,15 +12,19 @@ from .helper import format_prompt class MagickPen(AsyncGeneratorProvider, ProviderModelMixin): url = "https://magickpen.com" - api_endpoint = "https://api.magickpen.com/chat/free" + api_endpoint_free = "https://api.magickpen.com/chat/free" + api_endpoint_ask = "https://api.magickpen.com/ask" working = True supports_gpt_4 = True supports_stream = False - default_model = 'gpt-4o-mini' - models = ['gpt-4o-mini'] + default_model = 'free' + models = ['free', 'ask'] - model_aliases = {} + model_aliases = { + "gpt-4o-mini": "free", + "gpt-4o-mini": "ask", + } @classmethod def get_model(cls, model: str) -> str: @@ -102,10 +106,25 @@ class MagickPen(AsyncGeneratorProvider, ProviderModelMixin): } async with ClientSession(headers=headers) as session: - data = { - "history": [{"role": "user", "content": format_prompt(messages)}] - } - async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response: - response.raise_for_status() - result = await response.text() - yield result + if model == 'free': + data = { + "history": [{"role": "user", "content": format_prompt(messages)}] + } + async with session.post(cls.api_endpoint_free, json=data, proxy=proxy) as response: + response.raise_for_status() + result = await response.text() + yield result + + elif model == 'ask': + data = { + "query": format_prompt(messages), + "plan": "Pay as you go" + } + async with session.post(cls.api_endpoint_ask, json=data, proxy=proxy) as response: + response.raise_for_status() + async for chunk in response.content: + if chunk: + yield chunk.decode() + + else: + raise ValueError(f"Unknown model: {model}") |