mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-02 01:27:14 +00:00
feat(model): Support deploy rerank model (#1522)
This commit is contained in:
@@ -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."""
|
||||
|
||||
|
Reference in New Issue
Block a user