mirror of
https://github.com/BlueSkyXN/AI2API.git
synced 2025-09-26 20:51:19 +08:00
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
from fastapi import FastAPI, Request, Header
|
||
import httpx
|
||
import uvicorn
|
||
from typing import Optional
|
||
from fastapi.responses import StreamingResponse, Response
|
||
import logging
|
||
|
||
# 设置日志
|
||
logging.basicConfig(level=logging.INFO,
|
||
format='%(asctime)s [%(levelname)s] [%(name)s] %(message)s')
|
||
logger = logging.getLogger(__name__)
|
||
|
||
app = FastAPI()
|
||
|
||
TARGET_API_URL = "https://rad.huddlz.xyz" # 替换为您的目标API地址
|
||
|
||
@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
|
||
async def proxy_request(request: Request, path: str, authorization: Optional[str] = Header(None)):
|
||
# 记录请求信息
|
||
logger.info(f"接收到请求: {request.method} {request.url.path}")
|
||
|
||
# 获取原始请求体
|
||
body = await request.body()
|
||
|
||
# 获取并处理原始请求头
|
||
headers = dict(request.headers)
|
||
logger.debug(f"原始请求头: {headers}")
|
||
|
||
# 从Authorization中提取token并设置为x-api-key
|
||
if authorization:
|
||
token = authorization.replace("Bearer ", "")
|
||
headers["x-api-key"] = token
|
||
logger.debug(f"从Authorization提取并设置x-api-key")
|
||
|
||
# 移除可能导致问题的请求头
|
||
headers.pop("host", None)
|
||
|
||
# 处理内容编码相关头信息
|
||
# 移除Accept-Encoding头，让httpx自己处理内容压缩/解压缩
|
||
if "accept-encoding" in headers:
|
||
logger.debug(f"移除Accept-Encoding: {headers.get('accept-encoding')}")
|
||
headers.pop("accept-encoding", None)
|
||
|
||
# 构建完整的目标URL
|
||
url = f"{TARGET_API_URL}/{path}"
|
||
logger.info(f"转发请求到: {url}")
|
||
|
||
# 获取查询参数
|
||
params = dict(request.query_params)
|
||
|
||
try:
|
||
# 转发请求到目标API，禁用自动处理压缩内容
|
||
async with httpx.AsyncClient(headers={"Accept-Encoding": "identity"}) as client:
|
||
response = await client.request(
|
||
method=request.method,
|
||
url=url,
|
||
params=params,
|
||
headers=headers,
|
||
content=body,
|
||
timeout=60.0,
|
||
follow_redirects=True
|
||
)
|
||
|
||
logger.info(f"收到响应: 状态码 {response.status_code}")
|
||
logger.debug(f"响应头: {response.headers}")
|
||
|
||
# 处理响应头，移除可能导致问题的头信息
|
||
response_headers = dict(response.headers)
|
||
|
||
# 移除可能导致冲突的头信息
|
||
headers_to_remove = [
|
||
"content-length",
|
||
"transfer-encoding",
|
||
"content-encoding", # 重要：移除内容编码头
|
||
"server",
|
||
"connection"
|
||
]
|
||
|
||
for header in headers_to_remove:
|
||
if header in response_headers:
|
||
logger.debug(f"移除响应头: {header}")
|
||
response_headers.pop(header, None)
|
||
|
||
# 获取响应内容（已自动解压缩）
|
||
content = await response.aread()
|
||
|
||
# 返回未压缩的响应
|
||
return Response(
|
||
content=content,
|
||
status_code=response.status_code,
|
||
headers=response_headers
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"请求处理过程中发生错误: {str(e)}", exc_info=True)
|
||
return {"detail": f"代理服务器错误: {str(e)}"}, 500
|
||
|
||
if __name__ == "__main__":
|
||
uvicorn.run(app, host="0.0.0.0", port=9898) |