mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-25 20:43:56 +01:00
32 lines
885 B
Python
32 lines
885 B
Python
import os
|
|
import aiohttp
|
|
|
|
import proxies
|
|
|
|
from dotenv import load_dotenv
|
|
from request_manager import Request
|
|
|
|
load_dotenv()
|
|
|
|
async def stream_closedai_request(request: Request):
|
|
async with aiohttp.ClientSession(
|
|
connector=await proxies.default_proxy.get_connector(),
|
|
timeout=aiohttp.ClientTimeout(total=request.timeout),
|
|
raise_for_status=False
|
|
) as session:
|
|
async with session.request(
|
|
method=request.method,
|
|
url=request.url,
|
|
json=request.payload,
|
|
headers={
|
|
'Content-Type': 'application/json',
|
|
'Authorization': f'Bearer {os.getenv("CLOSEDAI_KEY")}'
|
|
}
|
|
) as response:
|
|
async for chunk in response.content.iter_any():
|
|
chunk = f'{chunk.decode("utf8")}\n\n'
|
|
yield chunk
|
|
|
|
if __name__ == '__main__':
|
|
pass
|