geqintan commited on
Commit
52ef174
·
1 Parent(s): 1868dc4
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from fastapi import FastAPI, Request, HTTPException, Depends, Header
2
  from pydantic import BaseModel, Field
3
  from sentence_transformers import SentenceTransformer
 
4
  import numpy as np
5
  import logging, os
6
 
@@ -29,7 +30,7 @@ except Exception as e:
29
  raise HTTPException(status_code=500, detail="Model loading failed")
30
 
31
  class EmbeddingRequest(BaseModel):
32
- input: str | list[str]
33
 
34
  @app.post("/v1/embeddings")
35
  async def embeddings(request, authorization: str = Depends(check_authorization)):
 
1
  from fastapi import FastAPI, Request, HTTPException, Depends, Header
2
  from pydantic import BaseModel, Field
3
  from sentence_transformers import SentenceTransformer
4
+ from typing import Union, List # 添加必要的类型导入
5
  import numpy as np
6
  import logging, os
7
 
 
30
  raise HTTPException(status_code=500, detail="Model loading failed")
31
 
32
  class EmbeddingRequest(BaseModel):
33
+ input: Union[str, List[str]] # 修复类型定义
34
 
35
  @app.post("/v1/embeddings")
36
  async def embeddings(request, authorization: str = Depends(check_authorization)):