diff --git a/gateway/backend.py b/gateway/backend.py index 580449c..0f9541b 100644 --- a/gateway/backend.py +++ b/gateway/backend.py @@ -38,7 +38,7 @@ if enable_gateway: return await login_html(request) user_remix_context = remix_context.copy() - set_value_for_key(user_remix_context, "user", {"id": "user-chatgpt"}) + set_value_for_key(user_remix_context, "user", {}) set_value_for_key(user_remix_context, "accessToken", token) response = templates.TemplateResponse("chatgpt.html", {"request": request, "remix_context": user_remix_context}) @@ -296,6 +296,9 @@ if enable_gateway: @app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH", "TRACE"]) async def reverse_proxy(request: Request, path: str): + if re.match("/v1/rgstr", path): + return Response(status_code=202, content=b'{"success":true}') + if re.match("ces/v1", path): return {"success": True} diff --git a/gateway/reverseProxy.py b/gateway/reverseProxy.py index ca3e81e..2557bff 100644 --- a/gateway/reverseProxy.py +++ b/gateway/reverseProxy.py @@ -148,6 +148,8 @@ async def chatgpt_reverse_proxy(request: Request, path: str): base_url = "https://cdn.oaistatic.com" if "file-" in path and "backend-api" not in path: base_url = "https://files.oaiusercontent.com" + if "v1/" in path: + base_url = "https://ab.chatgpt.com" token = request.cookies.get("token", "") req_token = await get_real_req_token(token) @@ -199,9 +201,14 @@ async def chatgpt_reverse_proxy(request: Request, path: str): response = Response(content=(await r.atext()), media_type=r.headers.get("content-type"), status_code=r.status_code, background=background) else: - content = ((await r.atext()).replace("chatgpt.com", origin_host) + content = await r.atext() + if "initialize" in content: + pass + content = (content + .replace("ab.chatgpt.com", origin_host) .replace("cdn.oaistatic.com", origin_host) # .replace("files.oaiusercontent.com", origin_host) + .replace("chatgpt.com", origin_host) .replace("https", petrol)) rheaders = dict(r.headers) content_type = rheaders.get("content-type", "") diff --git a/version.txt b/version.txt index 15a4939..75f2cd8 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -1.6.7-beta3 \ No newline at end of file +1.6.7-beta4 \ No newline at end of file