mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-15 14:51:19 -08:00
Code Issues Projects Releases Packages Wiki Activity

Add Groq and Openai interfaces, Add integration tests

Browse source
This commit is contained in:
Heiner Lohaus 2024-04-06 01:05:00 +02:00
parent 1e2cf48cba
commit d44b39b31c
10 changed files with 167 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

16
g4f/requests/aiohttp.py
View file

@ -15,11 +15,19 @@ class StreamResponse(ClientResponse):
async for chunk in self.content.iter_any():
yield chunk
async def json(self) -> Any:
return await super().json(content_type=None)
async def json(self, content_type: str = None) -> Any:
return await super().json(content_type=content_type)
class StreamSession(ClientSession):
def __init__(self, headers: dict = {}, timeout: int = None, proxies: dict = {}, impersonate = None, **kwargs):
def __init__(
self,
headers: dict = {},
timeout: int = None,
connector: BaseConnector = None,
proxies: dict = {},
impersonate = None,
**kwargs
):
if impersonate:
headers = {
**DEFAULT_HEADERS,
@ -29,7 +37,7 @@ class StreamSession(ClientSession):
**kwargs,
timeout=ClientTimeout(timeout) if timeout else None,
response_class=StreamResponse,
connector=get_connector(kwargs.get("connector"), proxies.get("https")),
connector=get_connector(connector, proxies.get("all", proxies.get("https"))),
headers=headers
)