CodeCow/chat2api
1
0
Fork 0
mirror of https://github.com/lanqian528/chat2api.git synced 2026-06-13 21:02:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update gateway

Browse Source
This commit is contained in:
LanQian 2024-05-17 04:11:25 +08:00
parent e8c450ae98
commit 80c01b38bb
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
chatgpt/reverseProxy.py
View File

@ -101,7 +101,6 @@ async def chatgpt_reverse_proxy(request: Request, path: str):
data = await request.body()
client = Client(proxy=random.choice(proxy_url_list) if proxy_url_list else None)
r = None
try:
r = await client.request(request.method, f"{base_url}/{path}", params=params, headers=headers,
cookies=request_cookies, data=data, stream=True, allow_redirects=False)
@ -127,13 +126,17 @@ async def chatgpt_reverse_proxy(request: Request, path: str):
return StreamingResponse(r.aiter_content(), media_type=r.headers.get("content-type", ""),
background=background)
else:
content = ((await r.atext()).replace("chatgpt.com", origin_host)
.replace("chat.openai.com", origin_host)
.replace("ab.chatgpt.com", origin_host)
.replace("cdn.oaistatic.com", origin_host)
.replace("https", petrol))
response = Response(content=content, media_type=r.headers.get("content-type"),
status_code=r.status_code)
if "/conversation" in path or "/register-websocket" in path:
response = Response(content=r.content, media_type=r.headers.get("content-type"),
status_code=r.status_code)
else:
content = ((await r.atext()).replace("chatgpt.com", origin_host)
.replace("chat.openai.com", origin_host)
.replace("ab.chatgpt.com", origin_host)
.replace("cdn.oaistatic.com", origin_host)
.replace("https", petrol))
response = Response(content=content, media_type=r.headers.get("content-type"),
status_code=r.status_code)
for cookie_name in r.cookies:
if cookie_name in request_cookies:
continue
@ -145,8 +148,7 @@ async def chatgpt_reverse_proxy(request: Request, path: str):
response.set_cookie(key=cookie_name, value=cookie_value)
return response
finally:
if r and 'stream' not in r.headers.get("content-type", ""):
await client.close()
await client.close()
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))