diff --git a/extensions/openai/typing.py b/extensions/openai/typing.py index 332a8c28..3a212dd9 100644 --- a/extensions/openai/typing.py +++ b/extensions/openai/typing.py @@ -120,7 +120,7 @@ class ChatCompletionResponse(BaseModel): class EmbeddingsRequest(BaseModel): - input: str | List[str] + input: str | List[str] | List[int] | List[List[int]] model: str | None = Field(default=None, description="Unused parameter. To change the model, set the OPENEDAI_EMBEDDING_MODEL and OPENEDAI_EMBEDDING_DEVICE environment variables before starting the server.") encoding_format: str = Field(default="float", description="Can be float or base64.") user: str | None = Field(default=None, description="Unused parameter.")