fiewolf1000 commited on
Commit
2a68bca
·
verified ·
1 Parent(s): 795e310

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import torch
6
  import os
7
  from typing import List, Optional
8
  from datetime import datetime # 需在文件开头导入
 
9
 
10
  # 1. 初始化FastAPI应用
11
  app = FastAPI(
@@ -170,6 +171,26 @@ async def health_check():
170
  "timestamp": str(pd.Timestamp.now())
171
  }
172
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
173
  # 7. 本地运行入口(开发环境用)
174
  if __name__ == "__main__":
175
  import uvicorn
 
6
  import os
7
  from typing import List, Optional
8
  from datetime import datetime # 需在文件开头导入
9
+ from fastapi.responses import MarkdownResponse
10
 
11
  # 1. 初始化FastAPI应用
12
  app = FastAPI(
 
171
  "timestamp": str(pd.Timestamp.now())
172
  }
173
 
174
+
175
+
176
+ # ------------------- 新增:根路径(/)首页路由 -------------------
177
+ @app.get("/", response_class=MarkdownResponse, description="API 首页(含调用指南)")
178
+ async def home_page():
179
+ """根路径首页:展示 API 功能、调用示例、认证方式等"""
180
+ return f"""
181
+ # Cross-Encoder 重排序 API(兼容 GPT 格式)
182
+
183
+ 基于 `cross-encoder/ms-marco-MiniLM-L-6-v2` 模型,提供文本相关性排序服务,支持 GPT 标准 API 调用格式。
184
+
185
+ ## 核心功能
186
+ - 输入「查询语句 + 候选文档列表」,返回按相关性降序排列的结果(含分数、排名)
187
+ - 兼容 OpenAI 风格 API 格式,可直接用 OpenAI 库调用
188
+ - 支持 API Key 认证,保障接口安全"""
189
+
190
+
191
+
192
+
193
+
194
  # 7. 本地运行入口(开发环境用)
195
  if __name__ == "__main__":
196
  import uvicorn