Embeddings: Fix base64 return

A base64 embedding can be a string post-encoding.

Signed-off-by: kingbri <8082010+bdashore3@users.noreply.github.com>
This commit is contained in:
kingbri 2025-01-01 16:15:12 -05:00
parent 245bd5c008
commit 6da65a8fd3

View file

@ -27,7 +27,7 @@ class EmbeddingsRequest(BaseModel):
class EmbeddingObject(BaseModel): class EmbeddingObject(BaseModel):
object: str = Field("embedding", description="Type of the object.") object: str = Field("embedding", description="Type of the object.")
embedding: List[float] = Field( embedding: Union[List[float], str] = Field(
..., description="Embedding values as a list of floats." ..., description="Embedding values as a list of floats."
) )
index: int = Field( index: int = Field(