Embeddings: Add string input as an option

Used in OAI's API Signed-off-by: kingbri <[email protected]>
theroyallab · Nov 17, 2024 · b94c646 · b94c646
1 parent f9fffd4
commit b94c646
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/endpoints/OAI/types/embedding.py b/endpoints/OAI/types/embedding.py
@@ -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(