feat(model): Support deploy rerank model (#1522)

This commit is contained in:
Fangyin Cheng
2024-05-16 14:50:16 +08:00
committed by GitHub
parent 559affe87d
commit 593e974405
29 changed files with 814 additions and 75 deletions

View File

@@ -166,6 +166,23 @@ class EmbeddingsResponse(BaseModel):
usage: UsageInfo = Field(..., description="Usage info")
class RelevanceRequest(BaseModel):
"""Relevance request entity."""
model: str = Field(..., description="Rerank model name")
query: str = Field(..., description="Query text")
documents: List[str] = Field(..., description="Document texts")
class RelevanceResponse(BaseModel):
"""Relevance response entity."""
object: str = Field("list", description="Object type")
model: str = Field(..., description="Rerank model name")
data: List[float] = Field(..., description="Data list, relevance scores")
usage: UsageInfo = Field(..., description="Usage info")
class ModelPermission(BaseModel):
"""Model permission entity."""