summaryrefslogblamecommitdiffstats
path: root/g4f/Provider/MetaAIAccount.py
blob: 369b3f2f97983433f213f3c8ba4d0eac6b81b934 (plain) (tree)
1
2
3
4
5
6
7
8
9







                                                   
                     
                           












                                                                                     
from __future__ import annotations

from ..typing import AsyncResult, Messages, Cookies
from .helper import format_prompt, get_cookies
from .MetaAI import MetaAI

class MetaAIAccount(MetaAI):
    needs_auth = True
    parent = "MetaAI"
    image_models = ["meta"]

    @classmethod
    async def create_async_generator(
        cls,
        model: str,
        messages: Messages,
        proxy: str = None,
        cookies: Cookies = None,
        **kwargs
    ) -> AsyncResult:
        cookies = get_cookies(".meta.ai", True, True) if cookies is None else cookies
        async for chunk in cls(proxy).prompt(format_prompt(messages), cookies):
            yield chunk