diff --git a/endpoints/OAI/types/embedding.py b/endpoints/OAI/types/embedding.py index 7937d50..860c3ed 100644 --- a/endpoints/OAI/types/embedding.py +++ b/endpoints/OAI/types/embedding.py @@ -27,7 +27,7 @@ class EmbeddingsRequest(BaseModel): class EmbeddingObject(BaseModel): 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." ) index: int = Field(