Update FreeChatgpt.py

This commit is contained in:
H Lohaus 2024-04-06 23:50:17 +02:00 committed by GitHub
parent 7f16b7bb21
commit c85765925f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5,6 +5,7 @@ from aiohttp import ClientSession, ClientTimeout
from ..typing import AsyncResult, Messages from ..typing import AsyncResult, Messages
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
from ..requests.raise_for_status import raise_for_status
class FreeChatgpt(AsyncGeneratorProvider, ProviderModelMixin): class FreeChatgpt(AsyncGeneratorProvider, ProviderModelMixin):
url = "https://free.chatgpt.org.uk" url = "https://free.chatgpt.org.uk"
@ -45,7 +46,7 @@ class FreeChatgpt(AsyncGeneratorProvider, ProviderModelMixin):
"top_p": kwargs.get("top_p", 1) "top_p": kwargs.get("top_p", 1)
} }
async with session.post(f'{cls.url}/api/openai/v1/chat/completions', json=data, proxy=proxy) as response: async with session.post(f'{cls.url}/api/openai/v1/chat/completions', json=data, proxy=proxy) as response:
response.raise_for_status() await raise_for_status(response)
started = False started = False
async for line in response.content: async for line in response.content:
if line.startswith(b"data: [DONE]"): if line.startswith(b"data: [DONE]"):