diff options
author | AndPim4912 <52836885+AndPim4912@users.noreply.github.com> | 2023-10-23 14:10:28 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-10-23 14:10:28 +0200 |
commit | 8d7ad98fcb314453c01f6dcd4bfe59016efd78ce (patch) | |
tree | 4d39804e1aa1724560e1688f0b6d4c124b9771c7 /g4f/Provider/unfinished | |
parent | Refactor API initialization to accept a list of ignored providers. (diff) | |
parent | Debug config in api (diff) | |
download | gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.tar gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.tar.gz gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.tar.bz2 gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.tar.lz gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.tar.xz gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.tar.zst gpt4free-8d7ad98fcb314453c01f6dcd4bfe59016efd78ce.zip |
Diffstat (limited to 'g4f/Provider/unfinished')
-rw-r--r-- | g4f/Provider/unfinished/ChatAiGpt.py | 8 | ||||
-rw-r--r-- | g4f/Provider/unfinished/MikuChat.py | 3 | ||||
-rw-r--r-- | g4f/Provider/unfinished/PerplexityAi.py | 2 |
3 files changed, 7 insertions, 6 deletions
diff --git a/g4f/Provider/unfinished/ChatAiGpt.py b/g4f/Provider/unfinished/ChatAiGpt.py index c77afb48..9d050093 100644 --- a/g4f/Provider/unfinished/ChatAiGpt.py +++ b/g4f/Provider/unfinished/ChatAiGpt.py @@ -43,10 +43,12 @@ class ChatAiGpt(AsyncGeneratorProvider): async with session.get(f"{cls.url}/", proxy=proxy) as response: response.raise_for_status() response = await response.text() - result = re.search(r'data-nonce=(.*?) data-post-id=([0-9]+)', response) - if not result: + if result := re.search( + r'data-nonce=(.*?) data-post-id=([0-9]+)', response + ): + cls._nonce, cls._post_id = result.group(1), result.group(2) + else: raise RuntimeError("No nonce found") - cls._nonce, cls._post_id = result.group(1), result.group(2) prompt = format_prompt(messages) data = { "_wpnonce": cls._nonce, diff --git a/g4f/Provider/unfinished/MikuChat.py b/g4f/Provider/unfinished/MikuChat.py index bf19631f..970fb0bb 100644 --- a/g4f/Provider/unfinished/MikuChat.py +++ b/g4f/Provider/unfinished/MikuChat.py @@ -48,8 +48,7 @@ class MikuChat(AsyncGeneratorProvider): async for line in response.iter_lines(): if line.startswith(b"data: "): line = json.loads(line[6:]) - chunk = line["choices"][0]["delta"].get("content") - if chunk: + if chunk := line["choices"][0]["delta"].get("content"): yield chunk def k(e: str, t: int): diff --git a/g4f/Provider/unfinished/PerplexityAi.py b/g4f/Provider/unfinished/PerplexityAi.py index 3e096808..e97dbf0d 100644 --- a/g4f/Provider/unfinished/PerplexityAi.py +++ b/g4f/Provider/unfinished/PerplexityAi.py @@ -21,7 +21,7 @@ class PerplexityAi(AsyncProvider): proxy: str = None, **kwargs ) -> str: - url = cls.url + "/socket.io/?EIO=4&transport=polling" + url = f"{cls.url}/socket.io/?EIO=4&transport=polling" headers = { "Referer": f"{cls.url}/" } |