Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Depends, Request  # 导入Request
+from fastapi import FastAPI, HTTPException, Depends, Request
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import os
@@ -95,30 +95,32 @@ async def create_embedding(
     req: EmbeddingRequest,
     _: bool = Depends(verify_api_key)
 ):
-    # 打印Authorization头部
-    auth_header = request.headers.get("Authorization")
-    logger.info(f"请求中的Authorization头部: {auth_header}")
+    # 打印完整请求信息
+    logger.info("\n===== 接收到的完整请求信息 =====")
+    logger.info(f"请求方法: {request.method}")
+    logger.info(f"请求URL: {request.url}")
+    logger.info("请求头部:")
+    for name, value in request.headers.items():
+        logger.info(f"  {name}: {value}")
+    logger.info(f"请求体: {await request.body()}")  # 打印原始请求体
+    logger.info("===============================\n")
+
+    # 原有逻辑保持不变
     logger.info(f"收到嵌入请求，模型: {req.model}, 输入类型: {type(req.input)}")
     try:
-        # 获取模型
         model = get_model(req.model)
-        
-        # 处理输入（支持单文本或文本列表）
         inputs = [req.input] if isinstance(req.input, str) else req.input
         logger.info(f"处理输入，文本数量: {len(inputs)}")
         
-        # 计算嵌入
         logger.info("开始计算嵌入")
         embeddings = model.encode(inputs, normalize_embeddings=True)
         logger.info(f"嵌入计算完成，嵌入形状: {embeddings.shape}")
         
-        # 构建响应
         data = [
             EmbeddingData(embedding=embedding.tolist(), index=i)
             for i, embedding in enumerate(embeddings)
         ]
         
-        # 估算token数（简单近似：每个单词约1 token）
         prompt_tokens = sum(len(text.split()) for text in inputs)
         logger.info(f"估算token数: {prompt_tokens}")
         
@@ -132,13 +134,19 @@ async def create_embedding(
         logger.error(error_msg)
         raise HTTPException(status_code=500, detail=error_msg)
 
-# 健康检查接口
+# 健康检查接口也打印完整请求
 @app.get("/health")
-async def health_check():
-    logger.info("健康检查请求")
+async def health_check(request: Request):
+    logger.info("\n===== 健康检查请求信息 =====")
+    logger.info(f"请求方法: {request.method}")
+    logger.info(f"请求URL: {request.url}")
+    logger.info("请求头部:")
+    for name, value in request.headers.items():
+        logger.info(f"  {name}: {value}")
+    logger.info("===============================\n")
     return {"status": "healthy", "models": list(MODEL_MAPPING.keys())}
 
 if __name__ == "__main__":
     import uvicorn
     logger.info("启动服务")
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)