From db2d6cffd9268e8842f0c2aa8e062c0e19d88d28 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Sun, 21 Apr 2024 07:22:59 +0200
Subject: Fix Bing Provider, Add Account Support for Meta AI Use default
 headers in OpenaiChat

---
 g4f/Provider/MetaAIAccount.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 g4f/Provider/MetaAIAccount.py

(limited to 'g4f/Provider/MetaAIAccount.py')

diff --git a/g4f/Provider/MetaAIAccount.py b/g4f/Provider/MetaAIAccount.py
new file mode 100644
index 00000000..8be2318e
--- /dev/null
+++ b/g4f/Provider/MetaAIAccount.py
@@ -0,0 +1,21 @@
+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
+
+    @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
\ No newline at end of file
-- 
cgit v1.2.3