summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Pi.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2023-12-01 23:56:12 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2023-12-01 23:56:12 +0100
commite74af803e7add7790b366545f1fe6c80934de798 (patch)
tree4cdd527acfd2e5514a3ca638c136a0c5d59faf04 /g4f/Provider/Pi.py
parentAdded new provider PI (Hacky way to use) (#1291) (diff)
downloadgpt4free-e74af803e7add7790b366545f1fe6c80934de798.tar
gpt4free-e74af803e7add7790b366545f1fe6c80934de798.tar.gz
gpt4free-e74af803e7add7790b366545f1fe6c80934de798.tar.bz2
gpt4free-e74af803e7add7790b366545f1fe6c80934de798.tar.lz
gpt4free-e74af803e7add7790b366545f1fe6c80934de798.tar.xz
gpt4free-e74af803e7add7790b366545f1fe6c80934de798.tar.zst
gpt4free-e74af803e7add7790b366545f1fe6c80934de798.zip
Diffstat (limited to 'g4f/Provider/Pi.py')
-rw-r--r--g4f/Provider/Pi.py93
1 files changed, 93 insertions, 0 deletions
diff --git a/g4f/Provider/Pi.py b/g4f/Provider/Pi.py
new file mode 100644
index 00000000..9ecebafb
--- /dev/null
+++ b/g4f/Provider/Pi.py
@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+from ..typing import CreateResult, Messages
+from .base_provider import BaseProvider, format_prompt
+
+import json
+from cloudscraper import CloudScraper, session, create_scraper
+
+class Pi(BaseProvider):
+ url = "https://chat-gpt.com"
+ working = True
+ supports_stream = True
+
+ @classmethod
+ def create_completion(
+ cls,
+ model: str,
+ messages: Messages,
+ stream: bool,
+ proxy: str = None,
+ scraper: CloudScraper = None,
+ conversation: dict = None,
+ **kwargs
+ ) -> CreateResult:
+ if not scraper:
+ scraper = cls.get_scraper()
+ if not conversation:
+ conversation = cls.start_conversation(scraper)
+ answer = cls.ask(scraper, messages, conversation)
+
+ last_answer = 0
+ for line in answer:
+ if "text" in line:
+ yield line["text"][last_answer:]
+ last_answer = len(line["text"])
+
+ def get_scraper():
+ scraper = create_scraper(
+ browser={
+ 'browser': 'chrome',
+ 'platform': 'windows',
+ 'desktop': True
+ },
+ sess=session()
+ )
+ scraper.headers = {
+ 'Accept': '*/*',
+ 'Accept-Encoding': 'deflate,gzip,br',
+ }
+ return scraper
+
+ def start_conversation(scraper: CloudScraper):
+ response = scraper.post('https://pi.ai/api/chat/start', data="{}", headers={
+ 'accept': 'application/json',
+ 'x-api-version': '3'
+ })
+ if 'Just a moment' in response.text:
+ raise RuntimeError('Error: Cloudflare detected')
+ return Conversation(
+ response.json()['conversations'][0]['sid'],
+ response.cookies
+ )
+
+ def get_chat_history(scraper: CloudScraper, conversation: Conversation):
+ params = {
+ 'conversation': conversation.sid,
+ }
+ response = scraper.get('https://pi.ai/api/chat/history', params=params, cookies=conversation.cookies)
+ if 'Just a moment' in response.text:
+ raise RuntimeError('Error: Cloudflare detected')
+ return response.json()
+
+ def ask(scraper: CloudScraper, messages: Messages, conversation: Conversation):
+ json_data = {
+ 'text': format_prompt(messages),
+ 'conversation': conversation.sid,
+ 'mode': 'BASE',
+ }
+ response = scraper.post('https://pi.ai/api/chat', json=json_data, cookies=conversation.cookies, stream=True)
+
+ for line in response.iter_lines(chunk_size=1024, decode_unicode=True):
+ if 'Just a moment' in line:
+ raise RuntimeError('Error: Cloudflare detected')
+ if line.startswith('data: {"text":'):
+ yield json.loads(line.split('data: ')[1])
+ if line.startswith('data: {"title":'):
+ yield json.loads(line.split('data: ')[1])
+
+class Conversation():
+ def __init__(self, sid: str, cookies):
+ self.sid = sid
+ self.cookies = cookies
+ \ No newline at end of file