99i commited on
Commit
3eeaab0
·
verified ·
1 Parent(s): 1fcbf43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -176,10 +176,15 @@ async def chat_completions(request: Request):
176
  async def embeddings(request: Request):
177
  """转发 embedding 请求."""
178
  return await forward_request(request, "/embeddings")
179
- @app.post("/hf/v1/models")
180
- async def embeddings(request: Request):
181
- """转发 embedding 请求."""
182
- return await forward_request(request, "/models")
 
 
 
 
 
183
 
184
  @app.post("/hf/v1/images/generations")
185
  async def images_generations(request: Request):
 
176
  async def embeddings(request: Request):
177
  """转发 embedding 请求."""
178
  return await forward_request(request, "/embeddings")
179
+ @app.get("/hf/v1/models")
180
+ async def models(request: Request):
181
+ """转发 models 请求."""
182
+ async with httpx.AsyncClient(timeout=REQUEST_TIMEOUT) as client:
183
+ response = await client.get(
184
+ "https://api.siliconflow.cn/v1/models", headers=headers
185
+ )
186
+ response.raise_for_status()
187
+ return response.json()
188
 
189
  @app.post("/hf/v1/images/generations")
190
  async def images_generations(request: Request):