fix: embedding 与 model_name 类型检查
This commit is contained in:
@@ -48,4 +48,4 @@ def acge_embedding(text: list[str]) -> list[list[float]]:
|
|||||||
norm="l2",
|
norm="l2",
|
||||||
axis=1,
|
axis=1,
|
||||||
)
|
)
|
||||||
return vector
|
return vector.tolist()
|
||||||
|
|||||||
6
app.py
6
app.py
@@ -32,12 +32,14 @@ async def embedding_api(req: EmbeddingAPIRequest) -> EmbeddingAPIResposne:
|
|||||||
req.input = [req.input]
|
req.input = [req.input]
|
||||||
|
|
||||||
# 进行 embedding 计算
|
# 进行 embedding 计算
|
||||||
embeddings: list[float] = []
|
embeddings: list[list[float]] = []
|
||||||
if req.model == "acge-large-zh":
|
if req.model == "acge-large-zh":
|
||||||
embeddings = acge_embedding(req.input).tolist()
|
embeddings = acge_embedding(req.input)
|
||||||
elif req.model == "text-embedding-ada-002":
|
elif req.model == "text-embedding-ada-002":
|
||||||
# [TODO]: Implement text-embedding-ada-002
|
# [TODO]: Implement text-embedding-ada-002
|
||||||
raise NotImplementedError("text-embedding-ada-002 not implemented yet!")
|
raise NotImplementedError("text-embedding-ada-002 not implemented yet!")
|
||||||
|
else:
|
||||||
|
raise ValueError("Unknown model name!")
|
||||||
|
|
||||||
# 与 OpenAI 接口返回格式一致
|
# 与 OpenAI 接口返回格式一致
|
||||||
# https://platform.openai.com/docs/api-reference/embeddings/create
|
# https://platform.openai.com/docs/api-reference/embeddings/create
|
||||||
|
|||||||
Reference in New Issue
Block a user