Embeddings: Add string input as an option

Used in OAI's API

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-11-16 23:48:31 -05:00
parent f9fffd42e0
commit b94c646210

View file

@ -1,4 +1,4 @@
from typing import List, Optional
from typing import List, Optional, Union
from pydantic import BaseModel, Field
@ -10,7 +10,7 @@ class UsageInfo(BaseModel):
class EmbeddingsRequest(BaseModel):
input: List[str] = Field(
input: Union[str, List[str]] = Field(
..., description="List of input texts to generate embeddings for."
)
encoding_format: str = Field(