File tree Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Original file line number Diff line number Diff line change @@ -227,13 +227,30 @@ class CompositeEmbedder(CamelBase):
227
227
RestEmbedder,
228
228
UserProvidedEmbedder,
229
229
]
230
+ search_embedder: Union[
231
+ OpenAiEmbedder,
232
+ HuggingFaceEmbedder,
233
+ OllamaEmbedder,
234
+ RestEmbedder,
235
+ UserProvidedEmbedder,
236
+ ]"""
237
+
238
+ source : str = "composite"
230
239
search_embedder : Union [
231
240
OpenAiEmbedder ,
232
241
HuggingFaceEmbedder ,
233
242
OllamaEmbedder ,
234
243
RestEmbedder ,
235
244
UserProvidedEmbedder ,
236
245
]
246
+ indexing_embedder : Union [
247
+ OpenAiEmbedder ,
248
+ HuggingFaceEmbedder ,
249
+ OllamaEmbedder ,
250
+ RestEmbedder ,
251
+ UserProvidedEmbedder ,
252
+ ]
253
+
237
254
238
255
# Type alias for the embedder union type
239
256
EmbedderType = Union [
Original file line number Diff line number Diff line change 6
6
CompositeEmbedder ,
7
7
HuggingFaceEmbedder ,
8
8
OpenAiEmbedder ,
9
+ PoolingType ,
9
10
UserProvidedEmbedder ,
10
11
)
11
12
@@ -104,7 +105,7 @@ def test_huggingface_embedder_format(empty_index):
104
105
assert embedders .embedders ["huggingface" ].distribution .mean == 0.5
105
106
assert embedders .embedders ["huggingface" ].distribution .sigma == 0.1
106
107
assert embedders .embedders ["huggingface" ].binary_quantized is False
107
- assert embedders .embedders ["huggingface" ].pooling is False
108
+ assert embedders .embedders ["huggingface" ].pooling is PoolingType . USE_MODEL
108
109
109
110
110
111
def test_ollama_embedder_format (empty_index ):
You can’t perform that action at this time.
0 commit comments