Update main.py
Browse files
main.py
CHANGED
|
@@ -49,7 +49,7 @@ class ChatRequest(BaseModel):
|
|
| 49 |
|
| 50 |
|
| 51 |
class EmbeddingRequest(BaseModel):
|
| 52 |
-
input: str
|
| 53 |
model: str = "text-embedding-004"
|
| 54 |
encoding_format: Optional[str] = "float"
|
| 55 |
|
|
|
|
| 49 |
|
| 50 |
|
| 51 |
class EmbeddingRequest(BaseModel):
|
| 52 |
+
input: Union[str, List[str]]
|
| 53 |
model: str = "text-embedding-004"
|
| 54 |
encoding_format: Optional[str] = "float"
|
| 55 |
|