Add openai embedding API (#997)

This commit is contained in:
Ying Sheng
2024-08-09 11:19:18 -07:00
committed by GitHub
parent 05c50a82b8
commit b16e856f11
8 changed files with 135 additions and 19 deletions

View File

@@ -319,8 +319,14 @@ class EmbeddingRequest(BaseModel):
user: Optional[str] = None
class EmbeddingResponse(BaseModel):
index: str
embedding: List[float] = None
class EmbeddingObject(BaseModel):
embedding: List[float]
index: int
object: str = "embedding"
class EmbeddingResponse(BaseModel):
data: List[EmbeddingObject]
model: str
object: str = "list"
usage: Optional[UsageInfo] = None