Spaces:
Running
Running
fahmiaziz98
commited on
Commit
·
ee35e05
1
Parent(s):
c7bad32
[UPDATE] Refactor request and response schemas to include /query endpoint in documentation
Browse files
src/models/schemas/__init__.py
CHANGED
|
@@ -13,14 +13,12 @@ from .common import (
|
|
| 13 |
EmbeddingOptions,
|
| 14 |
)
|
| 15 |
|
| 16 |
-
from .requests import BaseEmbedRequest, EmbedRequest
|
| 17 |
|
| 18 |
from .responses import (
|
| 19 |
BaseEmbedResponse,
|
| 20 |
DenseEmbedResponse,
|
| 21 |
SparseEmbedResponse,
|
| 22 |
-
BatchDenseEmbedResponse,
|
| 23 |
-
BatchSparseEmbedResponse,
|
| 24 |
ModelsListResponse,
|
| 25 |
RootResponse,
|
| 26 |
)
|
|
|
|
| 13 |
EmbeddingOptions,
|
| 14 |
)
|
| 15 |
|
| 16 |
+
from .requests import BaseEmbedRequest, EmbedRequest
|
| 17 |
|
| 18 |
from .responses import (
|
| 19 |
BaseEmbedResponse,
|
| 20 |
DenseEmbedResponse,
|
| 21 |
SparseEmbedResponse,
|
|
|
|
|
|
|
| 22 |
ModelsListResponse,
|
| 23 |
RootResponse,
|
| 24 |
)
|
src/models/schemas/requests.py
CHANGED
|
@@ -49,7 +49,7 @@ class EmbedRequest(BaseEmbedRequest):
|
|
| 49 |
"""
|
| 50 |
Request model for single/batch text and sparse embedding.
|
| 51 |
|
| 52 |
-
Used for /embed endpoint to process multiple texts at once.
|
| 53 |
|
| 54 |
Attributes:
|
| 55 |
texts: List of input texts to embed
|
|
@@ -61,7 +61,6 @@ class EmbedRequest(BaseEmbedRequest):
|
|
| 61 |
...,
|
| 62 |
description="List of input texts to generate embeddings for",
|
| 63 |
min_length=1,
|
| 64 |
-
max_length=100,
|
| 65 |
)
|
| 66 |
|
| 67 |
@field_validator("texts")
|
|
|
|
| 49 |
"""
|
| 50 |
Request model for single/batch text and sparse embedding.
|
| 51 |
|
| 52 |
+
Used for /embed and /query endpoint to process multiple texts at once.
|
| 53 |
|
| 54 |
Attributes:
|
| 55 |
texts: List of input texts to embed
|
|
|
|
| 61 |
...,
|
| 62 |
description="List of input texts to generate embeddings for",
|
| 63 |
min_length=1,
|
|
|
|
| 64 |
)
|
| 65 |
|
| 66 |
@field_validator("texts")
|
src/models/schemas/responses.py
CHANGED
|
@@ -27,7 +27,7 @@ class DenseEmbedResponse(BaseEmbedResponse):
|
|
| 27 |
"""
|
| 28 |
Response model for single/batch dense embeddings.
|
| 29 |
|
| 30 |
-
Used for /embed endpoint with dense models.
|
| 31 |
|
| 32 |
Attributes:
|
| 33 |
embeddings: List of generated dense embedding vectors
|
|
@@ -63,7 +63,7 @@ class SparseEmbedResponse(BaseEmbedResponse):
|
|
| 63 |
"""
|
| 64 |
Response model for single/batch sparse embeddings.
|
| 65 |
|
| 66 |
-
Used for /
|
| 67 |
|
| 68 |
Attributes:
|
| 69 |
embeddings: List of generated sparse embeddings
|
|
|
|
| 27 |
"""
|
| 28 |
Response model for single/batch dense embeddings.
|
| 29 |
|
| 30 |
+
Used for /embed & /query endpoint with dense models.
|
| 31 |
|
| 32 |
Attributes:
|
| 33 |
embeddings: List of generated dense embedding vectors
|
|
|
|
| 63 |
"""
|
| 64 |
Response model for single/batch sparse embeddings.
|
| 65 |
|
| 66 |
+
Used for /embed and /query endpoint with sparse models.
|
| 67 |
|
| 68 |
Attributes:
|
| 69 |
embeddings: List of generated sparse embeddings
|